如何让subprocess.run的输入值与输出一同写入stdout文件？

你的核心问题有两个：

1. 每次输入都重启Java程序，导致程序初始化提示重复输出
2. subprocess.run仅捕获程序输出，不会记录传入的输入内容，且非终端环境下Java程序的缓冲机制导致输出格式错乱

方案1：调整程序启动逻辑+手动拼接输入与输出 #

每个测试用例是一组连续输入，应该启动一次程序并传入整组输入，而非每个输入单独启动程序。之后将输入内容对应插入到程序的提示语后，还原真实交互格式。

修改后的代码：

input_value_lists = [["103\n", "-3\n", "85\n"], 
                     ["-10\n", "-1\n", "105\n","98\n"]]

with open(student_txt_file, "w") as txt_file:
    for input_list in input_value_lists:
        # 拼接当前测试用例的所有输入
        full_input = ''.join(input_list)
        # 单次启动程序，传入完整输入组
        result = subprocess.run(["java", hw_file], input=full_input, capture_output=True, text=True)
        stdout_lines = result.stdout.splitlines(keepends=True)
        
        # 匹配提示语，插入对应输入内容
        output_content = []
        input_idx = 0
        for line in stdout_lines:
            output_content.append(line)
            # 判断当前行是输入提示且还有未处理的输入
            if line.strip() == "Enter quiz grade:" and input_idx < len(input_list):
                output_content.append(input_list[input_idx])
                input_idx += 1
        # 将拼接好的内容写入文件
        txt_file.write(''.join(output_content))

方案2：用伪终端（pty）模拟真实运行环境 #

如果Java程序因为非终端环境的缓冲机制导致输出格式异常，使用pty模块模拟终端环境，可直接捕获和终端运行完全一致的交互内容（包含输入和输出）。

代码示例：

import pty
import os
import subprocess

input_value_lists = [["103\n", "-3\n", "85\n"], 
                     ["-10\n", "-1\n", "105\n","98\n"]]

with open(student_txt_file, "w") as txt_file:
    for input_list in input_value_lists:
        full_input = ''.join(input_list)
        # 创建伪终端
        master, slave = pty.openpty()
        # 绑定程序到伪终端，模拟真实运行环境
        proc = subprocess.Popen(["java", hw_file], stdin=slave, stdout=slave, stderr=slave, text=True)
        os.close(slave)
        
        # 写入测试用例输入
        os.write(master, full_input.encode())
        # 读取完整交互输出
        output = b""
        while True:
            try:
                data = os.read(master, 1024)
                if not data:
                    break
                output += data
            except OSError:
                break
        # 等待程序执行完毕
        proc.wait()
        os.close(master)
        
        # 将捕获的内容写入文件
        txt_file.write(output.decode())

• 重复初始化提示：原代码每个输入都启动一次Java程序，导致每次都会输出程序开头的初始化语句；改为每组输入启动一次程序即可解决。
• 输入未记录：subprocess.run的stdout仅捕获程序自身的输出，不会包含传入的输入内容，需要手动拼接或用伪终端捕获完整交互。
• 缓冲格式问题：Java程序在非终端环境下默认使用块缓冲，会导致输出不及时换行；伪终端会触发程序使用行缓冲，和终端运行格式完全一致。

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