Python多进程模块中input()函数导致进程调用失败问题排查

问题背景 #

在Python 3.11下编写了这段multiprocessing测试代码：

from multiprocessing import Process
import time

def proc(text):
    while True:
        print(text)
        time.sleep(1)

print(__name__)

if __name__ == '__main__':
    p = Process(target=proc, args=('hi',))
    p.start()

    input('Press any key to exit')
    p.terminate()

预期子进程持续打印hi，直到按下按键终止。但在Windows 10的PyCharm中运行时，看不到子进程的任何输出；移除input()后程序却能正常工作。

问题根源 #

1. Windows进程启动机制：Windows上multiprocessing默认采用spawn模式，子进程会重新导入主模块并执行目标函数。但主进程因input()调用进入阻塞状态后，子进程的print输出被系统输出缓冲拦截，无法实时显示。
2. PyCharm控制台的缓冲特性：PyCharm的集成控制台对多进程输出的缓冲策略会放大这个问题——当主进程等待输入时，子进程的输出不会被主动刷新到控制台，导致看起来像子进程根本没启动。

解决方案 #

方法1：强制刷新输出 #

修改proc函数中的print语句，添加flush=True参数强制立即刷新缓冲，确保输出能实时显示：

def proc(text):
    while True:
        print(text, flush=True)
        time.sleep(1)

方法2：用线程处理输入 #

把input()操作放到单独线程中，避免主进程长时间阻塞影响子进程输出的缓冲刷新：

from multiprocessing import Process
import time
import threading

def proc(text):
    while True:
        print(text)
        time.sleep(1)

def handle_input(process):
    input('Press any key to exit')
    process.terminate()

if __name__ == '__main__':
    p = Process(target=proc, args=('hi',))
    p.start()
    # 启动守护线程处理输入
    threading.Thread(target=handle_input, args=(p,), daemon=True).start()
    p.join()

方法3：调整PyCharm控制台设置 #

进入File > Settings > Build, Execution, Deployment > Console > Python Console，取消勾选Emulate terminal in output console。不过这个设置会改变控制台的整体行为，建议优先使用前两种方法。

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