python - 在 pytest 中禁用特定记录器

Question

在我的项目中，我使用 pdfplumber 解析一些 PDF 文件。在测试执行(pytest)期间，我有时希望查看代码中的日志以进行调试。这可以通过设置 --log-cli-level=DEBUG 来完成。然而，这会打开来自所有代码的消息，还有 pdfplumber - 这非常冗长并且使调试变得困难。有没有办法在测试运行期间有选择地启用/禁用记录器？

pytest 4.6.3
python 3.7.3

感谢您的帮助!

Answer 1

Pytest 默认情况下不支持此功能，但您可以向 conftest.py 添加自定义选项以关闭特定记录器。

import pytest
import logging

def pytest_addoption(parser):
    """Add a command line option to disable logger."""
    parser.addoption(
        "--log-disable", action="append", default=[], help="disable specific loggers"
    )

def pytest_configure(config):
    """Disable the loggers."""
    for name in config.getoption("--log-disable", default=[]):
        logger = logging.getLogger(name)
        logger.propagate = False