已经有一个问题可以解答如何对此处的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内运行的测试。
现在不能重写冗长的部分,但绝对是未来的目标。
这是我尝试使用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
答案 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')
请注意,如果您使用多个线程,则此类更改将导致其他线程的输出也开始重定向输出。