上一篇文章寫了給HTMLTestRunner增加餅狀圖，這讓我們的報告看起來更直觀一些，但是細節方面還不是太讓人滿意，比如，我們檢視執行通過、失敗、錯誤的用例時，HTMLTestRunner給出了一大段程式碼，讓人看著頭大，如果給出截圖是否更直觀，一目瞭然的瞭解用例執行情況呢，答案是肯定的。

　　給HTMLTestRunner增加截圖分兩部分，第一部分，需要編寫截圖功能，第二部分就是修改HTMLTestRunner給HTML樣式增加截圖路徑以及展示樣子

----------------------------------------------------------------------------------------------------------------------------------------------------------------------

效果展示：

第一步：增加截圖功能

selenium提供了截圖函式get_srceenshot_as_file()，我們只要呼叫就好了，需要我們自己寫的是，截圖儲存位置和截圖名稱，根據個人喜好來定。

下面程式碼，以及把截圖功能封裝在一個函式中，我們只要傳入driver、檔名稱、路徑就好，路徑是寫死的

# !/usr/bin/env python
# -*- coding: UTF-8 –*-
__author__ = 'Mr.Li'
from selenium import webdriver
import os

#截圖函式
def insert_img(driver,file_name，file_path):
    base_dir 
 
= os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
    file_path = base_dir + "/YJF-UI_TestCode/report/image/" + file_name
    driver.get_screenshot_as_file(file_path)
    print('Screenshot_Path：',file_path)

if __name__ == '__main__':
    driver = webdriver.Chrome()
    driver.get('http://www.baidu.com
 
')
    insert_img(driver,'baidu.png')
    driver.quit()