（HTMLTestRunner增加圖表樣式顯示）二：增加截圖功能

　　上一篇文章寫了給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()