python - 在 Python 中读取另一个程序的控制台输出

Question

有两个程序 - parent.py 和 myscript.py，如下所示。 parent.py 在控制台中不断打印消息。 myscript.py 需要访问 parent.py 打印的内容。

parent.py:

import time

while 1:
     time.sleep(1)
     print "test"

myscript.py:

import subprocess
p = None

p = subprocess.Popen(['python', 'parent.py'],
                 stdout=subprocess.PIPE,
                 stderr=subprocess.STDOUT,
                 )
for line in p.stdout:
       print line

我希望 myscript.py 动态捕获 parent.py 的输出。可以看出 parent.py 有一个无限循环 - 脚本永远不会停止，但每当它输出单词“test”时，myscript.py 也应该输出相同的内容。但是 myscript.py 没有给出任何输出。

我尝试用 just 替换 parent.py 中的代码

print "Hello"

由于程序执行完毕，myscript.py 也打印了“Hello”。所以我想除非 parent.py 的执行完成，否则 myscript.py 不会读取它。这是怎么回事？

我还尝试用这个片段替换 myscript.py 中的循环:

for line in iter(p.stdout.readline, ''):
    print line

仍然没有输出。

我的问题是:即使 parent.py 永远不会完成其执行，我如何让 myscript.py 动态读取 parent.py 正在打印的所有内容。

(我发现了类似的问题，但它们要么不起作用，要么其中一些根本没有答案。)

Answer 1

这是一个block buffering issue :当 stdout 被重定向到管道时 parent.py 将其输出累积在内部缓冲区 (~4K-8K) 中，即，在缓冲区溢出之前大约一个小时内您将看不到任何内容。

要禁用缓冲，请传递 -u 命令行选项:

#!/usr/bin/env python
import os
import sys
from subprocess import Popen, PIPE, STDOUT

script_path = os.path.join(get_script_dir(), 'parent.py')
p = Popen([sys.executable, '-u', script_path],
          stdout=PIPE, stderr=STDOUT, bufsize=1)
with p.stdout:
    for line in iter(p.stdout.readline, b''):
        print line,
p.wait()

参见:

• Python: read streaming input from subprocess.communicate()
• Call python script with input with in a python script using subprocess

注意:如果您不需要对输出做任何事情，那么只需删除 stdout=PIPE，即可在控制台中查看输出:

subprocess.check_call([sys.executable, script_path])