#!/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: 存儲執(zhí)行結(jié)果和執(zhí)行程序狀態(tài)碼code （0代表執(zhí)行成功，非零表示不成功）: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):'''發(fā)送信息到微信:param touser: 部門成員id，zabbix中定義的微信接收者,成員ID列表（消息接收者，多個接收者用‘|'分隔，最多支持1000個）。特殊情況：指定為@all，則向關(guān)注該企業(yè)應用的全部成員發(fā)送:param toparty: 部門id，定義了范圍，組內(nèi)成員都可接收到消息,部門ID列表，多個接收者用‘|'分隔，最多支持100個。當touser為@all時忽略本參數(shù):param agentid: 企業(yè)應用的id，整型。可在應用的設置頁面查看:param access_token: 根據(jù)CropID,Secret獲取的訪問token值:param content: 濾出zabbix傳遞的第三個參數(shù),表示發(fā)送微信消息的內(nèi)容消息內(nèi)容，最長不超過2048個字節(jié)，注意：主頁型應用推送的文本消息在微信端最多只顯示20個字（包含中英文）:param info: 返回執(zhí)行結(jié)果信息{'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時忽略本參數(shù),非必須"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方式獲取發(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'] = eself.info['code'] = 1if __name__ == '__main__':reload(sys)sys.setdefaultencoding('utf-8')def log(date, touser, content,info):'''發(fā)送的日志打印日志:param date: 時間:param touser: 發(fā)送給誰:param content: 發(fā)送的信息內(nèi)容:param info: 發(fā)送執(zhí)行的結(jié)果:return:'''msg = '%s %s %s 發(fā)送結(jié)果 - %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)