我有一个 Python 脚本，它使用“打印”来打印到标准输出。我最近通过 Python Logger 添加了日志记录，并希望这样做，如果启用了日志记录，这些打印语句将转到记录器。我不想修改或删除这些打印语句。
我可以通过 'log.info("some info msg")' 来记录。我希望能够做这样的事情：
if logging_enabled:
sys.stdout=log.info
print("test")
如果启用了日志记录，“test”应该像我做 log.info("test") 一样被记录。如果未启用日志记录，则应将“test”打印到屏幕上。
这可能吗？我知道我可以以类似的方式将标准输出定向到文件（请参阅：redirect prints to log file）
【问题讨论】：
你有两个选择：
打开一个日志文件并用它替换 sys.stdout，而不是一个函数：
log = open("myprog.log", "a")
sys.stdout = log
>>> print("Hello")
>>> # nothing is printed because it goes to the log file instead.
用你的日志函数替换打印：
# If you're using python 2.x, uncomment the next line
#from __future__ import print_function
print = log.info
>>> print("Hello!")
>>> # nothing is printed because log.info is called instead of print
【讨论】：
当然，您既可以打印到标准输出，也可以附加到日志文件，如下所示：
# Uncomment the line below for python 2.x
#from __future__ import print_function
import logging
logging.basicConfig(level=logging.INFO, format='%(message)s')
logger = logging.getLogger()
logger.addHandler(logging.FileHandler('test.log', 'a'))
print = logger.info
print('yo!')
【讨论】：
另一种方法是将记录器包装在一个对象中，该对象将对write 的调用转换为记录器的log 方法。
Ferry Boender 就是这样做的，provided under the GPL license 在a post 上his website。下面的代码就是基于此，但解决了原来的两个问题：
import logging
import sys
class StreamToLogger(object):
"""
Fake file-like stream object that redirects writes to a logger instance.
"""
def __init__(self, logger, log_level=logging.INFO):
self.logger = logger
self.log_level = log_level
self.linebuf = ''
def write(self, buf):
temp_linebuf = self.linebuf + buf
self.linebuf = ''
for line in temp_linebuf.splitlines(True):
# From the io.TextIOWrapper docs:
# On output, if newline is None, any '\n' characters written
# are translated to the system default line separator.
# By default sys.stdout.write() expects '\n' newlines and then
# translates them so this is still cross platform.
if line[-1] == '\n':
self.logger.log(self.log_level, line.rstrip())
else:
self.linebuf += line
def flush(self):
if self.linebuf != '':
self.logger.log(self.log_level, self.linebuf.rstrip())
self.linebuf = ''
logging.basicConfig(
level=logging.DEBUG,
format='%(asctime)s:%(levelname)s:%(name)s:%(message)s',
filename="out.log",
filemode='a'
)
stdout_logger = logging.getLogger('STDOUT')
sl = StreamToLogger(stdout_logger, logging.INFO)
sys.stdout = sl
stderr_logger = logging.getLogger('STDERR')
sl = StreamToLogger(stderr_logger, logging.ERROR)
sys.stderr = sl
这使您可以轻松地将所有输出路由到您选择的记录器。如果需要，您可以将 sys.stdout 和/或 sys.stderr 保存在此线程中，然后再替换它，如果您以后需要恢复它。
【讨论】：
stdout 的模块生成stdout 的模块中？或者，只要我在__main__ 函数中按上述方式配置，发送到stdout 的任何输出都会发送到file？
self.linebuf = '' 的用途。我看不到任何地方都在使用它。
一个更简单的选择，
import logging, sys
logging.basicConfig(filename='path/to/logfile', level=logging.DEBUG)
logger = logging.getLogger()
sys.stderr.write = logger.error
sys.stdout.write = logger.info
【讨论】：
sys.__stderr__ 用于StreamHandler，这个永远不会改变。
sys.stdout 的引用。
您确实应该这样做：通过调整您的日志记录配置以使用 print 语句或其他内容，具体取决于设置。不要覆盖print 行为，因为将来可能引入的某些设置（例如，由您或其他人使用您的模块）可能实际上会将其输出到stdout，您会遇到问题。
有一个处理程序应该将您的日志消息重定向到正确的流（文件，stdout 或任何其他类似文件）。它被称为StreamHandler，并与logging 模块捆绑在一起。
所以基本上在我看来你应该做，你说你不想做的事情：用实际的日志记录替换 print 语句。
【讨论】：
stdout 的第三方库。我在我的应用程序中使用logging，所以我的应用程序本身没有任何打印语句。您提到StreamHandler 可以将日志消息发送到file 或stdout。 StreamHandler 是否将来自 stdout 的输出发送到文件？
一旦你定义了你的记录器，即使有多个打印参数，也可以使用它来将打印重定向到记录器。
print = lambda *tup : logger.info(str(" ".join([str(x) for x in tup])))
【讨论】：
下面的片段在我的 PySpark 代码中完美运行。如果有人需要以防万一理解-->
import os
import sys
import logging
import logging.handlers
log = logging.getLogger(__name_)
handler = logging.FileHandler("spam.log")
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
handler.setFormatter(formatter)
log.addHandler(handler)
sys.stderr.write = log.error
sys.stdout.write = log.info
（将在同一目录中的“spam.log”中记录每个错误，控制台/stdout 上不会有任何内容）
（将“spam.log”中的所有信息记录在同一目录中，控制台/stdout上不会有任何信息）
在两个文件以及控制台中打印输出错误/信息删除上面的两行。
快乐编码 干杯！！！
【讨论】：