使用subprocess.Popen时将大量数据通过管道传输到stdin

Question

我有点想了解解决这个简单问题的python方法是什么。

我的问题很简单。如果使用以下代码，它将挂起。子流程模块doc中对此进行了详细记录。

import subprocess

proc = subprocess.Popen(['cat','-'],
                        stdin=subprocess.PIPE,
                        stdout=subprocess.PIPE,
                        )
for i in range(100000):
    proc.stdin.write('%d\n' % i)
output = proc.communicate()[0]
print output

搜索一个解决方案（有一个非常有见地的线程，但是我现在已经迷失了），我发现这个解决方案（以及其他）使用了一个显式的fork：

import os
import sys
from subprocess import Popen, PIPE

def produce(to_sed):
    for i in range(100000):
        to_sed.write("%d\n" % i)
        to_sed.flush()
    #this would happen implicitly, anyway, but is here for the example
    to_sed.close()

def consume(from_sed):
    while 1:
        res = from_sed.readline()
        if not res:
            sys.exit(0)
            #sys.exit(proc.poll())
        print 'received: ', [res]

def main():
    proc = Popen(['cat','-'],stdin=PIPE,stdout=PIPE)
    to_sed = proc.stdin
    from_sed = proc.stdout

    pid = os.fork()
    if pid == 0 :
        from_sed.close()
        produce(to_sed)
        return
    else :
        to_sed.close()
        consume(from_sed)

if __name__ == '__main__':
    main()

尽管此解决方案在概念上非常容易理解，但它使用了一个以上的进程，并且与子进程模块相比，其停留在太低的级别（那只是为了隐藏此类事情…）。

我想知道：是否有一个简单而干净的解决方案使用了不会挂起的子流程模块，或者要实现此模式，我必须退后一步并实现一个老式的select循环或显式fork？

谢谢

Answer 1

如果需要纯Python解决方案，则需要将读取器或写入器放在单独的线程中。该threading软件包是一种轻量级的方法，可以方便地访问常见对象并且不会造成混乱。

import subprocess
import threading
import sys

proc = subprocess.Popen(['cat','-'],
                        stdin=subprocess.PIPE,
                        stdout=subprocess.PIPE,
                        )
def writer():
    for i in range(100000):
        proc.stdin.write(b'%d\n' % i)
    proc.stdin.close()
thread = threading.Thread(target=writer)
thread.start()
for line in proc.stdout:
    sys.stdout.write(line.decode())
thread.join()
proc.wait()

看到subprocess现代化的模块以支持流和协程可能会很整洁，这将允许混合使用Python片段和Shell片段的管道更优雅地构建。