#!/usr/bin/python
# coding: utf-8




import urllib,urllib2
import json
import sys
 
def gettoken(corpid,corpsecret):
    gettoken_url = 'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=' + corpid + '&corpsecret=' + corpsecret
    try:
  token_file = urllib2.urlopen(gettoken_url)
    except urllib2.HTTPError as e:
        print e.code
        print e.read().decode("utf8")
  sys.exit()
    token_data = token_file.read().decode('utf-8')
    token_json = json.loads(token_data)
    token_json.keys()
    token = token_json['access_token']
    return token
 
def senddata(access_token,user,content):
    send_url = 'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + access_token
    send_values = {
        "touser":user,    #企業號中的使用者帳號，在zabbix使用者Media中配置，如果配置不正常，將按部門傳送。
        "toparty":"1",    #企業號中的部門id
        "msgtype":"text",  #企業號中的應用id，訊息型別。
        "agentid":"1",
        "text":{
            "content":content
           },
        "safe":"0"
        }
    send_data = json.dumps(send_values, ensure_ascii=False)
    send_request = urllib2.Request(send_url, send_data)
    response = json.loads(urllib2.urlopen(send_request).read())
    print str(response)
 
if __name__ == '__main__':
    user = str(sys.argv[1])   #zabbix傳過來的第一個引數
    content = str(sys.argv[2])  #zabbix傳過來的第三個引數
    corpid = 'xxxx'   #CorpID是企業號的標識
    corpsecret = 'xxxx'  #corpsecretSecret是管理組憑證金鑰
    accesstoken = gettoken(corpid,corpsecret)
    senddata(accesstoken,user,content)