如何重定向带默认sys.stdout参数的msg函数标准输出至变量？

问题根源 #

外部库中的msg函数在定义时，默认参数file=sys.stdout是在函数定义阶段就绑定了当时的sys.stdout对象。后续你通过contextmanager替换sys.stdout变量时，函数的默认参数依然指向最初的stdout对象，所以重定向不会生效。只有显式传入当前的sys.stdout，才会使用新的流。

可行解决方案 #

1. 包装并替换原函数 #

直接给msg函数做一层包装，调用时自动传入当前的sys.stdout作为file参数，无需手动传参：

import sys
# 导入外部库的原函数
from your_external_lib import msg as original_msg

def wrapped_msg(content, file=None):
    # 如果没传file参数，自动用当前的sys.stdout
    if file is None:
        file = sys.stdout
    original_msg(content, file=file)

# 替换外部库中的msg函数引用
import your_external_lib
your_external_lib.msg = wrapped_msg

之后不管是你自己调用msg，还是库内部调用，都会自动使用当前的sys.stdout，配合contextlib.redirect_stdout就能正常捕获输出。

2. 修改函数的默认参数 #

直接修改原函数的默认参数值，让它每次调用时都取当前的sys.stdout：

import sys
from your_external_lib import msg

# 查看函数的默认参数位置（假设msg的参数顺序是content, file=...）
# 用__defaults__修改位置参数的默认值
msg.__defaults__ = (sys.stdout,)

# 如果是关键字参数的默认值，用__kwdefaults__（比如函数定义是msg(content, *, file=sys.stdout)）
# msg.__kwdefaults__['file'] = sys.stdout

这种方法更简洁，但依赖函数的参数结构，如果后续库更新修改了参数顺序或定义方式，可能会失效。

3. 替换sys.stdout对象的方法（而非替换变量） #

不修改sys.stdout变量本身，而是替换它的write和flush方法，让原绑定的stdout对象输出到新流：

import sys
from contextlib import contextmanager

@contextmanager
def redirect_stdout(new_stream):
    # 保存原方法
    old_write = sys.stdout.write
    old_flush = sys.stdout.flush

    # 替换为新流的方法
    sys.stdout.write = new_stream.write
    sys.stdout.flush = new_stream.flush

    try:
        yield
    finally:
        # 恢复原方法
        sys.stdout.write = old_write
        sys.stdout.flush = old_flush

使用时和普通重定向一样：

from io import StringIO

output = StringIO()
with redirect_stdout(output):
    msg("Test content")  # 无需传参

print(output.getvalue())

这种方法不需要修改库函数，但如果库中直接使用sys.stdout的其他属性（比如fileno），需要在new_stream中实现对应的方法，否则会报错。

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