Python 发送企业微信单发和群发机器人

转载

mob604756f976e6 2021-08-09 08:09:00

import time
import requests
import json


class WeChat:
    def __init__(self):
        self.CORPID = 'ww2e1234567895498f5498f'  #企业ID，在管理后台获取
        self.CORPSECRET = 'xy11234567898hk_ecJ123456789DhKy4_1y12345OI'#自建应用的Secret，每个自建应用里都有单独的secret
        self.AGENTID = '1000002'  #应用ID，在后台应用中获取
        self.TOUSER = "maomao|dingding"  # 接收者用户名,多个用户用|分割

    def _get_access_token(self):
        url = 'https://qyapi.weixin.qq.com/cgi-bin/gettoken'
        values = {'corpid': self.CORPID,
                  'corpsecret': self.CORPSECRET,
                  }
        req = requests.post(url, params=values)
        data = json.loads(req.text)
        return data["access_token"]

    def get_access_token(self):
        try:
            with open('./tmp/access_token.conf', 'r') as f:
                t, access_token = f.read().split()
        except:
            with open('./tmp/access_token.conf', 'w') as f:
                access_token = self._get_access_token()
                cur_time = time.time()
                f.write('\t'.join([str(cur_time), access_token]))
                return access_token
        else:
            cur_time = time.time()
            if 0 < cur_time - float(t) < 7260:
                return access_token
            else:
                with open('./tmp/access_token.conf', 'w') as f:
                    access_token = self._get_access_token()
                    f.write('\t'.join([str(cur_time), access_token]))
                    return access_token

    def send_data(self, message):
        send_url = 'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + self.get_access_token()
        send_values = {
            "touser": self.TOUSER,
            "msgtype": "text",
            "agentid": self.AGENTID,
            "text": {
                "content": message
                },
            "safe": "0"
            }
        send_msges=(bytes(json.dumps(send_values), 'utf-8'))
        respone = requests.post(send_url, send_msges)
        respone = respone.json()   #当返回的数据是json串的时候直接用.json即可将respone转换成字典
        return respone["errmsg"]


if __name__ == '__main__':
    wx = WeChat()
    wx.send_data("这是程序发送的第1条消息！\n Python程序调用企业微信API,从自建应用“告警测试应用”发送给管理员的消息！")
    wx.send_data("这是程序发送的第2条消息！")

群发机器人：

import requests
import json

dsj_url = ''


def send_msg(send_message):
    data1 = json.dumps({'msgtype': "markdown",
                        "markdown": {
                            "content": send_message,
                            "mentioned_list": ["@all"]
                        }})
    # 指定机器人发送消息
    resp = requests.post(dsj_url, data1, auth=('Content-Type', 'application/json'))
    return '出现故障：%s，此次报警成功' % send_message


if __name__ == '__main__':
    # 报警机器人url
    send_message = '''【可疑】<font color=\"warning\">hids 检测到 异常登录！</font>\n
                      >源ip:<font color=\"comment\">10.70.234.21</font>
                      >目标IP:<font color=\"comment\">10.70.218.13</font>
                      >涉及资产:<font color=\"comment\">jssz-ai-newton-cpu-06</font>
                      >详情：
                      >登录主机所属业务组：<font color=\"comment\">未分组主机</font>
                      >登录使用账户:<font color=\"comment\">root</font>
                      >发现时间:<font color=\"comment\">2021-06-10 16:24:12</font>
                      '''
    res = send_msg(send_message)
    print(res)