如何在Python中交替读写子进程的stdin、stdout和stderr？

核心问题分析 #

你遇到的挂起、无法读取输出的问题，本质是子进程的缓冲机制导致的：像python3这类交互式工具，在非终端环境（比如直接用subprocess.PIPE）会启用全缓冲，输出内容会积压在缓冲区里，不会实时返回给父进程；只有当缓冲区满、进程主动刷新或退出时，父进程才能读到内容。而用伪终端（pty）模拟交互式终端环境，能让子进程切换为行缓冲，实时输出提示符和内容。

解决方案代码 #

以下是满足你需求的实现，无需多线程/多进程，适配Linux环境：

import pty
import os
import subprocess

def interactive_session(cmd):
    # 创建伪终端对，模拟交互式终端环境
    master_fd, slave_fd = pty.openpty()
    
    # 启动子进程，将其标准IO绑定到伪终端的slave端
    proc = subprocess.Popen(
        cmd,
        stdin=slave_fd,
        stdout=slave_fd,
        stderr=slave_fd,
        shell=True
    )
    os.close(slave_fd)  # 无需保留slave端，通过master端即可读写
    
    # 设置master端为非阻塞模式，避免无输出时挂起
    os.set_blocking(master_fd, False)
    
    try:
        while proc.poll() is None:  # 子进程未结束时循环
            # 读取当前所有可用输出（包括无换行符的内容）
            output = b""
            try:
                while True:
                    chunk = os.read(master_fd, 1024)
                    if not chunk:
                        break
                    output += chunk
            except BlockingIOError:
                # 无更多可读内容时跳出
                pass
            
            if output:
                print(output.decode("utf-8"), end="")
                # 检测到python提示符时，接受用户输入并发送给子进程
                if b">>> " in output or b"... " in output:
                    user_input = input("").encode("utf-8") + b"\n"
                    os.write(master_fd, user_input)
        
        # 读取子进程结束前的剩余输出
        final_output = b""
        try:
            while True:
                chunk = os.read(master_fd, 1024)
                if not chunk:
                    break
                final_output += chunk
        except BlockingIOError:
            pass
        if final_output:
            print(final_output.decode("utf-8"), end="")
        
        print(f"\n进程结束，返回码: {proc.returncode}")
    except KeyboardInterrupt:
        proc.terminate()
        proc.wait()
        print("\n进程被终止")
    finally:
        os.close(master_fd)

# 启动python3交互式会话
interactive_session("python3")

关键细节说明 #

1. 伪终端的作用：pty.openpty()创建的伪终端让子进程误以为在真实终端中，触发其交互式行为（比如python3显示>>>提示符、使用行缓冲），确保输出实时可读。
2. 非阻塞读取：通过os.set_blocking(master_fd, False)设置非阻塞，避免父进程在无输出时挂起，实现循环中的交织读写。
3. 读写逻辑：每次循环读取当前所有可用输出，检测到目标提示符后，将用户输入写入子进程，完成交互流程。

关于python3的特殊性 #

python3确实有特殊的缓冲行为：

• 当运行在终端时，stdout是行缓冲，实时输出内容；
• 当运行在管道下时，stdout是全缓冲，只有缓冲区满或进程退出才会输出。
  用伪终端模拟终端环境，就能规避这个问题，让python3进入交互式模式，正常响应输入和输出。

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