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

Question

关于sys.stdout和sys.stderr这里已经有一个问题可以回答如何做到这一点：https://stackoverflow.com/a/14197079/198348

但这并不适用于任何地方。日志模块似乎输出到sys.stdout和sys.stderr，但我无法用上面的上下文管理器捕获它。在

在下面的示例代码中，我尝试捕获上下文管理器中的所有输出，但未能为logger语句执行此操作：

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

以下是我试着用鼻子测试的方法：

^{pr2}$

然后运行上面的：

nosetests test_logging.py
nosetests --nocapture test_logging.py

Answer 1

重定向不起作用，因为您最初设置的记录器有一个直接指向stdout的指针。见下文。在

logging_test.py

import logging
import sys

SIMPLE_LOG_FORMAT = '[%(asctime)s] {%(filename)s:%(lineno)d} %(levelname)s - %(message)s'
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
formatter = logging.Formatter(SIMPLE_LOG_FORMAT)
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)

logger.info("Message before redirecting stdout and stderr")
# Checking what logger is writing to
logger.info('Before Redirection. logger writing to {} '.format(logger.handlers[0].stream))

log_file = open('log_file.log', 'w')
sys.stdout = log_file
sys.stderr = log_file
# Checking what logger is writing to
logger.info('After Redirection. logger writing to {} '.format(logger.handlers[0].stream))
logger.info("Message after redirecting stdout and stderr")

log_file.close()

Output:

^{pr2}$

正如您在输出的第二行和第三行上看到的，logger仍然直接引用它。在

解决这个问题的方法之一就是这样做

logger.handlers[0].stream = open('log_file.log', 'w')

请注意，如果您使用多个线程，在一个线程中这样的更改将导致其他线程的输出也开始重定向输出。在

Answer 2

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：

^{pr2}$

那么，上下文管理器可能看起来有点像：

@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')