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

时间:2014-02-28 20:46:42

标签: python python-2.7 logging

已经有一个问题可以解答如何对此处的sys.stdoutsys.stderr进行解答:https://stackoverflow.com/a/14197079/198348

但这无处不在。日志记录模块似乎输出到sys.stdoutsys.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

2 个答案:

答案 0 :(得分: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

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

答案 1 :(得分:0)

重定向不起作用,因为您最初设置的记录器有一个直接指向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()
  

输出:

[2018-06-01 16:27:10,670] {logging_test.py:12} INFO - Message before redirecting stdout and stderr
[2018-06-01 16:27:10,670] {logging_test.py:14} INFO - Before Redirection. logger writing to <open file '<stdout>', mode 'w' at 0x10cd74150> 
[2018-06-01 16:27:10,670] {logging_test.py:20} INFO - After Redirection. logger writing to <open file '<stdout>', mode 'w' at 0x10cd74150> 
[2018-06-01 16:27:10,670] {logging_test.py:21} INFO - Message after redirecting stdout and stderr

正如您在输出的第二行和第三行看到的那样,记录器仍然直接引用。

解决此问题的方法之一是做这样的事情

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

请注意,如果您使用多个线程,则此类更改将导致其他线程的输出也开始重定向输出。