work_webchat_robot.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2019/9/18 2:35 PM
# @Author  : Criss Chan
# @Site    : https://blog.csdn.net/crisschan
# @File    : work_webchat.py
# @Software: PyCharm
# @instruction：使用企业微信的机器人群发消息，在企业微信群里面，选择添加一个机器人，然后复制对应的webhook地址
#               每个机器人发送的消息不能超过20条/分钟。

from enum import Enum
import requests
import json

class MSGTYPE(Enum):
    TEXT = 'text'  # 文本类型。对应消息的限制：文本内容，最长不超过2048个字节，必须是utf8编码
    MARKDOWN = 'markdown'  # markdown类型,对应消息内容限制：markdown内容，最长不超过4096个字节，必须是utf8编码
    NEWS = 'news'  # 图文消息，限制：图文消息，一个图文消息支持1到8条图文
    # title	是标题，不超过128个字节，超过会自动截断
    # description，描述不超过512个字节，超过会自动截断
    # url，点击后跳转的链接。
    # picurl图文消息的图片链接，支持JPG、PNG格式，较好的效果为大图 1068*455，小图150*150。


class workWechat(object):

    def __init__(self, webhook, msgtype, dictdetail, mentioned_mobile_list=''):
        '''

        :param webhook: 机器人的webhook地址
        :param msgtype: 消息类型，具体参见MSGTYPE枚举类型
        :param dictdetail: 消息主体内容
        :param mentioned_mobile_list: 手机号列表，提醒手机号对应的群成员(@某个成员)，@all表示提醒所有人(该参数只在消息类型为text的时候有效
        '''
        self.webhook = webhook
        self.msgtype = msgtype
        self.mentioned_mobile_list = mentioned_mobile_list

        self.detail = dictdetail

        self._send_msg()

    def _send_msg(self):
        headers = {'Content-Type': 'application/json'}
        payload = self._payload()

        res = requests.post(self.webhook, data=payload.encode('utf-8'), headers=headers)

        if json.loads(res.text)['errmsg']=='ok':
            print('scuccess')
        else:
            print('faile')

    def _payload(self):
        payload = ''

        mentioned_mobile = self.mentioned_mobile_list

        mentioned_mobile = '"mentioned_mobile_list":["' + mentioned_mobile + '"],'


        if self.msgtype is MSGTYPE.NEWS:
            payload = '{"msgtype": "' + self.msgtype.value + '","' + self.msgtype.value + '": {  ' + mentioned_mobile + ' "articles" : [' + self.detail + ']}}'
        elif self.msgtype is MSGTYPE.MARKDOWN:
            payload = '{"msgtype": "' + self.msgtype.value + '","' + self.msgtype.value + '": {  ' + mentioned_mobile + '"content":"' + self.detail + '"}}'
        elif self.msgtype is MSGTYPE.TEXT:
            payload = '{"msgtype": "' + self.msgtype.value + '","' + self.msgtype.value + '": { ' + mentioned_mobile + '"content":"' + self.detail + '"}}'
        else:
            payload = ''

        return payload