进行一系列优化，且删除所有demo用例仅保持正常的示例

2023-11-10 15:58:03 +08:00
parent e97fe7290f
commit 4943adf36b
53 changed files with 906 additions and 868 deletions
@@ -0,0 +1,144 @@
+# -*- coding: utf-8 -*-
+# @Time    : 2023/5/10 10:43
+# @Author  : chenyinhua
+# @File    : send_result_handle.py
+# @Software: PyCharm
+# @Desc: 根据配置文件，发送指定类型的测试结果
+
+# 标准库导入
+# 第三方库导入
+from loguru import logger
+# 本地应用/模块导入
+from config.models import NotificationType
+from config.settings import SEND_RESULT_TYPE, email, ding_talk, wechat, email_subject, email_content, ding_talk_title, \
+    ding_talk_content, wechat_content
+from case_utils.requests_utils.data_handle import data_handle
+from case_utils.report_utils.get_results_handle import get_test_results_from_from_allure_report
+from common_utils.notify_utils.dingding_handle import DingTalkBot
+from common_utils.notify_utils.wechat_handle import WechatBot
+from common_utils.notify_utils.yagmail_handle import YagEmailServe
+
+
+def send_email(user, pwd, host, subject, content, to, attachments):
+    """
+    发送邮件
+    """
+    try:
+        yag = YagEmailServe(user=user, password=pwd, host=host)
+        info = {
+            "subject": subject,
+            "contents": content,
+            "to": to,
+            "attachments": attachments
+
+        }
+        yag.send_email(info)
+    except Exception as e:
+        logger.error(f"发送邮件通知异常， 错误信息：{e}")
+
+
+def send_dingding(webhook_url, secret, title, content):
+    """
+    发送钉钉消息
+    """
+    try:
+        dingding = DingTalkBot(webhook_url=webhook_url, secret=secret)
+        res = dingding.send_markdown(title=title, text=content, is_at_all=True)
+        if res:
+            logger.info(f"发送钉钉通知成功~")
+        else:
+            logger.error(f"发送钉钉通知失败~")
+    except Exception as e:
+        logger.error(f"发送钉钉通知异常， 错误信息：{e}")
+
+
+def send_wechat(webhook_url, content, attachment=None):
+    """
+    发送企业微信消息
+    """
+    try:
+        wechat = WechatBot(webhook_url=webhook_url)
+        msg = wechat.send_markdown(content=content)
+        if msg:
+            if attachment:
+                file = wechat.send_file(wechat.upload_file(attachment))
+                if file:
+                    logger.info(f"发送企业微信通知(包括文本以及附件)成功~")
+                else:
+                    logger.error(f"发送企业微信通知(附件)失败~")
+        else:
+            logger.error(f"发送企业微信（文本）失败~")
+    except Exception as e:
+        logger.error(f"发送企业微信通知异常， 错误信息：{e}")
+
+
+def send_result(report_path, attachment_path=None):
+    """
+    根据用户配置，采取指定方式，发送测试结果
+    :param report_path: 报告路径
+    :param attachment_path: 发送的附件， pytest-html就是报告本身作为附件发送， allure是压缩包发送
+    """
+    # 默认不发送任何通知
+    if SEND_RESULT_TYPE == NotificationType.DEFAULT.value:
+        logger.debug(f"SEND_RESULT_TYPE={SEND_RESULT_TYPE}， 配置了不发送任何邮件")
+        return
+
+    results = get_test_results_from_from_allure_report(report_path)
+
+    # 建立发送消息的内容、函数以及参数的映射关系
+    notification_mappings = {
+        NotificationType.EMAIL.value: {
+            'sender': send_email,
+            'sender_args': {
+                'user': email.get("user"),
+                'pwd': email.get("password"),
+                'host': email.get("host"),
+                'subject': email_subject,
+                'content': email_content,
+                'to': email.get("to"),
+                'attachments': attachment_path,
+            }
+        },
+        NotificationType.DING_TALK.value: {
+            'sender': send_dingding,
+            'sender_args': {
+                'webhook_url': ding_talk["webhook_url"],
+                'secret': ding_talk["secret"],
+                'title': ding_talk_title,
+                'content': ding_talk_content,
+            }
+        },
+        NotificationType.WECHAT.value: {
+            'sender': send_wechat,
+            'sender_args': {
+                'webhook_url': wechat["webhook_url"],
+                'content': wechat_content,
+                'attachment': attachment_path,
+            }
+        }
+    }
+    # 单一渠道发送消息
+    if SEND_RESULT_TYPE in notification_mappings:
+        notification = notification_mappings[SEND_RESULT_TYPE]
+        # 获取消息内容并替换
+        notification['sender_args']['content'] = data_handle(notification['sender_args']['content'],
+                                                             source=results)
+        # 获取消息发送函数
+        sender = notification['sender']
+        # 获取对应消息发送函数的参数
+        sender_args = notification['sender_args']
+        # 调用消息发送函数
+        sender(**sender_args)
+    # 全渠道发送消息
+    else:
+        # 遍历所有消息发送方式
+        for notification in notification_mappings.values():
+            # 获取消息内容并替换
+            notification['sender_args']['content'] = data_handle(notification['sender_args']['content'],
+                                                                 source=results)
+            # 获取消息发送函数
+            sender = notification['sender']
+            # 获取对应消息发送函数的参数
+            sender_args = notification['sender_args']
+            # 调用消息发送函数
+            sender(**sender_args)