如何为Python subprocess进程传递多个输入？

我来帮你搞定这个问题，你遇到的核心痛点是communicate()只能调用一次，而直接写stdin时又因为缓冲和等待逻辑缺失导致执行顺序混乱。下面给你两种实用的解决方案：

方案一：一次性通过communicate()传递所有输入 #

这是最省心的方法——因为Python的input()函数会读到换行符为止，你只需要把多个输入用换行符\n拼接起来，一次性传给communicate()的input参数就行，子进程会自动依次读取每个输入。

示例代码：

import subprocess

# 启动子进程，指定stdin和stdout为管道
p = subprocess.Popen(
    ['python', 'registration.py'],
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    text=True  # 自动处理字符串，无需手动编码解码
)

# 用\n分隔两个输入，一次性传递
output, _ = p.communicate(input="phone_number\nverification_code\n")

# 打印子进程的输出结果
print("子进程输出：")
print(output)

这个方法的优势是简洁可靠，communicate()会自动处理输入输出的缓冲，还会等待子进程完全结束，不用额外操心。

方案二：分步写入stdin并处理缓冲与等待 #

如果你确实需要分步发送输入（比如要等第一个输入处理完成、拿到响应后再发第二个），那就要手动处理缓冲区，并确保子进程有足够时间处理输入。

你之前的代码有几个问题：

1. 没调用flush()，写入的内容留在缓冲区里，没立即传给子进程；
2. 没有等待子进程处理输入，直接执行后续代码，导致"Done"先打印；
3. 没必要加-i参数（交互模式），反而会引入额外的缓冲问题。

修正后的示例代码：

import subprocess
import time

# 启动子进程，去掉多余的-i参数
p = subprocess.Popen(
    ['python', 'registration.py'],
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    text=True
)

# 发送第一个输入后必须调用flush()，强制把内容传给子进程
p.stdin.write("phone_number\n")
p.stdin.flush()

# 可选：读取子进程的第一个输出，确认输入已被处理
first_response = p.stdout.readline()
print(f"收到第一个响应：{first_response.strip()}")

# 按照需求最多等待15秒
time.sleep(15)

# 发送第二个输入并刷新缓冲区
p.stdin.write("verification_code\n")
p.stdin.flush()

# 等待子进程结束，获取剩余输出
remaining_output, _ = p.communicate()
print(f"剩余输出：{remaining_output.strip()}")

print("Done")

这里的关键是flush()——它会把缓冲区里的内容立即推送给子进程；而readline()会阻塞等待子进程的输出，确保第一个输入已经被处理，避免出现顺序颠倒的问题。

总结 #

• 如果不需要分步处理输入，优先用方案一，代码简洁且不易出错；
• 如果需要分步操作或依赖前一个输入的结果，用方案二，记得加上flush()和必要的等待/输出读取逻辑。

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