Python3 测试报告BeautifulReport中添加截图

在测试类中，添加save_img方法，在测试过程中出现错误时，自动截图并返回失败

默认存放的图片路径是img

def save_img(self, img_name):
    """
        传入一个img_name, 并存储到默认的文件路径下
    :param img_name:
    :return:
    """
    self.driver.get_screenshot_as_file('{}/{}.png'.format(os.path.abspath(self.img_path), img_name))

调用时，使用如下：

@BeautifulReport.add_test_img('test_errors_save_imgs')
def test_errors_save_imgs(self):
    """
        如果在测试过程中, 出现不确定的错误, 程序会自动截图, 并返回失败, 如果你需要程序自动截图, 则需要咋测试类中定义 save_img方法
    """
    self.driver.find_element_by_xpath('//abc')

效果：