Python工厂模式封装各类Webhook群聊机器人代码片段

　　引言

　　企业存在给 特定群组 自动推送消息的需求，比如：监控报警推送、销售线索推送、运营内容推送等。 你可以在群聊中添加一个自定义机器人，通过服务端调用 webhook 地址，即可将外部系统的通知消息即时推送到群聊中。

　　飞书自定义机器人

　　#!/usr/bin/python3

　　# -*- coding: utf-8 -*-

　　# @Author: Hui

　　# @Desc: { webhook机器人模块 }

　　# @Date: 2023/02/19 19:48

　　import hmac

　　import base64

　　import hashlib

　　import time

　　from urllib.parse import quote_plus

　　import requests

　　from exceptions.base import SendMsgException

　　class BaseChatBot(object):

　　    """群聊机器人基类"""

　　    def __init__(self, webhook_url: str, secret: str = None):

　　        """

　　        初始化机器人

　　        Args:

　　            webhook_url: 机器人webhook地址

　　            secret: 安全密钥

　　        """

　　        self.webhook_url = webhook_url

　　        self.secret = secret

　　    def _get_sign(self, timestamp: str, secret: str):

　　        """

　　        获取签名(NotImplemented)

　　        Args:

　　            timestamp: 签名时使用的时间戳

　　            secret: 签名时使用的密钥

　　        Returns:

　　        """

　　        raise NotImplementedError

　　    def send_msg(self, content: str, timeout=10):

　　        """

　　        发送消息(NotImplemented)

　　        Args:

　　            content: 消息内容

　　            timeout: 发送消息请求超时时间 默认10秒

　　        Returns:

　　        """

　　        raise NotImplementedError

　　class FeiShuChatBot(BaseChatBot):

　　    """飞书机器人"""

　　    def _get_sign(self, timestamp: str, secret: str) -> str:

　　        """

　　        获取签名

　　        把 timestamp + "\n" + 密钥 当做签名字符串，使用 HmacSHA256 算法计算签名，再进行 Base64 编码

　　        Args:

　　            timestamp: 签名时使用的时间戳

　　            secret: 签名时使用的密钥

　　        Returns: sign

　　        """

　　        string_to_sign = '{}\n{}'.format(timestamp, secret)

　　        hmac_code = hmac.new(string_to_sign.encode("utf-8"), digestmod=hashlib.sha256).digest()

　　        # 对结果进行base64处理

　　        sign = base64.b64encode(hmac_code).decode('utf-8')

　　        return sign

　　    def send_msg(self, content: str, timeout=10):

　　        """

　　        发送消息

　　        Args:

　　            content: 消息内容