如何通过ProcessBuilder为Python脚本输入Yes/No以继续执行

这个问题我之前也碰到过，本质就是要给启动的Python进程主动写入输入，同时还要处理好进程的输出避免阻塞。下面一步步给你讲清楚怎么做：

核心思路 #

当你用ProcessBuilder启动Python脚本时，这个脚本的标准输入其实对应Java程序里的OutputStream（没错：Java的输出流就是进程的输入流）。我们需要：

• 把Java的输出流和Python进程的标准输入绑定
• 读取Python脚本的输出，判断什么时候需要输入（或者直接提前写入，如果你确定脚本一定会请求输入）
• 写入对应的"Yes"或"No"并刷新缓冲区

完整代码示例 #

假设你的Python脚本test.py是这样的：

user_input = input("请输入Yes或No继续: ")
if user_input.strip().lower() == "yes":
    print("执行Yes分支")
else:
    print("执行No分支")

对应的Java代码如下：

import java.io.*;

public class ProcessBuilderDemo {
    public static void main(String[] args) throws IOException, InterruptedException {
        // 1. 创建ProcessBuilder，指定Python命令（注意加上-u参数禁用缓冲，避免输出延迟）
        ProcessBuilder pb = new ProcessBuilder("python", "-u", "test.py");
        Process process = pb.start();

        // 2. 获取进程的输入、输出、错误流
        OutputStream processInput = process.getOutputStream(); // 给Python脚本发输入
        InputStream processOutput = process.getInputStream(); // 读取Python脚本的正常输出
        InputStream processError = process.getErrorStream(); // 读取Python脚本的错误输出

        // 3. 启动线程读取输出和错误，避免阻塞主线程
        new Thread(() -> readStream("输出", processOutput)).start();
        new Thread(() -> readStream("错误", processError)).start();

        // 4. 等待片刻（或者根据输出判断时机），写入"Yes"并刷新
        Thread.sleep(100); // 简单等待脚本输出请求，实际可以通过读取输出判断是否需要输入
        processInput.write("Yes\n".getBytes());
        processInput.flush();

        // 5. 等待进程执行完成
        int exitCode = process.waitFor();
        System.out.println("进程退出码: " + exitCode);

        // 关闭流
        processInput.close();
        processOutput.close();
        processError.close();
    }

    // 工具方法：读取流内容并打印
    private static void readStream(String type, InputStream stream) {
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(stream))) {
            String line;
            while ((line = reader.readLine()) != null) {
                System.out.println(type + ": " + line);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

关键细节解释 #

• python -u参数：这个非常重要！Python默认会缓冲标准输出，导致Java无法及时收到脚本的"请输入Yes或No继续:"提示，从而错过输入时机。加上-u参数可以禁用缓冲，让输出实时传递给Java程序。
• 单独线程处理输出流：如果主线程只处理输入，不读取输出，当Python脚本的输出缓冲区满时，进程会卡住无法继续执行。所以必须用单独的线程来读取输出和错误流。
• 写入后要flush()：OutputStream的缓冲区不会自动刷新，必须手动调用flush()才能让内容立刻发送给Python进程，否则脚本会一直等待输入。
• 换行符\n：Python的input()函数会读取到换行符为止，所以写入的时候一定要加上\n，否则脚本会一直处于等待输入的状态。

进阶优化：根据输出判断输入时机 #

上面的代码用Thread.sleep(100)来等待脚本输出请求，这不是很可靠。更好的方式是在读取输出的线程里，当捕捉到"请输入Yes或No继续:"这样的提示时，再触发输入操作。比如可以修改读取线程的逻辑：

private static void readStreamWithInputTrigger(String type, InputStream stream, OutputStream inputStream) {
    try (BufferedReader reader = new BufferedReader(new InputStreamReader(stream))) {
        String line;
        while ((line = reader.readLine()) != null) {
            System.out.println(type + ": " + line);
            // 当捕捉到输入提示时，写入Yes
            if (line.contains("请输入Yes或No继续:")) {
                inputStream.write("Yes\n".getBytes());
                inputStream.flush();
            }
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}

然后启动线程时传入processInput：

new Thread(() -> readStreamWithInputTrigger("输出", processOutput, processInput)).start();

这样就可以精确地在脚本请求输入时自动响应，不需要依赖固定的等待时间。

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