当前位置:Gxlcms > Python > 基于python实现微信模板消息

基于python实现微信模板消息

时间:2021-07-01 10:21:17 帮助过:96人阅读

我的风格,废话不多说了,直接给大家贴代码了,并在一些难点上给大家附了注释,具体代码如下所示:

  1. #!/usr/bin/env python
  2. #-*- coding:utf-8 -*-
  3. import urllib2,json
  4. import datetime,time
  5. from config import *
  6. import sys
  7. reload(sys)
  8. sys.setdefaultencoding("utf-8")
  9. class WechatPush():
  10. def __init__(self,appid,secrect,file_name):
  11. # 传入appid
  12. self.appid = appid
  13. # 传入密码
  14. self.secrect = secrect
  15. # 传入记录token和过期时间的文件名
  16. self.file_name=file_name
  17. def build_timestamp(self,interval):
  18. # 传入时间间隔,得到指定interval后的时间 格式为"2015-07-01 14:41:40"
  19. now = datetime.datetime.now()
  20. delta = datetime.timedelta(seconds=interval)
  21. now_interval=now + delta
  22. return now_interval.strftime(‘%Y-%m-%d %H:%M:%S‘)
  23. def check_token_expires(self):
  24. # 判断token是否过期
  25. with open(self.file_name,‘r‘) as f:
  26. line=f.read()
  27. if len(line)>0:
  28. expires_time=line.split(",")[1]
  29. token=line.split(",")[0]
  30. else:
  31. return "","true"
  32. curr_time=time.strftime(‘%Y-%m-%d %H:%M:%S‘)
  33. # 如果过期返回false
  34. if curr_time>expires_time:
  35. return token,"false"
  36. # 没过期返回true
  37. else:
  38. return token,"true"
  39. def getToken(self):
  40. # 获取accessToken
  41. url = ‘https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=‘+self.appid + "&secret="+self.secrect
  42. try:
  43. f = urllib2.urlopen(url)
  44. s = f.read()
  45. # 读取json数据
  46. j = json.loads(s)
  47. j.keys()
  48. # 从json中获取token
  49. token = j[‘access_token‘]
  50. # 从json中获取过期时长
  51. expires_in =j[‘expires_in‘]
  52. # 将得到的过期时长减去300秒然后与当前时间做相加计算然后写入到过期文件
  53. write_expires=self.build_timestamp(int(expires_in-300))
  54. content="%s,%s" % (token,write_expires)
  55. with open(self.file_name,‘w‘) as f:
  56. f.write(content)
  57. except Exception,e:
  58. print e
  59. return token
  60. def post_data(self,url,para_dct):
  61. """触发post请求微信发送最终的模板消息"""
  62. para_data = para_dct
  63. f = urllib2.urlopen(url,para_data)
  64. content = f.read()
  65. return content
  66. def do_push(self,touser,template_id,url,topcolor,data):
  67. ‘‘‘推送消息 ‘‘‘
  68. #获取存入到过期文件中的token,同时判断是否过期
  69. token,if_token_expires=self.check_token_expires()
  70. #如果过期了就重新获取token
  71. if if_token_expires=="false":
  72. token=self.getToken()
  73. # 背景色设置,貌似不生效
  74. if topcolor.strip()==‘‘:
  75. topcolor = "#7B68EE"
  76. #最红post的求情数据
  77. dict_arr = {‘touser‘: touser, ‘template_id‘:template_id, ‘url‘:url, ‘topcolor‘:topcolor,‘data‘:data}
  78. json_template = json.dumps(dict_arr)
  79. requst_url = "https://api.weixin.qq.com/cgi-bin/message/template/send?access_token="+token
  80. content = self.post_data(requst_url,json_template)
  81. #读取json数据
  82. j = json.loads(content)
  83. j.keys()
  84. errcode = j[‘errcode‘]
  85. errmsg = j[‘errmsg‘]
  86. #print errmsg
  87. if __name__ == "__main__":
  88. def alarm(title,hostname,timestap,level,message,state,tail):
  89. """报警函数"""
  90. color="#FF0000"
  91. 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}}
  92. return data
  93. def recover(title,message,alarm_time,recover_time,continue_time,tail):
  94. """恢复函数"""
  95. re_color="#228B22"
  96. 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}}
  97. return data
  98. # data=alarm("测试的报警消息","8.8.8.8",time.ctime(),"最高级别","然并卵","挂了","大傻路赶紧处理")
  99. # 实例化类
  100. webchart=WechatPush(appid,secrect,file_name)
  101. url="http://www.xiaoniu88.com"
  102. print len(sys.argv)
  103. # 发送报警消息
  104. if len(sys.argv) == 9:
  105. title=sys.argv[1]
  106. hostname=sys.argv[2]
  107. timestap=sys.argv[3]
  108. level=sys.argv[4]
  109. message=sys.argv[5]
  110. state=sys.argv[6]
  111. tail=sys.argv[7]
  112. print "sys.argv[1]"+sys.argv[1]
  113. print "sys.argv[2]"+sys.argv[2]
  114. print "sys.argv[3]"+sys.argv[3]
  115. print "sys.argv[4]"+sys.argv[4]
  116. print "sys.argv[5]"+sys.argv[5]
  117. print "sys.argv[6]"+sys.argv[6]
  118. print "sys.argv[7]"+sys.argv[7]
  119. print "sys.argv[8]"+sys.argv[8]
  120. with open("/etc/zabbix/moniter_scripts/test.log",‘a+‘) as f:
  121. f.write(title+"\n")
  122. f.write(hostname+"\n")
  123. f.write(timestap+"\n")
  124. f.write(level+"\n")
  125. f.write(message+"\n")
  126. f.write(state+"\n")
  127. f.write(tail+"\n")
  128. f.write("%s_%s" % ("group",sys.argv[8])+"\n")
  129. data=alarm(title,hostname,timestap,level,message,state,tail)
  130. group_name="%s_%s" % ("group",sys.argv[8])
  131. for touser in eval("%s_%s" % ("group",sys.argv[8])):
  132. webchart.do_push(touser,alarm_id,url,"",data)
  133. for touser in group_super:
  134. webchart.do_push(touser,alarm_id,url,"",data)
  135. #发送恢复消息
  136. elif len(sys.argv) == 8:
  137. title=sys.argv[1]
  138. message=sys.argv[2]
  139. alarm_time=sys.argv[3]
  140. recover_time=sys.argv[4]
  141. continue_time=sys.argv[5]
  142. tail=sys.argv[6]
  143. print "sys.argv[1]"+sys.argv[1]
  144. print "sys.argv[2]"+sys.argv[2]
  145. print "sys.argv[3]"+sys.argv[3]
  146. print "sys.argv[4]"+sys.argv[4]
  147. print "sys.argv[5]"+sys.argv[5]
  148. print "sys.argv[6]"+sys.argv[6]
  149. print "sys.argv[7]"+sys.argv[7]
  150. data=recover(title,message,alarm_time,recover_time,continue_time,tail)
  151. for touser in eval("%s_%s" % ("group",sys.argv[7])):
  152. webchart.do_push(touser,recover_id,url,"",data)
  153. for touser in group_super:
  154. webchart.do_push(touser,recover_id,url,"",data)

好了,代码到此结束了,希望以上所述关于python模板消息的相关叙述能够给大家带来帮助。哪里写的不好,还请各位大侠多多见谅,提出宝贵意见,谢谢。

人气教程排行