首页 > 编程 > Python > 正文

python smtplib模块自动收发邮件功能(二)

2020-01-04 15:01:09
字体:
来源:转载
供稿:网友

接上篇python smtplib模块自动收发邮件功能(一) ,用python smtplib模块实现了发送邮件程序了,那么接下来我们需要现在要解决的问题如何在 test_report/目录下找到最新生成的报告,只有找到了才能把发邮件功能,然后将其集成到我们的自动化测试应用中. 

一、获取最新的test_report

#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,运行,得到:

python,smtplib,收发邮件

那么 C:/Python34/test_report/2016-03-24-16_00_34_result.html是最新的test_report

二、整合自动发送邮件功能

主要实现以下几部分:

1.运行相关的 cases 生成HTMLtest report。 
2.将test report发送到指定邮箱。

直接上脚本

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运行,得到: 

python,smtplib,收发邮件

查看邮箱,如图所示:

python,smtplib,收发邮件

打开邮件内容,如图所示:

python,smtplib,收发邮件

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 报告文件的机制所引起的。在测试用例运行之前生成报告文件,在整个程序没有彻底运行结束前,程序并没有把运行的结果写入到文件中,所以,在用例运行完成后发邮件,造成邮件内容是空的。

最开始的脚本,其中两行是这样的:

print (u'最新测试生成的报告:'+lists[-1])#找到最新生成的文件file_new=os.path.join(result_dir,lists[-1])

于是,运行结束后,出现了问题,指定的邮箱可以正常收到邮件,但所得到的邮件内容是空的。也就是说,脚本运行还没有结束,就已经执行了邮件的自动发送功能。

于是,将上述的两行,改后的脚本:

print (u'最新测试生成的报告:'+lists[-2]) #找到最新生成的文件file_new=os.path.join(result_dir,lists[-2])

所以,我们不能在整个程序未运行结束时发送当前的测试报告,我们可以选择上一次运行结果的报告进行发送。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持VEVB武林网。


注:相关教程知识阅读请移步到python教程频道。
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表