Python如何同时捕获print输出与stdout.buffer写入内容到变量

原有方案失效原因 #

• 你手动给StringIO附加的buffer是完全独立的BufferedWriter实例，文本输出、二进制输出分别存储在两个隔离的缓冲区，没有统一存储逻辑，自然无法拿到完整输出。
• BufferedWriter默认带缓存，如果不主动调用flush()，写入的内容不会同步到底层BytesIO，读取时会为空。

最优实现方案 #

我们可以自定义符合sys.stdout接口的上下文管理器，统一用字节缓冲区存储所有输出，同时兼容文本写入和二进制写入场景，用完自动恢复原始标准输出，不会污染全局环境。

实现代码 #

import sys
from io import BytesIO, TextIOWrapper

class StdoutCapturer:
    def __init__(self, encoding: str = "utf-8"):
        self.encoding = encoding
        self._original_stdout = None
        # 底层统一用字节缓冲区存储所有输出
        self._bytes_buffer = BytesIO()
        # 文本层，负责把print等文本写入编码后存入字节缓冲区
        self._text_wrapper = TextIOWrapper(self._bytes_buffer, encoding=encoding)
        # 暴露buffer属性，匹配原生sys.stdout的接口
        self.buffer = self._bytes_buffer

    def write(self, data: str):
        # 处理文本写入请求（print最终会调用这个方法）
        self._text_wrapper.write(data)

    def flush(self):
        # 刷出文本层的缓存到底层字节缓冲区
        self._text_wrapper.flush()

    def __enter__(self):
        # 进入上下文时替换全局stdout
        self._original_stdout = sys.stdout
        sys.stdout = self
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 退出上下文时恢复原始stdout，刷出所有缓存
        self.flush()
        sys.stdout = self._original_stdout

    def get_text_output(self) -> str:
        """获取解码后的文本格式输出"""
        self.flush()
        return self._bytes_buffer.getvalue().decode(self.encoding)

    def get_raw_output(self) -> bytes:
        """获取原始二进制格式输出"""
        self.flush()
        return self._bytes_buffer.getvalue()

使用示例 #

if __name__ == "__main__":
    a = 123
    # 启动捕获
    with StdoutCapturer() as capturer:
        # 场景1：普通print输出
        print("Normal Print")
        # 场景2：直接操作stdout.buffer写入二进制
        fds = [sys.stdin.buffer, sys.stdout.buffer, sys.stderr.buffer]
        fds[1].write(b"%d\n" % a)
        fds[1].flush()
    
    # 读取捕获结果
    print("捕获到的文本内容：")
    print(capturer.get_text_output())
    print("捕获到的原始字节：")
    print(capturer.get_raw_output())

方案优势 #

• 100%兼容sys.stdout的接口规范，两种输出场景都能完整捕获，不会丢内容
• 基于上下文管理器实现，用完自动恢复环境，无全局污染
• 同时支持文本和二进制两种结果输出，适配不同需求

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