allure-pytest的報錯截圖添加到報告

# 寫在conftest.py

import os
import allure
from selenium import webdriver
import pytest
# 添加報錯截圖到allure報告裡
driver = None


@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
    '''
    hook pytest失敗
    :param item:
    :param call:
    :return:
    '''
    # execute all other hooks to obtain the report object
    outcome = yield
    rep = outcome.get_result()
    # we only look at actual failing test calls, not 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")
        # pic_info = adb_screen_shot()
        with allure.step('添加失敗截圖...'):
            allure.attach(driver.get_screenshot_as_png(), "失敗截圖", allure.attachment_type.PNG)


#def adb_screen_shot():
    # driver.get_screenshot_as_png()
    # driver.get_screenshot_as_base64()
    # driver.get_screenshot_as_file("122.jpg")
    # os.popen("adb screen -p testfailue.jpg")


@pytest.fixture(scope='session', autouse=True)
def browser():
    global driver
    if driver is None:
        driver = webdriver.Chrome()
    return driver