FastapiAdmin/backend/app/utils/ding_talk_util.py

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

from typing import Any
from dingtalkchatbot.chatbot import DingtalkChatbot, FeedLink

# 钉钉发送消息模块
class DingTalkPack:

    def __init__(self, webhook: str, secret: str, at_mobiles: list = None):

        self.webhook=webhook
        self.secret=secret
        self.at_mobiles=at_mobiles
        
        self.ding_news = DingtalkChatbot( webhook=self.webhook, secret=self.secret, pc_slide=False, fail_notice=False)

    def send_text(self, msg: str, mobiles: list = None) -> None:
        """
        发送文本信息
        :param msg: 文本内容
        :param mobiles: 艾特用户电话
        :return:
        """
        if not mobiles:
            self.ding_news.send_text(msg=msg, is_at_all=True)
        else:
            if isinstance(mobiles, list):
                self.ding_news.send_text(msg=msg, at_mobiles=mobiles)
            else:
                raise TypeError("mobiles类型错误 不是list类型.")

    def send_link(self, title: str, text: str, message_url: str, pic_url: str) -> None:
        """
        发送link通知
        :return:
        """
        try:
            self.ding_news.send_link(title=title, text=text, message_url=message_url, pic_url=pic_url)
        except Exception:
            raise Exception("发送link类型消息失败")

    def send_markdown(self, title: str, msg: str, mobiles: list = None) -> None:
        """
        :param mobiles:
        :param title:
        :param msg:
        markdown 格式
        """
        if mobiles is None:
            self.ding_news.send_markdown(title=title, text=msg, is_at_all=True)
        else:
            if isinstance(mobiles, list):
                self.ding_news.send_markdown(title=title, text=msg, at_mobiles=mobiles)
            else:
                raise TypeError("mobiles类型错误 不是list类型.")

    def feed_link(self, title: str, message_url: str, pic_url: str) -> Any:
        """
        发送link类型
        :param title:
        :param message_url:
        :param pic_url:
        :return:
        """
        return FeedLink(title=title, message_url=message_url, pic_url=pic_url)

    def send_feed_link(self, *arg) -> None:
        """
        发送feedlink类型
        :param arg:
        :return:
        """
        try:
            self.ding_news.send_feed_card(list(arg))
        except Exception:
            raise Exception("发送feedlink类型消息失败")

    def send_dingding(self,
                      title,
                      environment,
                      tester,
                      total,
                      pass_num,
                      fail_num,
                      error_num,
                      skip_num,
                      rate,
                      duration,
                      ):
        """
        发送钉钉通知
        :return:
        """

        self.ding_news.send_markdown(
            title=f'【{title}测试执行完毕提醒',
            text=f'''<font color='#FFA500'>[通知] </font>测试执行完成 \n\n
                    --- \n\n
                    执行环境：<font color="#1d953f">{environment}</font>\n\n
                    执行人员：<font color="#f2eada">@{tester}</font>\n\n
                    --- \n\n
                    运行总数：<font color="#d71345">{total}</font>\n\n
                    通过数量：<font color="#1d953f">{pass_num}</font>\n\n
                    失败数量：<font color="#c63c26">{fail_num}</font>\n\n
                    异常数量：<font color="#fdb933">{error_num}</font>\n\n
                    跳过数量：<font color="#2585a6">{skip_num}</font>\n\n
                    成功率为：<font color="#1d953f">{rate}</font>\n\n
                    运行时间：<font color="#464547">{duration}</font>\n\n
                    **********************************\n\n
                    详细情况可登录平台查看，非相关负责人员可忽略此消息。谢谢。
                    ''',
            at_mobiles=self.at_mobiles
        )