使用subprocess.Popen调用python脚本并刷新数据

时间:2013-01-22 15:17:18

标签: python subprocess pipe

好的,所以我看过这样的十几个帖子,但是没有一个给出完整的答案,我到目前为止尝试的一切都不适合我。

1)不断输出一些数据并刷新的脚本:

import time
import sys

if __name__ == '__main__':
    for i in range(5):
        print i,
        sys.stdout.flush()
        time.sleep(1)

2)用Popen调用第一个脚本的脚本,应该逐个打印数字,但由于某种原因没有,并一次打印出来:

import sys
import subprocess

if __name__ == '__main__':
    process = subprocess.Popen(['python', 'flush.py'], stdout = subprocess.PIPE )
    for line in iter(process.stdout.readline, ''):
        print line,
        sys.stdout.flush()

我有点困惑的第一件事是在第一个脚本中,如果你删除了flush它会在一行中返回输出O_O ...我很确定这是因为时间.sleep但仍然有点期待它像一个标准输出一样返回不断返回值0,1,2,3,4但不是全部在一起,当然冲洗解决它,但只是奇怪,至少对我来说...

主要问题: 第二个脚本是不是一个一个地返回数字,而是一次性返回一个输出.....我需要的是看到数字一个一个地弹出...

我在某处读到它没有返回EOF,Popen等待关闭管道,这就是为什么它会跑到最后......

那我该怎么办或接下来尝试一下? 提前致谢。

1 个答案:

答案 0 :(得分:8)

正如@Warren Weckesser's comment所述,您的问题与buffering issues无关。

父进程中的

.readline()在读取换行符或达到EOF之前不会返回。您的子进程根本不打印任何换行符,因此您的父进程在子进程结束之前不会打印任何

最小修复只是删除子脚本中print i,末尾的逗号。

这也有效:

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

p = Popen([sys.executable or 'python',
           '-u', # unbuffer stdout (or make it line-buffered on Python 3)
           '-c',
           """
import time

for i in range(5):
    print(i) # <-- no comma i.e., each number is on its own line
    time.sleep(1)
"""], stdout=PIPE, bufsize=1)
for line in iter(p.stdout.readline, b''):
    print(int(line)**2)

示例:

 $ python parent.py
 0
 1
 4
 9
 16

数字每秒打印一次,无需等待子进程结束。

如果您不想更改子脚本,那么您应该使用在空格处停止的readline()而不是换行符,例如:

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

p = Popen(['python2', 'child.py'], stdout=PIPE, bufsize=0)
for token in generate_tokens(p.stdout):
    print(int(token))

其中generate_tokens()产生以空格分隔的标记:

def generate_tokens(pipe):
    buf = []
    while True:
        b = pipe.read(1) # read one byte
        if not b: # EOF
            pipe.close()
            if buf:
                yield b''.join(buf)
            return
        elif not b.isspace(): # grow token
            buf.append(b)
        elif buf: # full token read
            yield b''.join(buf)
            buf = []

一旦孩子打印出来,它也会打印整数。