#!/usr/bin/python2.7
#_*_coding:utf-8 _*_
import requests,sys,json,logging,time
import urllib3
urllib3.disable_warnings()
reload(sys)
sys.setdefaultencoding('utf-8')
####这个位置配置测试增加log日志信息和日期
logfile = 'log.txt'
def write_log(log):
timestamp = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
log_with_time = { 'timestamp': timestamp, 'log': log }
with open(logfile, 'a') as f:
f.write(json.dumps(log_with_time) + '\n')
##############################
这个测试配置只增加log
#def write_log(log):
# with open('/tmp/zabbix_wechat.log', 'a') as f:
# f.write(json.dumps(log) + '\n')
def GetTokenFromServer(Corpid,Secret):
Url = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
Data = {
"corpid":Corpid,
"corpsecret":Secret
}
r = requests.get(url=Url,params=Data,verify=False)
write_log(r.json())
if r.json()['errcode'] != 0:
return False
else:
Token = r.json()['access_token']
file = open('/tmp/zabbix_wechat_config.json', 'w')
file.write(r.text)
file.close()
return Token
def SendMessage(User,Agentid,Subject,Content):
try:
file = open('/tmp/zabbix_wechat_config.json', 'r')
Token = json.load(file)['access_token']
file.close()
except:
Token = GetTokenFromServer(Corpid, Secret)
n = 0
Url = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=%s" % Token
Data = {
"touser": "******", # 企业号中的用户帐号，在zabbix用户Media中配置，如果配置不正常，将按部门发送。
#"totag": Tagid, # 企业号中的标签id，群发使用（推荐）
"toparty": 2, # 企业号中的部门id，群发时使用。
"msgtype": "text", # 消息类型。
"agentid": 1000003, # 企业号中的应用id。
"text": {
"content": Subject + '\n' + Content
},
"safe": "0"
}
r = requests.post(url=Url,data=json.dumps(Data, ensure_ascii=False),verify=False)
write_log(r.json())
while r.json()['errcode'] != 0 and n < 1:
n+=1
Token = GetTokenFromServer(Corpid, Secret)
if Token:
Url = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=%s" % Token
r = requests.post(url=Url,data=json.dumps(Data, ensure_ascii=False),verify=False)
write_log(r.json())
return r.json()
if __name__ == '__main__':
User = sys.argv[1] # zabbix传过来的第一个参数
Subject = str(sys.argv[2]) # zabbix传过来的第二个参数
Content = str(sys.argv[3]) # zabbix传过来的第三个参数
write_log('User: %s, Subject: %s, Content: %s' % (User, Subject, Content))
Corpid = "****************" # CorpID是企业号的标识
Secret = "****************************" # Secret是管理组凭证密钥
#Tagid = "*" # 通讯录标签ID
Agentid = "*******" # 应用ID
Partyid = "*" # 部门ID
Status = SendMessage(User,Agentid,Subject,Content)
write_log(str(Status))