mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-27 22:55:59 +00:00
100 lines
3.3 KiB
Python
Executable File
100 lines
3.3 KiB
Python
Executable File
# -*- coding: utf-8 -*-
|
|
|
|
import time
|
|
import smtplib
|
|
from email.header import Header
|
|
from email.mime.application import MIMEApplication
|
|
from email.mime.multipart import MIMEMultipart
|
|
from email.mime.text import MIMEText
|
|
|
|
from app.core.exceptions import CustomException
|
|
|
|
class EmailPack:
|
|
# 初始化发件人，密码，收件人列表
|
|
def __init__(self, fromaddr: str, password: str, toaddrs: list, server_host: str):
|
|
"""
|
|
:param REPORT_END_PATH:
|
|
"""
|
|
self.fromaddr = fromaddr
|
|
self.password = password
|
|
self.toaddrs = toaddrs
|
|
self.server_host = server_host
|
|
|
|
self.server = smtplib.SMTP(self.server_host)
|
|
self.message = MIMEMultipart() # 邮件体
|
|
|
|
# 设置发件人名称，主题，内容，附件
|
|
def _set_message(self, name: str, title: str, content: str, filelist: list):
|
|
"""
|
|
:param name:
|
|
:param title:
|
|
:param content:
|
|
:param filelist:
|
|
:return:
|
|
"""
|
|
tm = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time()))
|
|
self.message['From'] = Header(f"{name}<{self.fromaddr}>", 'utf-8') # 发件人名称和地址
|
|
self.message['Subject'] = Header(title + "_" + tm, 'utf-8') # 邮件主题
|
|
self.message.attach(MIMEText(content)) # 邮件内容
|
|
if filelist is not None: # 邮件附件
|
|
for file in filelist:
|
|
fileApart = MIMEApplication( open(file, 'rb').read(), file.split('.')[-1])
|
|
fileApart.add_header('Content-Disposition', 'attachment', filename=file.split("\\")[-1])
|
|
self.message.attach(fileApart)
|
|
|
|
# 发送邮件
|
|
def _send_message(self):
|
|
"""
|
|
:return:
|
|
"""
|
|
try:
|
|
self.server.login(self.fromaddr, self.password)
|
|
self.server.sendmail(self.fromaddr, self.toaddrs, self.message.as_string())
|
|
self.server.quit()
|
|
except smtplib.SMTPException as e:
|
|
raise CustomException(msg=f'邮件发送异常！{e}')
|
|
|
|
# 默认发送邮件
|
|
def send_default_email(
|
|
self,
|
|
title: str,
|
|
environment: str,
|
|
tester: str,
|
|
total: int,
|
|
pass_num: int,
|
|
fail_num: int,
|
|
error_num: int,
|
|
skip_num: int,
|
|
rate: str,
|
|
duration: str,
|
|
):
|
|
"""
|
|
:rtype: object
|
|
:return:
|
|
"""
|
|
|
|
self._set_message(
|
|
name="自动化测试",
|
|
title=f'{title}测试执行完毕提醒！',
|
|
content=f'''
|
|
各位同事, 大家好:
|
|
|
|
自动化用例执行完成，执行结果如下:
|
|
**********************************
|
|
执行环境: {environment}
|
|
执行人员: {tester}
|
|
运行总数: {total}
|
|
通过: {pass_num}
|
|
失败: {fail_num}
|
|
异常: {error_num}
|
|
跳过: {skip_num}
|
|
成功率: {rate}
|
|
总耗时: {duration}
|
|
**********************************
|
|
详细情况可登录平台查看，非相关负责人员可忽略此消息。谢谢。
|
|
''',
|
|
filelist=["xxx"]
|
|
)
|
|
self._send_message()
|
|
|