如何与正在运行的 Python 进程交互？

Linux

2024-04-05 22:49:43

与 Python 进程进行交互

问题概述

在进行自动化任务或与其他应用程序协作时，你需要与正在运行的 Python 进程进行交互，提供输入，接收输出，并根据响应动态调整你的脚本。

解决方案：管道通信

使用 Python 的 subprocess 模块，你可以创建管道，在父进程（你的脚本）和子进程（Python 脚本）之间建立通信通道。

步骤：

1. 创建管道

使用 subprocess.Popen() 创建一个管道，指定脚本路径、标准输入 (stdin) 和标准输出 (stdout) 流为管道。

2. 与进程交互

使用 communicate() 方法将数据写入子进程的 stdin 并读取 stdout 的输出。
stdin.write() 发送输入，stdin.flush() 刷新缓冲区，stdout.read() 读取输出。

3. 循环交互

在循环中进行交互，直到收到所需的输出或子进程退出。
根据输出采取相应操作，例如跳出循环。

示例脚本：

import subprocess

pipe = subprocess.Popen(["./my_script.py"], stdin=subprocess.PIPE, stdout=subprocess.PIPE)

while True:
    pipe.stdin.write(b"1\n")
    pipe.stdin.flush()

    output = pipe.stdout.read()
    print(output.decode("utf-8"))

    if "You win" in output:
        break

pipe.stdin.close()
pipe.stdout.close()