Zabbix实现微信报警功能

#!/usr/bin/env python# encoding: utf-8# Create time 2016-10-08#Auth chenpengimport urllib2import jsonimport sysimport timeclass WebChat(object):def __init__(self,CropID,Secret):self.CropID = CropIDself.Secret = Secretdef Get_Token(self,info):''':param info: 存储执行结果和执行程序状态码code （0代表执行成功，非零表示不成功）:return:'''self.info = infogurl = "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s" % (self.CropID,self.Secret)try:#通过Get方式获取tokenreq = 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'] = 0else:self.info['result'] = g_resultself.info['code'] = 1except Exception,e:self.info['code'] = 1self.info['result'] = edef Send_Msg(self,touser,toparty,agentid,access_token,content,info,*args,**kwargs):'''发送信息到微信:param touser: 部门成员id，zabbix中定义的微信接收者,成员ID列表（消息接收者，多个接收者用‘|'分隔，最多支持1000个）。特殊情况：指定为@all，则向关注该企业应用的全部成员发送:param toparty: 部门id，定义了范围，组内成员都可接收到消息,部门ID列表，多个接收者用‘|'分隔，最多支持100个。当touser为@all时忽略本参数:param agentid: 企业应用的id，整型。可在应用的设置页面查看:param access_token: 根据CropID,Secret获取的访问token值:param content: 滤出zabbix传递的第三个参数,表示发送微信消息的内容消息内容，最长不超过2048个字节，注意：主页型应用推送的文本消息在微信端最多只显示20个字（包含中英文）:param info: 返回执行结果信息{'result':None,'code':None};'code':0或者非零 ;0表示成功 非零表示失败:param args::param kwargs::return:'''self.touser = touserself.toparty = topartyself.agentid = agentidself.conntent = contentself.access_token = access_tokenself.info = infopurl = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=%s" % (access_token)data = {"touser": "","toparty": "","totag": "", #标签ID列表，多个接收者用‘|'分隔，最多支持100个。当touser为@all时忽略本参数,非必须"msgtype": "text", #必须"agentid": "", #必须"text": {"content": "" #必须},"safe": "0" # 表示是否是保密消息，0表示否，1表示是，默认0}data['touser'] = self.touserdata['agentid'] = self.agentiddata['toparty'] = self.topartydata['text']['content']=self.conntentdata = json.dumps(data,ensure_ascii=False)try:#通过PUT方式获取发送数据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'] = eself.info['code'] = 1if __name__ == '__main__':reload(sys)sys.setdefaultencoding('utf-8')def log(date, touser, content,info):'''发送的日志打印日志:param date: 时间:param touser: 发送给谁:param content: 发送的信息内容:param info: 发送执行的结果:return:'''msg = '%s %s %s 发送结果 - %s/n' % (date, touser, content, info)with open('msg.log', 'a') as f:f.write(msg)agentid = sys.argv[1]#agentid = 1touser = 'xxxxxxx@qq.com'toparty = ''content = sys.argv[2:]content = '/n'.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)