循环中如何将stdout重定向至另一个函数？

不需要依赖文件，用内存流、管道或自定义流类就能实现两个函数间的数据流传递，以下是几种可行方案：

方案1：一次性收集输出后处理（小数据量适用） #

用io.StringIO收集foo的所有print输出，再传递给bar当作输入流：

import sys
from io import StringIO
from contextlib import redirect_stdout

def foo():
    for line in sys.stdin:
        # 模拟处理逻辑
        processed_line = line.strip().upper()
        print(processed_line)

def bar(input_stream):
    # 将传入的流当作bar的"stdin"
    for line in input_stream:
        # 模拟bar的处理逻辑
        print(f"[BAR处理结果] {line.strip()}")

if __name__ == "__main__":
    output_buffer = StringIO()
    # 重定向foo的print到内存缓冲区
    with redirect_stdout(output_buffer):
        foo()
    # 将缓冲区指针移到开头，供bar读取
    output_buffer.seek(0)
    bar(output_buffer)

这种方式实现简单，但需要等foo完全执行完，bar才会开始处理，适合输出量不大的场景。

方案2：逐行实时处理（大数据量/实时场景适用） #

用操作系统管道+多线程实现实时数据流：foo的每一行print输出会立刻被bar读取处理，无需等待foo执行完毕。

import sys
import os
import threading
from contextlib import redirect_stdout

def foo(write_fd):
    # 重定向stdout到管道写端
    with os.fdopen(write_fd, 'w') as stdout_file, redirect_stdout(stdout_file):
        for line in sys.stdin:
            processed_line = line.strip().upper()
            print(processed_line)
            # 强制刷新缓冲区，确保内容实时写入管道
            sys.stdout.flush()

def bar(read_fd):
    # 将管道读端当作bar的"stdin"
    with os.fdopen(read_fd, 'r') as stdin_file:
        for line in stdin_file:
            print(f"[BAR处理结果] {line.strip()}")

if __name__ == "__main__":
    # 创建管道，得到读、写两个文件描述符
    read_fd, write_fd = os.pipe()
    
    # 启动bar线程负责读取管道内容
    bar_thread = threading.Thread(target=bar, args=(read_fd,))
    bar_thread.start()
    
    # 主线程执行foo，输出写入管道
    foo(write_fd)
    
    # 关闭写端，让bar知道输入已结束
    os.close(write_fd)
    # 等待bar线程处理完所有内容
    bar_thread.join()

方案3：直接传递输出行到处理函数（最灵活） #

自定义一个流类，让foo的print输出直接调用bar的处理逻辑，无需模拟stdin读取：

import sys
from contextlib import redirect_stdout

class OutputRedirector:
    def __init__(self, line_handler):
        self.line_handler = line_handler
    
    def write(self, text):
        # 按行分割输出内容，传递给处理函数
        for line in text.strip().splitlines():
            if line:
                self.line_handler(line)
    
    def flush(self):
        # 实现flush方法，避免缓冲问题
        pass

def bar_process_line(line):
    # 这里直接处理每一行输出，替代原bar的循环逻辑
    print(f"[BAR处理结果] {line}")

def foo():
    for line in sys.stdin:
        processed_line = line.strip().upper()
        print(processed_line)

if __name__ == "__main__":
    # 重定向foo的print到自定义处理类
    redirector = OutputRedirector(bar_process_line)
    with redirect_stdout(redirector):
        foo()

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