如何在Python中通过子进程运行多语言脚本并为输出添加来源标识

没问题，这事儿完全能实现！咱们可以借助Python的subprocess模块启动多个子进程，同时捕获它们的输出，再给每一行输出打上对应的来源标识——完全不用修改原来的脚本文件。下面给你详细讲怎么搞：

核心思路 #

1. 用subprocess.Popen启动目标脚本/程序，捕获它们的标准输出（stdout）和错误输出（stderr）；
2. 用线程来实时处理每个子进程的输出流，避免主进程阻塞，保证多个程序的输出能实时显示；
3. 给每一行输出加上对应的来源文件名标识，比如main.java > Hello, World!。

完整代码示例 #

import subprocess
import threading

def process_output(process_name, output_stream):
    """处理子进程的输出流，给每一行加上来源标识后打印"""
    # 逐行读取输出流，直到流结束（b''表示结束）
    for line in iter(output_stream.readline, b''):
        # 把字节流解码为字符串，同时去掉末尾的换行符
        cleaned_line = line.decode('utf-8').rstrip('\n')
        # 打印带标识的输出
        print(f"{process_name} > {cleaned_line}")

def main():
    # 定义要运行的程序列表：(显示用的标识名称, 命令行参数列表)
    programs = [
        ("main.py", ["python", "main.py"]),
        ("main.java", ["java", "Main"])  # 注意：这里假设已经编译好Main.class文件
    ]

    # 保存子进程和线程的引用，方便后续等待完成
    processes = []
    threads = []

    for name, cmd in programs:
        # 启动子进程，捕获stdout和stderr
        proc = subprocess.Popen(
            cmd,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=False  # 以字节流方式读取，兼容不同编码场景
        )
        processes.append(proc)

        # 创建线程处理标准输出
        stdout_thread = threading.Thread(
            target=process_output,
            args=(name, proc.stdout)
        )
        stdout_thread.start()
        threads.append(stdout_thread)

        # 创建线程处理错误输出，额外加上[ERROR]标识区分
        stderr_thread = threading.Thread(
            target=process_output,
            args=(f"{name} [ERROR]", proc.stderr)
        )
        stderr_thread.start()
        threads.append(stderr_thread)

    # 等待所有子进程执行完成
    for proc in processes:
        proc.wait()

    # 等待所有线程处理完剩余的输出
    for thread in threads:
        thread.join()

if __name__ == "__main__":
    main()

关键细节说明 #

• 线程的作用：如果不用线程，主进程会卡在等待某个子进程输出的环节，没法同时处理多个程序的实时输出。用线程可以让多个输出流的处理并行进行。
• Java文件的编译：如果你的main.java还没编译成.class文件，可以在启动Java进程前先加一步编译：

  # 编译main.java
  subprocess.run(["javac", "main.java"], check=True)
  

  把这段代码放在programs定义之后，启动子进程之前即可。
• 编码处理：如果你的脚本输出不是UTF-8编码，可以修改line.decode('utf-8')里的编码参数，比如改成gbk（适合中文Windows环境）。
• 错误输出区分：代码里给错误输出加了[ERROR]标识，这样能一眼区分正常输出和报错信息，排查问题更方便。

备注：内容来源于stack exchange，提问作者Tousend1000