如何实现Python子进程STDIN/STDOUT交错输出以模拟终端交互？

问题本质 #

你当前用subprocess.run批量传入输入、一次性捕获输出的方式，无法还原程序逐行交互的时序——程序先输出提示，等待输入，再输出下一个提示，以此类推。批量输入会让程序把所有输入一次性读走，输出也会一次性吐出，导致提示和输入无法对应。

已有的成熟方案 #

这个问题是交互式程序自动化测试/演示的常见需求，有现成的工具和方法可以解决，最优方向分为两类：

1. 用伪终端（PTY）模拟真实终端环境 #

真实终端下程序会通过PTY和用户交互，用PTY启动子进程可以让程序认为自己在和真实终端对话，自动保持输出和输入的时序对应。Python里可以用标准库pty结合subprocess实现：

import pty
import os
import subprocess

def run_interactive(cmd, input_lines):
    # 创建伪终端
    master, slave = pty.openpty()
    # 启动子进程，绑定伪终端的slave端作为stdin/stdout/stderr
    proc = subprocess.Popen(cmd, stdin=slave, stdout=slave, stderr=slave, text=True)
    os.close(slave)
    
    output = []
    input_iter = iter(input_lines)
    
    try:
        while proc.poll() is None:
            # 读取程序输出的提示
            data = os.read(master, 1024).decode()
            if not data:
                break
            output.append(data)
            # 发送对应的输入
            try:
                line = next(input_iter) + '\n'
                os.write(master, line.encode())
                output.append(line)
            except StopIteration:
                break
        # 读取剩余输出
        remaining = os.read(master, 1024).decode()
        if remaining:
            output.append(remaining)
    finally:
        os.close(master)
        proc.wait()
    
    return ''.join(output)

# 使用示例
result = run_interactive(['python', 'student_program.py'], ['blue', '7'])
print(result)

这个方法能完美还原终端交互的时序，输出会和用户手动操作时完全一致。

2. 用Expect风格的库控制交互流程 #

如果需要更精细地控制“等待特定输出→发送对应输入”的逻辑，可以用pexpect库（Python的Expect实现）：

import pexpect

def run_with_pexpect(cmd, input_mapping):
    child = pexpect.spawn(cmd, encoding='utf-8')
    output = []
    for prompt, response in input_mapping.items():
        # 等待指定的提示输出
        child.expect(prompt)
        # 记录提示内容
        output.append(child.before + child.match.group())
        # 发送输入并记录
        child.sendline(response)
        output.append(response + '\n')
    # 读取剩余输出
    output.append(child.read())
    child.close()
    return ''.join(output)

# 使用示例
result = run_with_pexpect('python student_program.py', {
    'name a color: ': 'blue',
    'pick a number: ': '7'
})
print(result)

这种方式适合需要严格匹配提示内容的场景，能确保输入和提示精准对应，避免时序混乱。

方案选择建议 #

• 如果只需要还原终端交互的视觉效果，不需要提前知道程序的提示内容，伪终端方案更通用简单。
• 如果需要确保输入和特定提示严格对应（比如学生程序的提示可能有细微差异），pexpect方案更可靠。

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