python发送邮件

最新推荐文章于 2023-12-21 11:00:00 发布

富贵诶

最新推荐文章于 2023-12-21 11:00:00 发布

阅读量746

点赞数

文章标签： python3 自动化

本文链接：https://blog.csdn.net/qq_36562656/article/details/93495893

版权

本文介绍了Python使用smtplib和email模块发送HTML格式邮件和带附件的邮件的方法，并展示了如何找到最新测试报告并结合自动化测试项目实现邮件发送功能。

摘要由CSDN通过智能技术生成

对于一个自动化测试工程师来说，发送邮件是脚本中的基本功能。在python中，发送邮件需要用到smtplib模块，它对SMTP 协议进行了简单的封装。
发送邮件时会涉及诸多需求，例如，邮件的正文的格式、是否带图片、邮件是否需要添加附件（及多附件）、邮件是否需要
同时向多人发送等。

1、发送HTML格式的邮件

import smtplib
from email.mime.text import MIMEText
from email.header import Header

# 发送邮箱服务器
smtpserver = 'smtp.163.com'
# 发送邮箱用户/授权码
user = 'username@163.com'
password = '123456'
# 接收邮箱
receiver = '1234567890@qq.com'
# 发送邮件主题
subject = 'python发来的邮件'
# 编写HTML类型的邮件正文
msg = MIMEText('<html><h1>你好！</h1></html>','html','utf-8')
msg['Subject'] = Header(subject, 'utf-8')
# 连接发送邮件
smtp = smtplib.SMTP()
smtp.connect(smtpserver)
smtp.login(user, password)
smtp.sendmail(user , receiver, msg.as_string())
smtp.quit()

除SMTP模块外，我们还用到了email模块，它主要用来定义邮件的标题和正文；
Header()方法用来定义邮件标题；
MIMEText()用于定义邮件正文，参数为html格式的文本。

2、发送带附件的邮件

在发送文件时，有时需要发送附件。下面的实例实现了带附件的邮件发送。

import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart

# 发送邮箱服务器
smtpserver = 'smtp.163.com'
# 发送邮箱用户/授权码
user = 'username@163.com'
password = '123456'
# 接收邮箱
receiver = '1234567890@qq.com'
# 邮件主题
subject = 'Python send email test'
# 发送的附件
sendfile = open('D:\\testpro\\report\\log.txt', 'rb').read()
att = MIMEText(sendfile, 'base64', 'utf-8')
att["Content-Type"] = 'application/octet-stream'
att["Content-Disposition"] = 'attachment; filename="log.txt"'
msg = MIMEMultipart('related')
msg['Subject'] = subject
msg.attach(att)
smtp = smtplib.SMTP()
smtp.connect(smtpserver)
smtp.login(user, password)
smtp.sendmail(user , receiver, msgRoot.as_string())
smtp.quit()

相比上一个实例，通过MIMEMultipart()模块构造的带附件的邮件

3、查找最新的测试报告

现在已经知道如何通过Python编写发邮件程序，但要想和自动化测试项目结合还需要解决一个问题，因为测试报告的名称是根据当前时间生成的，所以如何找到最新生成的测试报告是实现发邮件功能的关键。

import os

# 定义文件目录
result_dir = 'D:\\test\\report'
lists = os.listdir(result_dir)
# 重新按时间对目录下的文件进行排序
lists.sort(key=lambda fn: os.path.getmtime(result_dir+"\\"+fn))
print(('最新的文件为： ' + lists[-1]))
file = os.path.join(result_dir, lists[-1])
print(file)

首先定义测试报告的目录result_dir，os.listdir()可以获取目录下的所有文件及文件夹。利用sort()方法对目录下的文件及文件夹按时间重新排序。list[-1]取到的就是最新生成的文件或文件夹。

4、实际代码运用

项目结构
在这里插入图片描述
代码：
SendMail.py

import os
import smtplib
from email.header import Header
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from logs import logs
from data import User

# path = os.path.dirname(os.path.abspath(__file__))
# 获取上上级目录
path = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))


# 查找最新的测试报告
def newReport():
    # 定义文件目录
    result_dir = path + '\\report'
    lists = os.listdir(result_dir)
    # 重新按时间对目录下的文件进行排序
    lists.sort(key=lambda fn: os.path.getmtime(result_dir + "\\" + fn))
    # print(('最新的文件为： ' + lists[-1]))
    file = os.path.join(result_dir, lists[-1])
    # print(file)
    return file


# 上传附件
def att(path, name):
    sendfile = open(path, 'rb').read()
    att = MIMEText(sendfile, 'base64 ', 'utf-8')
    att['Content-type'] = 'application/octet-stream'
    # 附件在邮件中显示的名字
    att['Content-Disposition'] = 'attachment;filename= "' + name + '"'
    return att


def sendMail():
    smtpserver = User.smtpserver  # 发送者邮件服务器
    mailuser = User.mailuser  # 发送着邮件账户
    password = User.password  # 发送者邮件授权码
    receiver = User.receiver  # 接收者邮件账户
    subject = User.subject  # 邮件标题
    mailbody = User.mailbody  # 邮件正文
    reportAttachment = User.reportAttachment  # 报告附件名称
    AlllogAttachment = User.AlllogAttachment  # 所有日志附件名称
    ErrorlogAttachment = User.ErrorlogAttachment  # 错误日志附件名称
    file1 = newReport()
    file2 = path + '\\logs\\Error_Logs\\Errorlog.log'
    file3 = path + '\\logs\\All_Logs\\Alllog.log'
    msg = MIMEMultipart()
    # 附件加入到邮件中
    msg.attach(att(file1, reportAttachment))
    msg.attach(att(file2, AlllogAttachment))
    msg.attach(att(file3, ErrorlogAttachment))
    msg['Subject'] = Header(subject, 'utf-8')
    msg['From'] = mailuser  # 之前没有写From和To,发送邮件出现554错误
    msg['To'] = receiver
    msg.attach(MIMEText(mailbody, 'plain', 'utf-8'))
    try:
        # 连接发送邮件
        smtp = smtplib.SMTP()
        smtp.connect(smtpserver)
        smtp.login(mailuser, password)
        smtp.sendmail(mailuser, receiver, msg.as_string())
        smtp.quit()
    except Exception:
        logs.smtplib('Error')
    else:
        logs.smtplib('Success')

run.py

import unittest
from time import strftime
from logs.logs import DelAllLog
from smtp.SendMail import sendMail
from BeautifulReport import BeautifulReport

# 删除所有旧日志
DelAllLog()
# 格式化获取时间 以时间戳命名时不能用冒号（:）
ctime = strftime('%Y-%m-%d %H-%M-%S')
report_name = ctime + '-report.html'
test_suite = unittest.defaultTestLoader.discover('./case', pattern='Test*.py')
# 调用测试报告框架
runner = BeautifulReport(test_suite)
# 定义测试报告相关的参数
runner.report(filename=report_name, description='琴行端自动化测试报告', log_path='./report')
# 发送邮件
sendMail()

# 程序的执行入口
if __name__ == "__main__":
    unittest.main()

整个脚本执行完成后，打开接收邮箱，即可看到最新测试执行的测试报告，如图：

在这里插入图片描述