如何将Python子库的控制台输出重定向至logging模块？

遇到子库直接输出到控制台（而非通过logging模块）的内容无法被自定义logger捕获的问题，可以通过以下几种方式解决：

方法1：全局重定向标准输出/标准错误到logging流 #

子库的这类输出一般是通过print()或直接写入sys.stdout/sys.stderr实现的，我们可以替换这两个流，把内容转发给logger。

实现代码如下：

import sys
import logging

# 先配置好你的logger
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s [%(levelname)s] - %(message)s',
    handlers=[
        logging.FileHandler('app.log'),  # 输出到文件
        logging.StreamHandler()          # 输出到控制台
    ]
)

class StreamToLogger:
    def __init__(self, logger, log_level=logging.INFO):
        self.logger = logger
        self.log_level = log_level
        self.linebuf = ''

    def write(self, buf):
        # 按行拆分输出内容，避免多行内容被合并成一条日志
        for line in buf.rstrip().splitlines():
            self.logger.log(self.log_level, line.rstrip())

    def flush(self):
        # 实现flush方法避免报错
        pass

# 重定向stdout和stderr到logger
sys.stdout = StreamToLogger(logging.getLogger(), logging.INFO)
sys.stderr = StreamToLogger(logging.getLogger(), logging.ERROR)

# 调用子库的代码放在这里，子库的print输出会被logger捕获

方法2：临时重定向（避免影响全局输出） #

如果不想全局替换stdout/stderr，可以只在调用子库的代码块内临时重定向：

import sys
import logging

# 配置logger同上

def call_sub_library():
    # 保存原始流
    original_stdout = sys.stdout
    original_stderr = sys.stderr
    try:
        # 临时重定向
        sys.stdout = StreamToLogger(logging.getLogger(), logging.INFO)
        sys.stderr = StreamToLogger(logging.getLogger(), logging.ERROR)
        # 这里执行子库调用
        # sub_library.some_function()
        print("blablabla")  # 模拟子库输出
    finally:
        # 恢复原始流
        sys.stdout = original_stdout
        sys.stderr = original_stderr

特殊情况处理 #

如果子库是通过C扩展直接输出到控制台（绕过了Python的sys.stdout机制），上述Python层面的重定向会失效。这种情况下只能通过操作系统层面的重定向，比如在终端执行脚本时：

python your_script.py > app.log 2>&1

但这种方式会把所有输出（包括logger的格式化日志）都写入文件，无法区分格式，仅作为最后的备选方案。

内容的提问来源于stack exchange，提问作者likewest