feishu.py

# -*- coding:utf-8 -*-
'''
@File      :   feiShuTalk.py 
@Time      :   2020/11/9 11:45   
@Author    :   DY
@Version   :   V1.0.0
@Desciption:  
'''

import requests
import json
from loguru import logger
import time
import urllib
import urllib3
urllib3.disable_warnings()


try:
    JSONDecodeError = json.decoder.JSONDecodeError
except AttributeError:
    JSONDecodeError = ValueError


def is_not_null_and_blank_str(content):
    """
    非空字符串
    :param content: 字符串
    :return: 非空 - True，空 - False
    """
    if content and content.strip():
        return True
    else:
        return False


class FeiShutalkChatbot(object):

    def __init__(self, webhook, secret=None, pc_slide=False, fail_notice=False):
        '''
        机器人初始化
        :param webhook: 飞书群自定义机器人webhook地址
        :param secret:  机器人安全设置页面勾选“加签”时需要传入的密钥
        :param pc_slide:  消息链接打开方式，默认False为浏览器打开，设置为True时为PC端侧边栏打开
        :param fail_notice:  消息发送失败提醒，默认为False不提醒，开发者可以根据返回的消息发送结果自行判断和处理
        '''
        super(FeiShutalkChatbot, self).__init__()
        self.headers = {'Content-Type': 'application/json; charset=utf-8'}
        self.webhook = webhook
        self.secret = secret
        self.pc_slide = pc_slide
        self.fail_notice = fail_notice

    def send_text(self, msg, open_id=[]):
        """
        消息类型为text类型
        :param msg: 消息内容
        :return: 返回消息发送结果
        """
        data = {"msg_type": "text", "at": {}}
        if is_not_null_and_blank_str(msg):    # 传入msg非空
            data["content"] = {"text": msg}
        else:
            logger.error("text类型，消息内容不能为空！")
            raise ValueError("text类型，消息内容不能为空！")

        logger.debug('text类型：%s' % data)
        return self.post(data)

    def send_interactive(self, title, text):
        """
        消息类型为interactive类型
        :param title: 消息标题
        :param text: 消息内容
        :return: 返回消息发送结果
        """
        data = {"msg_type": "interactive", "card":
                {"config": {"wide_screen_mode": True},
                "elements": [{"tag": "div", "text": {"content": text, "tag": "lark_md"}}], 
                "header": {"template": "blue", "title": {"content": "🎉 " + title, "tag": "plain_text"}}}}
                
        # dic 转成 JSON 字符串 utf-8
        data = json.dumps(data, ensure_ascii=False)
        return self.post(data.encode('utf-8'))

    def send_post(self, title, text):
        """
        消息类型为post类型
        :param title: 消息标题
        :param text: 消息内容
        :return: 返回消息发送结果
        """
        data = {"msg_type": "post", "content": {"post": {
            "zh_cn": {"title": title, "content": [[{"tag": "text", "text": text}]]}}}}
        # dic 转成 JSON 字符串 utf-8
        data = json.dumps(data, ensure_ascii=False)
        return self.post(data.encode('utf-8'))

    def post(self, data):
        """
        发送消息（内容UTF-8编码）
        :param data: 消息数据（字典）
        :return: 返回消息发送结果
        """
        try:
            response = requests.post(
                self.webhook, headers=self.headers, data=data, verify=False)
        except requests.exceptions.HTTPError as exc:
            logger.error("消息发送失败， HTTP error: %d, reason: %s" %
                         (exc.response.status_code, exc.response.reason))
            raise
        except requests.exceptions.ConnectionError:
            logger.error("消息发送失败，HTTP connection error!")
            raise
        except requests.exceptions.Timeout:
            logger.error("消息发送失败，Timeout error!")
            raise
        except requests.exceptions.RequestException:
            logger.error("消息发送失败, Request Exception!")
            raise
        else:
            try:
                result = response.json()
            except JSONDecodeError:
                logger.error("服务器响应异常，状态码：%s，响应内容：%s" %
                             (response.status_code, response.text))
                return {'errcode': 500, 'errmsg': '服务器响应异常'}
            else:
                logger.debug('发送结果：%s' % result)
                # 消息发送失败提醒（errcode 不为 0，表示消息发送异常），默认不提醒，开发者可以根据返回的消息发送结果自行判断和处理
                if self.fail_notice and result.get('errcode', True):
                    time_now = time.strftime(
                        "%Y-%m-%d %H:%M:%S", time.localtime(time.time()))
                    error_data = {
                        "msgtype": "text",
                        "text": {
                            "content": "[注意-自动通知]飞书机器人消息发送失败，时间：%s，原因：%s，请及时跟进，谢谢!" % (
                                time_now, result['errmsg'] if result.get('errmsg', False) else '未知异常')
                        },
                        "at": {
                            "isAtAll": False
                        }
                    }
                    logger.error("消息发送失败，自动通知：%s" % error_data)
                    requests.post(self.webhook, headers=self.headers,
                                  data=json.dumps(error_data))
                return result


# main 函数
if __name__ == '__main__':
    feishu = FeiShutalkChatbot(
        "https://open.feishu.cn/open-apis/bot/v2/hook/79129e41-d4f3-429a-8963-ba04a4dcf4ed")
    feishu.send_post("测试消息", "测试消息")
    feishu.send_interactive("测试消息", "**测试**消息")