如何用Python在同一命令窗口实现命令交互式输入输出？

问题根源 #

1. 输出延迟：cmd.exe执行完命令后会输出结果，随后打印新的命令提示符（如C:\>）。原代码中stdout.readline()会阻塞等待下一行输出，直到下一次输入命令时，才会一次性读取上一次的结果和新提示符，导致输出延迟。
2. 空输入阻塞：发送空命令后，cmd.exe仅输出新提示符，但原读取循环会持续等待更多输出，导致进程卡死。

修复代码 #

import subprocess
import threading
import queue

def stream_output(pipe, queue):
    """异步读取子进程输出到队列"""
    for line in iter(pipe.readline, b''):
        queue.put(line.decode('utf-8', errors='replace'))
    pipe.close()

# 启动cmd进程，禁用shell（直接调用cmd.exe无需shell=True）
cmd_proc = subprocess.Popen(
    'cmd.exe',
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT,  # 合并错误输出到标准输出
    shell=False,
    text=False
)

# 启动异步读取线程
output_queue = queue.Queue()
thread = threading.Thread(target=stream_output, args=(cmd_proc.stdout, output_queue))
thread.daemon = True
thread.start()

while True:
    try:
        user_cmd = input("Enter a command: ")
        if not user_cmd.strip():
            print("请输入有效命令")
            continue
        
        # 发送命令到cmd进程
        cmd_proc.stdin.write(user_cmd.encode('utf-8') + b'\n')
        cmd_proc.stdin.flush()
        
        # 收集输出直到出现cmd提示符（根据实际环境调整判断逻辑）
        result = []
        while True:
            line = output_queue.get(timeout=3)
            result.append(line)
            # 判断是否到达新的命令提示符（示例为以'>'结尾）
            if line.rstrip().endswith('>'):
                break
        
        # 打印执行结果（移除最后一行的提示符）
        print(''.join(result[:-1]), end='')
    except KeyboardInterrupt:
        print("\n程序退出")
        cmd_proc.stdin.write(b'exit\n')
        cmd_proc.stdin.flush()
        break
    except queue.Empty:
        print("命令执行超时，未收到结束提示符")

核心改进说明 #

• 异步输出读取：用线程持续读取cmd的输出，避免主线程因等待输出而阻塞。
• 空输入过滤：直接拦截空输入，提示用户输入有效命令，避免发送空命令导致的阻塞。
• 提示符识别：通过检测cmd的命令提示符（如>结尾）来判断命令执行完成，确保实时输出结果。
• 超时保护：读取队列时设置超时，防止命令无输出时进程卡死。
• 合并错误输出：将stderr重定向到stdout，确保错误信息也能被捕获显示。

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