接上篇 python smtplib模块自动收发邮件功能（一），用python smtplib模块实现了发送邮件程序了，那么接下来我们需要现在要解决的问题如何在 test_report\目录下找到最新生成的报告，只有找到了才能把发邮件功能,然后将其集成到我们的自动化测试应用中.
一、获取最新的test_report
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
#coding=utf-8import smtplibfrom email.mime.text import mimetextfrom email.header import headerimport os,datetime,timeresult_dir='c:\\python34\\test_report' # test_report的绝对路径 lists=os.listdir(result_dir)print (lists)lists.sort(key=lambda fn: os.path.getmtime(result_dir+"\\"+fn) if not os.path.isdir(result_dir+"\\"+fn) else 0)print ('最新的文件为:'+lists[-1])file=os.path.join(result_dir,lists[-1])print (file) |
f5,运行，得到：
那么 c:\python34\test_report\2016-03-24-16_00_34_result.html是最新的test_report
二、整合自动发送邮件功能
主要实现以下几部分：
1.运行相关的 cases 生成htmltest report。
2.将test report发送到指定邮箱。
直接上脚本：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import unittestimport htmltestrunnerimport osimport timeimport datetimeimport smtplibfrom email.mime.text import mimetextfrom email.mime.multipart import mimemultipartfrom email.mime.image import mimeimagefrom email.header import header#定义发送邮件def sentemail(file_new): #发信邮箱 sender='abc@ciexxx.com' #收信邮箱 receiver='12345@qq.com' #定义正文 f=open(file_new,'rb') mail_body=f.read() f.close() msg=mimetext(mail_body,_subtype='html',_charset='utf-8') #定义标题 msg['subject']=u"搜狗搜索测试报告" msg['date']=time.strftime('%a, %d %b %y %h:%m:%s %z') smtp=smtplib.smtp() #smtpserver='smtp.263xmail.com' smtp.connect('smtp.263xmail.com') username='abc@ciexxx.com' password='123456' smtp.login(username,password) smtp.sendmail(sender,receiver,msg.as_string()) smtp.quit() print ('email has been sent out!')#查找测试报告，调用收发邮件功能def sendreport(): result_dir='c:\\python34\\test_report' lists=os.listdir(result_dir) lists.sort(key=lambda fn: os.path.getmtime(result_dir+"\\"+fn) if not os.path.isdir(result_dir+"\\"+fn) else 0) #print (u'最新测试生成的报告:'+lists[-1]) print (u'最新测试生成的报告:'+lists[-2]) #找到最新生成的文件 #file_new=os.path.join(result_dir,lists[-1]) file_new=os.path.join(result_dir,lists[-2]) print (file_new) sentemail(file_new)listaa='c:\\python34\\test_case'def createsuitel(): testunit=unittest.testsuite() '''discover方法定义''' discover=unittest.defaulttestloader.discover(listaa, pattern='unittesthtml_*.py', top_level_dir=none) for test_suite in discover: for test_case in test_suite: testunit.addtests(test_case) print (testunit) return testunitalltestnames=createsuitel()now = time.strftime('%y-%m-%d-%h_%m_%s',time.localtime(time.time()))file_name='c:\\python34\\test_report\\'+now+'_result.html'fp = open(file_name, 'wb')runner =htmltestrunner.htmltestrunner(stream=fp,title=u'搜狗搜索测试报告',description=u'用例执行情况：')if __name__=="__main__": runner.run(alltestnames) time.sleep(2) sendreport() fp.close() |
f5运行，得到：
查看邮箱，如图所示：
打开邮件内容，如图所示：
ok，就这样实现了实际项目中的自动收发邮件功能。
另，几个知识点：
1. os.listdir()
用于获取目录下的所有文件列表
2. lists.sort()
python 列表有一个内置的列表。sort()方法用于改变列表中元素的位置。
3. key=lambda fn:
key 是带一个参数的函数，用来为每个元素提取比较值.默认为 none, 即直接比较每个元素.
4.os.path.isdir()
isdir()函数判断某一路径是否为目录
5.lists[-1]
-1 表示取文件列表中的最大值，也就是最新被创建的文件.
6.os.path.join()
join()方法用来连接字符串，通过路径与文件名的拼接，我们将得到目录下最新被创建的的文件名的完整路径。
7.sentmail(file_new)
定义一个 sentmail()发邮件函数，接收一个参数 file_new，表示接收最新生成的测试报告文件.
8.open(file_new, ‘rb')
以读写（rb）方式打开最新生成的测试报告文件.
9.sendreport()
定义 sendreport()用于找最新生成的测试报告文件 file_new.
在成功实现这个sample之前，遇到过1个问题：
指定的邮箱可以正常收到邮件，但所得到的邮件内容是空的，这是由于 htmltestrunner 报告文件的机制所引起的。在测试用例运行之前生成报告文件，在整个程序没有彻底运行结束前，程序并没有把运行的结果写入到文件中，所以，在用例运行完成后发邮件，造成邮件内容是空的。
最开始的脚本，其中两行是这样的：
|
1
2
3
|
print (u'最新测试生成的报告:'+lists[-1])#找到最新生成的文件file_new=os.path.join(result_dir,lists[-1]) |
于是，运行结束后，出现了问题，指定的邮箱可以正常收到邮件，但所得到的邮件内容是空的。也就是说，脚本运行还没有结束，就已经执行了邮件的自动发送功能。
于是，将上述的两行，改后的脚本：
|
1
2
3
|
print (u'最新测试生成的报告:'+lists[-2]) #找到最新生成的文件file_new=os.path.join(result_dir,lists[-2]) |
所以，我们不能在整个程序未运行结束时发送当前的测试报告，我们可以选择上一次运行结果的报告进行发送。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/liujingqiu/article/details/50973848
