Zabbix實現(xiàn)微信報警功能

#!/usr/bin/env python
# encoding: utf-8
# Create time 2016-10-08
#Auth chenpeng
import urllib2
import json
import sys
import time
class WebChat(object):
def __init__(self,CropID,Secret):
self.CropID = CropID
self.Secret = Secret
def Get_Token(self,info):
'''
:param info: 存儲執(zhí)行結果和執(zhí)行程序狀態(tài)碼code （0代表執(zhí)行成功，非零表示不成功）
:return:
'''
self.info = info
gurl = "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s" % (self.CropID,self.Secret)
try:
#通過Get方式獲取token
req = urllib2.Request(gurl)
response = urllib2.urlopen(req)
g_result = json.loads(response.read(),"UTF-8")
if g_result .has_key('access_token'):
self.info['result']= g_result ['access_token']
self.info['code'] = 0
else:
self.info['result'] = g_result
self.info['code'] = 1
except Exception,e:
self.info['code'] = 1
self.info['result'] = e
def Send_Msg(self,touser,toparty,agentid,access_token,content,info,*args,**kwargs):
'''
發(fā)送信息到微信
:param touser: 部門成員id，zabbix中定義的微信接收者,
成員ID列表（消息接收者，多個接收者用‘|'分隔，最多支持1000個）。
特殊情況：指定為@all，則向關注該企業(yè)應用的全部成員發(fā)送
:param toparty: 部門id，定義了范圍，組內成員都可接收到消息,
部門ID列表，多個接收者用‘|'分隔，最多支持100個。當touser為@all時忽略本參數(shù)
:param agentid: 企業(yè)應用的id，整型。可在應用的設置頁面查看
:param access_token: 根據(jù)CropID,Secret獲取的訪問token值
:param content: 濾出zabbix傳遞的第三個參數(shù),
表示發(fā)送微信消息的內容消息內容，最長不超過2048個字節(jié)，
注意：主頁型應用推送的文本消息在微信端最多只顯示20個字（包含中英文）
:param info: 返回執(zhí)行結果信息{'result':None,'code':None};'code':0或者非零 ;0表示成功 非零表示失敗
:param args:
:param kwargs:
:return:
'''
self.touser = touser
self.toparty = toparty
self.agentid = agentid
self.conntent = content
self.access_token = access_token
self.info = info
purl = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=%s" % (access_token)
data = {
"touser": "",
"toparty": "",
"totag": "", #標簽ID列表，多個接收者用‘|'分隔，最多支持100個。當touser為@all時忽略本參數(shù),非必須
"msgtype": "text", #必須
"agentid": "", #必須
"text": {
"content": "" #必須
},
"safe": "0" # 表示是否是保密消息，0表示否，1表示是，默認0
}
data['touser'] = self.touser
data['agentid'] = self.agentid
data['toparty'] = self.toparty
data['text']['content']=self.conntent
data = json.dumps(data,ensure_ascii=False)
try:
#通過PUT方式獲取發(fā)送數(shù)據(jù)
req = urllib2.Request(purl, data)
response = urllib2.urlopen(req)
res = json.loads(response.read())
self.info['code'] = res['errcode']
self.info['result'] = res['errmsg']
except Exception,e:
self.info['result'] = e
self.info['code'] = 1
if __name__ == '__main__':
reload(sys)
sys.setdefaultencoding('utf-8')
def log(date, touser, content,info):
'''
發(fā)送的日志打印日志
:param date: 時間
:param touser: 發(fā)送給誰
:param content: 發(fā)送的信息內容
:param info: 發(fā)送執(zhí)行的結果
:return:
'''
msg = '%s %s %s 發(fā)送結果 - %s
' % (date, touser, content, info)
with open('msg.log', 'a') as f:
f.write(msg)
agentid = sys.argv[1]
#agentid = 1
touser = 'xxxxxxx@qq.com'
toparty = ''
content = sys.argv[2:]
content = '
'.join(content)
#content = '測試'
CropID = 'xxxxxxxxxxxxxxxxxxx'
Secret = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'
info={'result':None,'code':None}
date = time.strftime('%Y-%m-%d %H:%M:%S')
res=WebChat(CropID,Secret)
res.Get_Token(info)
if info['code'] == 0:
access_token = info['result']
res.Send_Msg(touser=touser, toparty=toparty, agentid=agentid, access_token=access_token,
content=content,info=info)
if info['code'] == 0:
content = eval(content)
log(date, touser, content,info)
else:
log(date, touser, content, info)
else:
log(date,touser,content,info)