Selenium实战（七）——自动发送邮件

SMPT（Simple Mail Transfer Protocol）简单邮件传输协议，是一组由源地址到目的地址传送邮件的规则，可以控制信件的中转方式。Python的smptlib模块提供了简单的API用来实现发送邮件的功能，它对SMPT进行了简单的封装。

一、python自带的发送邮件功能

1、发送邮件正文

 import smtplib

 from email.mime.text import MIMEText

 from email.header import Header

 # 发送邮件主题

 subject = 'hello my dear;'

 # 编写HTML类型的邮件正文

 msg = MIMEText('<html><h1>helloMMAMAMAAMAM</h1></html>', 'html', 'utf-8')

 msg['Subject'] = Header(subject, 'utf-8')

 msg['from'] = 'zudangli@126.com'

 msg['to'] = 'zudangli@126.com'

 # 发送邮件

 smtp = smtplib.SMTP()

 smtp.connect("smtp.126.com")

 smtp.login("zudangli@126.com", "19970507zudangli")

 smtp.sendmail("zudangli@126.com", "zudangli@126.com", msg.as_string())

 smtp.quit()

，

email模块下面的MIMEText类，定义发送邮件的正文、格式，以及编码，Header类定义邮件的主题和编码类型

smptlib模块用于发送邮件的。connect()方法指定连接的邮箱服务；login()方法指定登录邮箱的账号和密码；sendmail()方法指定发件人、收件人，以及邮件的正文；quit()方法用于关闭邮件服务器的连接。

2、发送带附件的邮件

 import smtplib

 from email.mime.text import MIMEText

 from email.mime.multipart import MIMEMultipart

 # 邮件主题

 subject = 'python带附件的发送邮件'

 # 发送的附件

 with open('test.txt', 'rb') as f:

     send_att = f.read()

 att = MIMEText(send_att, 'text', 'utf-8')

 att["Content-Type"] = 'application/octet-stream'

 att["Content-Disposition"] = 'attachment; filename="I am attachment.txt"'

 msg = MIMEMultipart()

 msg['Subject'] = subject

 msg['from'] = 'zudangli@126.com'

 msg['to'] = 'zudangli@126.com'

 msg.attach(att)

 # 发送邮件

 smtp = smtplib.SMTP()

 smtp.connect("smtp.126.com")

 smtp.login("zudangli@126.com", "19970507zudangli")

 smtp.sendmail("zudangli@126.com", "zudangli@126.com", msg.as_string())

 smtp.quit()

　　首先，读取附件的内容。通过MIMEText类，定义发送邮件的正文、格式，以及编码；

Content-Type指定附件内容类型；
application/octet-stream表示二进制流；
Content-Disposition指定显示附件的文件

　　然后，使用MIMEMultipart类定义邮件的主题，attach()指定附件信息。

　　最后，通过smtplib模块发送邮件。

二、用yagmail发送邮件

　　yagmail是Python的一个第三方库，GitHub项目地址：https://github.com/kootenpv/yagmail

安装命令：pip install yagmail

 import yagmail

 # 连接邮箱服务器

 yag = yagmail.SMTP(user="zudangli@126.com", password="19970507zudangli", host="smtp.126.com")

 # 邮件正文

 contents = ['This is the body,and here is just text http://somedomain/image.png', 'You can find an audio file attached.']

 # 发送邮件

 yag.send('zudangli@126.com', 'subject', contents)

如果想给多个用户发送邮件，只需要把收件人放到一个list中即可

yag.send(['pegawayatstudying@126.com', '1271916637@qq.com', 'zudangli@126.com'], 'subject', contents)

如果想发送带附件的邮件，只需要指定本地附件的路径即可。

 yag.send(['pegawayatstudying@126.com', '1271916637@qq.com', 'zudangli@126.com'], 'subject', contents, ["C://Users//zudl//Pictures//timg.jpg","F://秒通OApython//formal.py"])

三、整合自动发送邮件

 import time

 import unittest

 import yagmail

 from HTMLTestRunner import HTMLTestRunner

 # 把测试报告作为附件发送到指定邮箱

 def send_mail(report):

     yag = yagmail.SMTP(user="zudangli@126.com",

                        password="19970507zudangli",

                        host='smtp.126.com')

     subject = "主题，自动化测试报告"

     contents = "正文，请查看附件"

     yag.send(['1271916637@qq.com', 'pegawayatstudying@126.com', 'zudangli@126.com'], subject, contents, report)

     print('email has send out !')

 if __name__ == '__main__':

     # 定义测试用例的目录为当前目录中的unit_test

     test_dir = './'

     suits = unittest.defaultTestLoader.discover(test_dir, pattern='test*.py')

     # 取当前日期时间

     now_time = time.strftime("%Y-%m-%d %H_%M_%S")

     html_report = './' + now_time + 'result.html'

     # 生成HTML格式的测试报告

     fp = open('./' + now_time + 'result.html', 'wb')

     runner = HTMLTestRunner(stream=fp,

                             title="百度搜索测试报告",

                             description="运行环境：Windows 10，Chrome浏览器")

     runner.run(suits)

     fp.close()

     send_mail(html_report)

整个程序的执行过程分为两部分：

（1）定义测试报告文件，并赋值给变量html_report，通过HTMLTestRunner运行测试用例，将结果写入文件后关闭。

（2）调用send_mail()函数，并传入html_report文件。在send_mail()函数中，把测试报告作为邮件的附件发送到指定邮箱。

AS a side Note: 不直接把测试报告的内容读取出来作为邮件的正文发送的原因：HTMLTeatRunner报告在展示时引用了Bootstrap样式库，当作为邮件正文“写死”在邮件中时，会导致样式丢失。