Python subprocess调用交互式脚本无法读取输出的问题

以下是我用来模拟cmd.exe的Python代码：

import subprocess
import sys
import threading

def handle_output(shell):
    while shell:
        output = shell.stdout.readline(1)
        print(output, end="")
        sys.stdout.flush()

subprocess_shell = subprocess.Popen("cmd.exe", shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)

threading.Thread(target=handle_output, args=(subprocess_shell,), daemon=True).start()

while True:
    subprocess_shell.stdin.flush()
    cmd = input()
    subprocess_shell.stdin.write(cmd + "\n")

执行dir、echo这类普通命令时一切正常，但启动Python交互环境时完全没有输出。我尝试设置os.environ['PYTHONUNBUFFERED'] = '1'并传入env=os.environ，能解决python test.py这类脚本的实时输出问题，但启动Python交互环境时依旧没有任何输出。

预期输入python命令后应该输出：

Python 3.10.6 (tags/v3.10.6:9c7b4bd, Aug  1 2022, 21:53:49) [MSC v.1932 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>

但实际没有任何输出。

核心原因 #

Python交互模式会检测标准输入/输出是否连接到终端（TTY/PTY）：

• 如果是终端，才会输出欢迎信息和>>>提示符，同时采用适配交互的缓冲策略；
• 如果是普通管道（代码中使用的subprocess.PIPE），Python会判定为非终端环境，交互模式的欢迎信息、提示符不会输出，输入输出的缓冲机制也会异常。

PYTHONUNBUFFERED仅能解决非交互脚本的缓冲问题，无法让Python识别当前环境为终端，因此对交互模式无效。

解决方案（Windows环境） #

针对Windows平台，有三种可行方案，可根据需求选择：

方案1：用CREATE_NEW_CONSOLE创建新控制台窗口 #

让cmd.exe在独立控制台窗口运行，Python会识别到终端环境，输出正常：

import subprocess
import sys
import threading

def handle_output(shell):
    while shell.poll() is None:
        output = shell.stdout.read(1)
        if output:
            print(output, end="")
            sys.stdout.flush()

# 新增creationflags参数，让子进程在新窗口启动
subprocess_shell = subprocess.Popen(
    "cmd.exe",
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT,
    text=True,
    creationflags=subprocess.CREATE_NEW_CONSOLE
)

threading.Thread(target=handle_output, args=(subprocess_shell,), daemon=True).start()

while True:
    cmd = input()
    subprocess_shell.stdin.write(cmd + "\n")
    subprocess_shell.stdin.flush()

缺点：会弹出额外的cmd窗口，不过交互输出可正常捕获显示。

方案2：借助winpty模拟终端（无额外窗口，推荐） #

winpty是Windows下的伪终端工具，能让命令行程序误以为运行在终端环境：

1. 安装winpty：通过Chocolatey执行choco install winpty，或从官方渠道下载后添加到系统PATH；
2. 修改代码，对Python交互命令自动添加winpty前缀：

import subprocess
import sys
import threading

def handle_output(shell):
    while shell.poll() is None:
        output = shell.stdout.read(1)
        if output:
            print(output, end="")
            sys.stdout.flush()

subprocess_shell = subprocess.Popen(
    "cmd.exe",
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT,
    text=True
)

threading.Thread(target=handle_output, args=(subprocess_shell,), daemon=True).start()

while True:
    cmd = input()
    # 对python交互命令自动添加winpty前缀
    if cmd.strip() == "python":
        cmd = "winpty python"
    subprocess_shell.stdin.write(cmd + "\n")
    subprocess_shell.stdin.flush()

输入python后，即可正常显示欢迎信息和>>>提示符，交互操作完全正常。

方案3：用pty模块创建伪终端（跨平台） #

Python的pty模块可创建伪终端，Windows环境需先安装pywinpty依赖：

1. 安装依赖：pip install pywinpty；
2. 修改代码：

import subprocess
import sys
import threading
import pty
import os

def handle_output(fd):
    while True:
        try:
            output = os.read(fd, 1024).decode()
            if not output:
                break
            print(output, end="")
            sys.stdout.flush()
        except OSError:
            break

# 创建伪终端
master_fd, slave_fd = pty.openpty()

# 启动cmd.exe连接到伪终端
subprocess_shell = subprocess.Popen(
    "cmd.exe",
    stdin=slave_fd,
    stdout=slave_fd,
    stderr=slave_fd,
    text=False,
    close_fds=True
)

# 关闭slave端文件描述符
os.close(slave_fd)

# 启动线程处理输出
threading.Thread(target=handle_output, args=(master_fd,), daemon=True).start()

# 处理用户输入
while True:
    try:
        cmd = input()
        os.write(master_fd, (cmd + "\n").encode())
    except EOFError:
        break

该方案跨平台兼容Windows和Linux，无需额外弹出窗口，交互体验接近原生终端。

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