#?coding:utf-8
from?selenium?import??webdriver
import?unittest
import?os
from?HTMLTestRunner?import?HTMLTestRunner
from?email.mime.text?import?MIMEText
from?email.mime.multipart?import?MIMEMultipart
from?email.header?import?Header
from?HRXT.uidriver.loggin?import?Log
import?smtplib
import?time
log?=?Log（）
#==============定義發(fā)送郵件==========
def?send_mail（file_new）:
????log.info（“------獲取郵箱賬戶密碼------“）
????_user?=?‘1024547862@qq.com‘??#?發(fā)件地址
????_pwd=“l(fā)fwwfjridtiqbdid“?????#?服務(wù)器授權(quán)碼
????_to=“1620596776@qq.com“

????with?open（file_new?“rb“）?as?f:
????????mail_body?=?f.read（）
????msg?=?MIMEMultipart（）
????body?=?MIMEText（mail_body?‘html‘?‘utf-8‘）
????msg[‘Subject‘]?=?Header（“自動化測試報告“?‘utf-8‘）
????msg[‘From‘]?=?_user
????msg[‘To‘]?=?_to
????msg.attach（body）
????#
????#
????#添加附件
????log.info（“------添加測試報告附件------“）
????att=MIMEText（open（report_file“rb“）.read（）“base64““utf-8“）
????att[“Content-Type“]?=?“application/octet-stream“
????att[“Content-Disposition“]?=?‘a(chǎn)ttachment;?filename=?“testcase.html“‘
????msg.attach（att）

????#
????#
????#發(fā)送郵件
????s?=?smtplib.SMTP_SSL（“smtp.qq.com“）
????#?s.set_debuglevel（1）
????s.login（_user_pwd）??#?登錄郵箱的賬戶和密碼
????s.sendmail（_user_to?msg.as_string（））

????s.quit（）

????log.info（“------發(fā)送郵件------“）
????print（‘自動化郵件已發(fā)送‘）

#======查找測試目錄，找到最新生成的測試報告文件======
def?new_report（test_report）:
????lists?=?os.listdir（test_report）????????????????????????????????????#列出目錄的下所有文件和文件夾保存到lists
????lists.sort（key=lambda?fn:os.path.getmtime（test_report?+?“\\“?+?fn））#按時間排序
????file_new?=?os.path.join（test_reportlists[-1]）?????????????????????#獲取最新的文件保存到file_new
????print（file_new）
????return?file_new
if?__name__?==?“__main__“:
????#?unittest.TextTestRunner（）

????test_report=os.path.join（os.getcwd（）“Warehouse\\report“）???#?報告存放路徑

????case_path?=os.path.join（os.getcwd（）“case\\test_V1_0_1“）??#用例路徑

????report_file?=?new_report（test_report）#?獲取最新的測試報告

????discover?=?unittest.defaultTestLoader.discover（case_path
???????????????????????????????????????????????????pattern=“test_UserInfo_1.py“??#?匹配test*.py
???????????????????????????????????????????????????top_level_dir=None）
????print（discover）
????#?return?discover
#======================測試報告============================
????#?html報告文件內(nèi)容
????now=time.strftime（“%Y-%m-%d-%H-%M-%S“）???#加時間戳
????report_abspath?=?os.path.join（test_report?“result“+now+“.html“）
????fp?=?open（report_abspath?“wb“）
????runner?=?HTMLTestRunner（stream=fptitle=‘自動化測試報告測試結(jié)果如下：‘description=‘用例執(zhí)行情況：‘verbosity=2）

????#?調(diào)用函數(shù)返回值
????runner.run（discover）
????fp.close（）
????new_report?=?new_report（test_report）
????send_mail（new_report）??#?發(fā)送測試報告

#通過主函數(shù)執(zhí)行用例
#從注冊到發(fā)布信息
#執(zhí)行完成發(fā)生郵件，生成測試報告