Python：捕获含多调试打印语句的函数输出并保存至文件的Pythonic实现方案（无法使用脚本重定向）

看起来你之前的尝试因为输出格式被拆分破坏、以及列表写入文件的类型问题没达到预期，再加上并行执行的场景要求输出隔离，这里给你一个更贴合需求的改进方案：

import sys
import io
import SomeClass

class CaptureStdout:
    def __enter__(self):
        # 保存当前的标准输出流
        self.original_stdout = sys.stdout
        # 创建StringIO对象用于临时存储输出
        self.captured_stream = io.StringIO()
        sys.stdout = self.captured_stream
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 恢复原始的标准输出流
        sys.stdout = self.original_stdout
        # 获取完整的捕获内容（保留所有原始格式：换行、空格等）
        self.full_output = self.captured_stream.getvalue()
        # 关闭流释放内存，避免大循环下的资源占用
        self.captured_stream.close()

# 初始化目标类实例
my_class = SomeClass()

# 循环执行任务并捕获输出
for i in range(0, 9999):
    with CaptureStdout() as capture:
        result = my_class.function_to_test(some_variable=i)
    
    # 将捕获的完整输出写入对应文件
    with open(f"task_{i}_output.txt", "w", encoding="utf-8") as f:
        f.write(capture.full_output)
    
    # 在这里处理你的result
    # *do something else with the result*

方案核心优势： #

• 完整保留格式：直接通过getvalue()获取未被修改的完整输出字符串，不会像之前那样拆分换行破坏原有打印格式。
• 类型安全无报错：捕获的是纯字符串，写入文件时无需额外转换，解决了之前列表写入文件的类型错误问题。
• 并行场景友好：每个with块都是独立的上下文，标准输出的替换与恢复仅在当前线程/进程内生效，不会干扰其他并行任务的输出捕获（适配multiprocessing等并行框架）。
• 资源高效：主动关闭StringIO流，避免大循环场景下的内存占用累积。

扩展：同时捕获标准错误流（stderr） #

如果你的function_to_test还会往stderr打印内容（比如错误日志），可以扩展上下文管理器同时捕获两者：

class CaptureStdoutStderr:
    def __enter__(self):
        self.original_stdout = sys.stdout
        self.original_stderr = sys.stderr
        self.captured_stdout = io.StringIO()
        self.captured_stderr = io.StringIO()
        sys.stdout = self.captured_stdout
        sys.stderr = self.captured_stderr
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        sys.stdout = self.original_stdout
        sys.stderr = self.original_stderr
        self.stdout_output = self.captured_stdout.getvalue()
        self.stderr_output = self.captured_stderr.getvalue()
        self.captured_stdout.close()
        self.captured_stderr.close()

# 使用示例
with CaptureStdoutStderr() as capture:
    result = my_class.function_to_test(some_variable=i)
with open(f"task_{i}_full_output.txt", "w", encoding="utf-8") as f:
    f.write("=== 标准输出（STDOUT） ===\n")
    f.write(capture.stdout_output)
    f.write("\n=== 标准错误（STDERR） ===\n")
    f.write(capture.stderr_output)

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