生成报错截图

step1：前提：准备好执行的测试用例脚本

step2：创建一个新文件夹，放置screenshot.py文件，文件脚本如下图

step3：在编写的测试用例脚本中，引用创建的文件及py包(from picture.screenshot import takeScreenshot, createDir)

step4：在脚本中将需要截图的脚本用try：...except:循环（意思是：如果try这部分脚本执行成功，将不会有截图，否则执行except截图）

step5：运行用例脚本就可以啦（注意：生成报告是运行报告脚本才能成功）

import os
import traceback
import time

def currentDate():
    '''生成当前日期字符串'''
    date = time.localtime()
    return '-'.join([str(date.tm_year), str(date.tm_mon), str(date.tm_mday)])


def currentTime():
    '''生成当前时间字符串'''
    date = time.localtime()
    return '-'.join([str(date.tm_hour), str(date.tm_min), str(date.tm_sec)])


def createDir():
    '''创建当前日期和当前时间目录'''
    path = os.path.dirname(os.path.abspath(__file__))
    dateDir = os.path.join(path, currentDate())
    # 如果当前日期目录不存的话就创建
    if not os.path.exists(dateDir):
        os.mkdir(dateDir)
    timeDir = os.path.join(dateDir, currentTime())
    # 如果当前时间目录不存的话就创建
    if not os.path.exists(timeDir):
        os.mkdir(timeDir)
    return timeDir


def takeScreenshot(driver, savePath, pictureName):
    picturePath = os.path.join(savePath, pictureName + '.png')
    try:
        driver.get_screenshot_as_file(picturePath)
    except Exception as e:
        print(traceback.print_exc())

try:
　　.....

except:

    takeScreenshot(self.driver, createDir(), '报错截图')