如何临时重定向Python中的日志记录输出？

2019年8月4日 186次阅读

已经有一个问题可以解答如何在sys.stdout和sys.stderr中执行此操作：
https://stackoverflow.com/a/14197079/198348

但这并不适用于所有地方.日志记录模块似乎输出到sys.stdout和sys.stderr,但我无法使用上面的上下文管理器捕获它.

在下面的示例代码中,我试图捕获上下文管理器中的所有输出,但是没有为记录器语句执行此操作：

from __future__ import print_function
import contextlib
import sys
import logging
from StringIO import StringIO

# taken from https://stackoverflow.com/a/14197079/198348
@contextlib.contextmanager
def stdout_redirect(where):
    prev_stdout = sys.stdout
    prev_stderr = sys.stderr
    prev_stdout.flush()
    sys.stdout = where
    sys.stderr = where
    try:
        yield where
    finally:
        where.flush()
        sys.stdout = prev_stdout
        sys.stderr = prev_stderr

logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger()

print("\t\tOUTSIDE: stdout", file=sys.stdout)
print("\t\tOUTSIDE: stderr", file=sys.stderr)
logger.info("\tOUTSIDE: info")
logger.debug("\tOUTSIDE: debug")
logger.warn("\tOUTSIDE: warn")
logger.error("\tOUTSIDE: error")
logger.critical("\tOUTSIDE: critical")

print("=============== DIVIDER ================")

s = ""
with stdout_redirect(StringIO()) as new_stdout:
    print("\t\tINSIDE: stdout", file=sys.stdout)
    print("\t\tINSIDE: stderr", file=sys.stderr)
    logger.info("\tINSIDE: info")
    logger.debug("\tINSIDE: debug")
    logger.warn("\tINSIDE: warn")
    logger.error("\tINSIDE: error")
    logger.critical("\tINSIDE: critical")


print("=============== DIVIDER ===============")
print(new_stdout.getvalue())

print("=============== LOGGING ===============")

print(logger.handlers)
print(logger.root.handlers)

如何暂时将吐出的记录器的输出重定向到stdout并捕获它们？我看了logging/init.py,但它没有立即告诉我我需要做什么.

我这样做的动机是,我想为一个狡猾的大代码库配备测试,每个测试都会捕获每个测试调用的虚假记录输出量.我可以捕获外部程序,但我似乎无法捕获我在nose中运行的测试.

重写冗长的部分现在不是一个选项,但绝对是未来的目标.

编辑,关于ubuntu

这是我尝试使用nosetests运行的：

from __future__ import print_function
import sys

def test_funky_shurane():
    import logging
    logging.basicConfig(level=logging.DEBUG)
    logging.info("===== shurane info")
    logging.warn("===== shurane warn")
    logging.error("===== shurane error")
    logging.critical("===== shurane critical")
    print("===== shurane stdout", file=sys.stdout)
    print("===== shurane stderr", file=sys.stderr)
    assert True

然后运行上面的：

nosetests test_logging.py
nosetests --nocapture test_logging.py

最佳答案 logging.basicConfig()是一种方便,以非常简单的方式设置一些记录器处理.如果你需要更多,你不应该使用basicConfig().这不是什么大问题,因为它并没有做很多事情.我们需要的是配置两个流的日志记录;

import logging, sys
fmt = logging.Formatter(BASIC_FORMAT)

hdlr_stderr = logging.StreamHandler(sys.stderr)
hdlr_stderr.setFormatter(fmt)
hdlr_stdout = logging.StreamHandler(sys.stdout)
hdlr_stdout.setFormatter(fmt)
root.addHandler(hdlr_stderr)
root.addHandler(hdlr_stdout)
root.setLevel(logging.DEBUG)

默认情况下,记录器记录他们收到的所有消息;但最初,我们不想将任何消息记录到sys.stdout：

hdlr_stdout.level = float('inf')  # larger than any log level; nothing gets logged

然后,您的上下文管理器可能看起来像：

@contextlib.contextmanager
def redirect_stderr_logging(where):
    hdlr_stderr.level = float('inf')
    hdlr_stdout.level = logging.NOTSET
    try:
        yield where
    finally:
        hdlr_stderr.level = logging.NOTSET
        hdlr_stdout.level = float('inf')
点赞