用例失败自动抓图并集成到allure显示的代码

第一种方式

import pytest
from selenium import webdriver
import os
import allure
_driver = None


@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
    '''
    获取每个用例状态的钩子函数
    :param item:
    :param call:
    :return:
    '''
    # 获取钩子方法的调用结果
    outcome = yield
    rep = outcome.get_result()
    # 仅仅获取用例call 执行结果是失败的情况, 不包含 setup/teardown
    if rep.when == "call" and rep.failed:
        mode = "a" if os.path.exists("failures") else "w"
        with open("failures", mode) as f:
            # let's also access a fixture for the fun of it
            if "tmpdir" in item.fixturenames:
                extra = " (%s)" % item.funcargs["tmpdir"]
            else:
                extra = ""
            f.write(rep.nodeid + extra + "\n")
        # 添加allure报告截图
        if hasattr(_driver, "get_screenshot_as_png"):
            with allure.step('添加失败截图...'):
                allure.attach(_driver.get_screenshot_as_png(), "失败截图", allure.attachment_type.PNG)

第二种方式

#1、定义装饰2层函数
def screenshot_allure(func):
    def get_err_screenshot(self,*args,**kwargs):
#2、定义内部函数，拍图操作
        try:
            func(self,*args,**kwargs)
        except Exception as e:
            png = self.driver.get_screenshot_as_png()
            name = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
            allure.attach(png, name, allure.attachment_type.PNG)
            raise e
#3、返回内部函数名称
    return get_err_screenshot

 @screenshot_allure
    def step(self,data,run_case):
        tc_id = run_case[TestSteps.STEP_TC_ID]
        #获取步骤
        steps = data.get_steps_by_tc_id(tc_id)
        #allure报告
        #feature
        allure.dynamic.feature(run_case[TestCases.CASES_NOTE])
        #story
        allure.dynamic.story(run_case[TestCases.CASES_DESC])
        #title
        allure.dynamic.title(run_case[CaseData.DATA_CASE_ID]+"-"+run_case[CaseData.DATA_NAME])

        for step in steps:
            log.debug("执行步骤{}".format(step))
            #获取元素信息
            elements = step[TestSteps.STEP_ELEMENT_NAME]
            element = data.get_elements_by_element(step[TestSteps.STEP_TC_ID], elements)
            log.debug("元素信息{}".format(element))
            #操作步骤 关键表映射 click_btn
            operate = self.get_keyword(step[TestSteps.STEP_OPERATE])

            #操作判断，是否存在，不存在不执行步骤
            if operate:
                # 定义方法参数：字典
                param_value = dict()

                #根据getattr判断执行哪个方法
                action_method = getattr(Action(self.driver),operate)
                log.debug("该关键字是{}".format(operate))

                #定义具体的参数
                by = element[Elements.ELE_BY]
                value = element[Elements.ELE_VALUE]
                # 1、获取by,value,send_value内容
                send_value = step[TestSteps.STEP_DATA]
                # 2、send_value内容转换，通过case data数据内容

                expect = run_case[CaseData.DATA_EXPECT_RESULT]
                param_value["by"] = by
                param_value["value"] = value

                param_value["expect"] = expect
                #判断假如有输入内容 字符转换
                if send_value:
                    data_input = run_case[CaseData.DATA_INPUT]
                    send = self.str_to_dict(data_input)
                    param_value["send"] = send[send_value]
                #step
                with allure.step(step[TestSteps.STEP_NAME]):
                    action_method(**param_value)

            else:
                log.error("没有operate信息：{}".format(operate))