#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/8/20 上午12:42
# @Author  : wangying
# @Site    : 
# @File    : 企業(yè)微信.py
# @Software: PyCharm
 
#!/root/.virtualenvs/wechat/bin/python
# usage: send message via wechat
 
import requests, sys, json
import urllib3
urllib3.disable_warnings()
 
 
###填寫參數(shù)###
 
# Corpid是企業(yè)號的標識
Corpid = "ww32a580fceb30f350"
 
# Secret是管理組憑證密鑰
Secret = "QyBTbg1QL9lmNwaErqv1DMV0Y_G3lZre268E0hpTDqk"
 
# 應用ID
Agentid = "1000004"
 
# token_config文件放置路徑
Token_config = r'/tmp/zabbix_wechat_config.json'
 
###下面的代碼都不需要動###
 
 
def GetTokenFromServer(Corpid, Secret):
    """獲取access_token"""
    Url = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
    Data = {
        "corpid": Corpid,
        "corpsecret": Secret
    }
    r = requests.get(url=Url, params=Data, verify=False)
    print(r.json())
    if r.json()['errcode'] != 0:
        return False
    else:
        Token = r.json()['access_token']
        file = open(Token_config, 'w')
        file.write(r.text)
        file.close()
        return Token
 
 
def SendMessage(Partyid, Subject, Content):
    """發(fā)送消息"""
    # 獲取token信息
    try:
        file = open(Token_config, 'r')
        Token = json.load(file)['access_token']
        file.close()
    except:
        Token = GetTokenFromServer(Corpid, Secret)
 
    # 發(fā)送消息
    Url = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=%s" % Token
    Data = {
        "toparty": Partyid,
        "msgtype": "text",
        "agentid": Agentid,
        "text": {"content": Subject + '\n' + Content},
        "safe": "0"
    }
    r = requests.post(url=Url, data=json.dumps(Data), verify=False)
 
    # 如果發(fā)送失敗，將重試三次
    n = 1
    while r.json()['errcode'] != 0 and n < 4:
        n = 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), verify=False)
            print(r.json())
 
    return r.json()
 
 
if __name__ == '__main__':
    # 部門id
    Partyid = '20'
    # 消息標題
    Subject = '自應用程序代碼測試'
    # 消息內容
    Content = 'str(sys.argv[3])'
    Status = SendMessage(Partyid, Subject, Content)
    print(Status)