暂时忽略特定的日志记录行

Question

我有一个名为 import_customers() 的方法，它可以加载类似 csv 的数据。

此方法记录到日志级别的 INFO。

在一种情况下，我想避免这种日志记录。

我看到了几种方法：

变体 1：像 do_logging=True 这样的新 kwarg，我可以将其切换为 false。

变体 2：使用一些忽略此行的魔法上下文。

with IgnoreLoggingContext() as context:
    import_customers()

如何实现IgnoreLoggingContext()？

如果您认为 V1 更好，请发表评论。

Answer 1

这取决于您的需要。如果您想禁用整个日志记录，这将是最简单的：

from contextlib import contextmanager
import logging


@contextmanager
def IgnoreLoggingContext():
    root = logging.getLogger()
    origin_level = root.getEffectiveLevel()
    root.setLevel(logging.WARNING) # Or whatever you want
    yield
    root.setLevel(origin_level)

with IgnoreLoggingContext():
    do_something()

或者您可以将变量传递给管理器以指定要禁用的记录器：

@contextmanager
def IgnoreLoggingContext(name):
    logger = logging.getLogger(name)
    origin_level = logger.getEffectiveLevel()
    logger.setLevel(logging.WARNING) # Or whatever you want
    yield
    logger.setLevel(origin_level)

Answer 2

可以通过disabled 属性完全禁用记录器。另一件事是记录器不一定会传播到根记录器并且可以定义自己的记录处理，因此完全禁用可能如下所示：

@contextlib.contextmanager
def off():
    loggers = [logging.getLogger(name) for name in logging.root.manager.loggerDict]
    # turn loggers off
    for logger in loggers:
        logger.disabled = True
    logging.root.disabled = True
    yield
    # turn loggers back on
    for logger in loggers:
        logger.disabled = False
    logging.root.disabled = False

使用示例：

if __name__ == '__main__':
    logging.basicConfig(level=logging.DEBUG)
    logging.info('foo')
    with off():
        logging.info('bar')

将打印

INFO:root:foo

Answer 3

只需将日志记录级别从 INFO 更改为 ERROR，完成后将其更改回来。

# Beginning of file
rootLogger.setLevel(logging.INFO)

# Change logging level to only log to error, just prior to calling your method
rootLogger.setLevel(logging.ERROR)

# Change it back when you're done
rootLogger.setLevel(logging.INFO)