python - 将网页截图附加到所有“then”的诱惑步骤（pytest-bdd、selenium、allure）

Question

我正在编写 pytest-bdd + selenium + allure 测试。我需要将屏幕截图附加到每个验证步骤 - @then("...")。Pytest bdd 有我这样实现的“步骤后”钩子：

def pytest_bdd_after_step(request, feature, scenario, step, step_func, step_func_args):
if step.keyword == 'Then':
    driver = request.getfixturevalue('driver')
    allure.attach(driver.get_screenshot_as_png(), name='screenshot',
        attachment_type=allure.attachment_type.PNG)

它工作正常，但屏幕截图出现在报告中的所有步骤之后，而不是步骤内：

所以我的问题是：是否可以在“之前/之后”挂钩中将某些东西附加到步骤本身？谢谢。