时间:2021-07-01 10:21:17 帮助过:96人阅读
- #!/usr/bin/env python
- #-*- coding:utf-8 -*-
- import urllib2,json
- import datetime,time
- from config import *
- import sys
- reload(sys)
- sys.setdefaultencoding("utf-8")
- class WechatPush():
- def __init__(self,appid,secrect,file_name):
- # 传入appid
- self.appid = appid
- # 传入密码
- self.secrect = secrect
- # 传入记录token和过期时间的文件名
- self.file_name=file_name
- def build_timestamp(self,interval):
- # 传入时间间隔,得到指定interval后的时间 格式为"2015-07-01 14:41:40"
- now = datetime.datetime.now()
- delta = datetime.timedelta(seconds=interval)
- now_interval=now + delta
- return now_interval.strftime(‘%Y-%m-%d %H:%M:%S‘)
- def check_token_expires(self):
- # 判断token是否过期
- with open(self.file_name,‘r‘) as f:
- line=f.read()
- if len(line)>0:
- expires_time=line.split(",")[1]
- token=line.split(",")[0]
- else:
- return "","true"
- curr_time=time.strftime(‘%Y-%m-%d %H:%M:%S‘)
- # 如果过期返回false
- if curr_time>expires_time:
- return token,"false"
- # 没过期返回true
- else:
- return token,"true"
- def getToken(self):
- # 获取accessToken
- url = ‘https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=‘+self.appid + "&secret="+self.secrect
- try:
- f = urllib2.urlopen(url)
- s = f.read()
- # 读取json数据
- j = json.loads(s)
- j.keys()
- # 从json中获取token
- token = j[‘access_token‘]
- # 从json中获取过期时长
- expires_in =j[‘expires_in‘]
- # 将得到的过期时长减去300秒然后与当前时间做相加计算然后写入到过期文件
- write_expires=self.build_timestamp(int(expires_in-300))
- content="%s,%s" % (token,write_expires)
- with open(self.file_name,‘w‘) as f:
- f.write(content)
- except Exception,e:
- print e
- return token
- def post_data(self,url,para_dct):
- """触发post请求微信发送最终的模板消息"""
- para_data = para_dct
- f = urllib2.urlopen(url,para_data)
- content = f.read()
- return content
- def do_push(self,touser,template_id,url,topcolor,data):
- ‘‘‘推送消息 ‘‘‘
- #获取存入到过期文件中的token,同时判断是否过期
- token,if_token_expires=self.check_token_expires()
- #如果过期了就重新获取token
- if if_token_expires=="false":
- token=self.getToken()
- # 背景色设置,貌似不生效
- if topcolor.strip()==‘‘:
- topcolor = "#7B68EE"
- #最红post的求情数据
- dict_arr = {‘touser‘: touser, ‘template_id‘:template_id, ‘url‘:url, ‘topcolor‘:topcolor,‘data‘:data}
- json_template = json.dumps(dict_arr)
- requst_url = "https://api.weixin.qq.com/cgi-bin/message/template/send?access_token="+token
- content = self.post_data(requst_url,json_template)
- #读取json数据
- j = json.loads(content)
- j.keys()
- errcode = j[‘errcode‘]
- errmsg = j[‘errmsg‘]
- #print errmsg
- if __name__ == "__main__":
- def alarm(title,hostname,timestap,level,message,state,tail):
- """报警函数"""
- color="#FF0000"
- data={"first":{"value":title},"keyword1":{"value":hostname,"color":color},"keyword2":{"value":timestap,"color":color},"keyword3":{"value":level,"color":color},"keyword4":{"value":message,"color":color},"keyword5":{"value":state,"color":color},"remark":{"value":tail}}
- return data
- def recover(title,message,alarm_time,recover_time,continue_time,tail):
- """恢复函数"""
- re_color="#228B22"
- data={"first":{"value":title},"content":{"value":message,"color":re_color},"occurtime":{"value":alarm_time,"color":re_color},"recovertime":{"value":recover_time,"color":re_color},"lasttime":{"value":continue_time,"color":re_color},"remark":{"value":tail}}
- return data
- # data=alarm("测试的报警消息","8.8.8.8",time.ctime(),"最高级别","然并卵","挂了","大傻路赶紧处理")
- # 实例化类
- webchart=WechatPush(appid,secrect,file_name)
- url="http://www.xiaoniu88.com"
- print len(sys.argv)
- # 发送报警消息
- if len(sys.argv) == 9:
- title=sys.argv[1]
- hostname=sys.argv[2]
- timestap=sys.argv[3]
- level=sys.argv[4]
- message=sys.argv[5]
- state=sys.argv[6]
- tail=sys.argv[7]
- print "sys.argv[1]"+sys.argv[1]
- print "sys.argv[2]"+sys.argv[2]
- print "sys.argv[3]"+sys.argv[3]
- print "sys.argv[4]"+sys.argv[4]
- print "sys.argv[5]"+sys.argv[5]
- print "sys.argv[6]"+sys.argv[6]
- print "sys.argv[7]"+sys.argv[7]
- print "sys.argv[8]"+sys.argv[8]
- with open("/etc/zabbix/moniter_scripts/test.log",‘a+‘) as f:
- f.write(title+"\n")
- f.write(hostname+"\n")
- f.write(timestap+"\n")
- f.write(level+"\n")
- f.write(message+"\n")
- f.write(state+"\n")
- f.write(tail+"\n")
- f.write("%s_%s" % ("group",sys.argv[8])+"\n")
- data=alarm(title,hostname,timestap,level,message,state,tail)
- group_name="%s_%s" % ("group",sys.argv[8])
- for touser in eval("%s_%s" % ("group",sys.argv[8])):
- webchart.do_push(touser,alarm_id,url,"",data)
- for touser in group_super:
- webchart.do_push(touser,alarm_id,url,"",data)
- #发送恢复消息
- elif len(sys.argv) == 8:
- title=sys.argv[1]
- message=sys.argv[2]
- alarm_time=sys.argv[3]
- recover_time=sys.argv[4]
- continue_time=sys.argv[5]
- tail=sys.argv[6]
- print "sys.argv[1]"+sys.argv[1]
- print "sys.argv[2]"+sys.argv[2]
- print "sys.argv[3]"+sys.argv[3]
- print "sys.argv[4]"+sys.argv[4]
- print "sys.argv[5]"+sys.argv[5]
- print "sys.argv[6]"+sys.argv[6]
- print "sys.argv[7]"+sys.argv[7]
- data=recover(title,message,alarm_time,recover_time,continue_time,tail)
- for touser in eval("%s_%s" % ("group",sys.argv[7])):
- webchart.do_push(touser,recover_id,url,"",data)
- for touser in group_super:
- webchart.do_push(touser,recover_id,url,"",data)
好了,代码到此结束了,希望以上所述关于python模板消息的相关叙述能够给大家带来帮助。哪里写的不好,还请各位大侠多多见谅,提出宝贵意见,谢谢。