#!/usr/bin/env python#-*- coding:utf-8 -*-import urllib2,jsonimport datetime,timefrom config import *import sysreload(sys)sys.setdefaultencoding("utf-8")class WechatPush():  def __init__(self,appid,secrect,file_name):    # 傳入appid    self.appid = appid    # 傳入密碼    self.secrect = secrect    # 傳入記錄token和過期時(shí)間的文件名    self.file_name=file_name  def build_timestamp(self,interval):    # 傳入時(shí)間間隔,得到指定interval后的時(shí)間 格式為"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數(shù)據(jù)      j = json.loads(s)      j.keys()      # 從json中獲取token      token = j[‘a(chǎn)ccess_token‘]      # 從json中獲取過期時(shí)長(zhǎng)      expires_in =j[‘expires_in‘]      # 將得到的過期時(shí)長(zhǎng)減去300秒然后與當(dāng)前時(shí)間做相加計(jì)算然后寫入到過期文件      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):    """觸發(fā)post請(qǐng)求微信發(fā)送最終的模板消息"""    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,同時(shí)判斷是否過期    token,if_token_expires=self.check_token_expires()    #如果過期了就重新獲取token    if if_token_expires=="false":      token=self.getToken()    # 背景色設(shè)置,貌似不生效      if topcolor.strip()==‘‘:      topcolor = "#7B68EE"    #最紅post的求情數(shù)據(jù)    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數(shù)據(jù)    j = json.loads(content)    j.keys()    errcode = j[‘errcode‘]    errmsg = j[‘errmsg‘]    #print errmsgif __name__ == "__main__":  def alarm(title,hostname,timestap,level,message,state,tail):    """報(bào)警函數(shù)"""    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):    """恢復(fù)函數(shù)"""    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("測(cè)試的報(bào)警消息","8.8.8.8",time.ctime(),"最高級(jí)別","然并卵","掛了","大傻路趕緊處理")  # 實(shí)例化類  webchart=WechatPush(appid,secrect,file_name)  url="http://www.xiaoniu88.com"  print len(sys.argv)  # 發(fā)送報(bào)警消息  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(chǎn)+‘) 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)  #發(fā)送恢復(fù)消息  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)