（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()