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

Question

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

pytest 4.6.3
蟒蛇 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

Answer 2

不，据我所知，pytest 无法为您执行此操作。我能想到的是引入您自己的环境变量并相应地更改日志级别。像这样的东西：

import os
import logging


logger = logging.getLogger('mylogger')

if os.environ.get('mylogger_level'):
    logger.setLevel(os.environ.get('mylogger_level'))