14

我正在尝试使用subprocess.popen ()将 io.BytesIO() 字节流通过管道传输到单独的程序，但我不知道如何或是否有可能。文档和示例都是关于文本和换行符的。

当我掀起这样的事情时：

import io
from subprocess import *

stream = io.BytesIO()
someStreamCreatingProcess(stream)

command = ['somecommand', 'some', 'arguments']  
process = Popen(command, stdin=PIPE)
process.communicate(input=stream)

我明白了

Traceback (most recent call last):
  File "./test.py", line 9, in <module>
    procOut         = process.communicate(input=stream)
  File "/usr/lib/python2.7/subprocess.py", line 754, in communicate
    return self._communicate(input)
  File "/usr/lib/python2.7/subprocess.py", line 1322, in _communicate
    stdout, stderr = self._communicate_with_poll(input)
  File "/usr/lib/python2.7/subprocess.py", line 1384, in _communicate_with_poll
    chunk = input[input_offset : input_offset + _PIPE_BUF]
TypeError: '_io.BytesIO' object has no attribute '__getitem__'

我认为popen()仅适用于文本。我错了吗？
有不同的方法可以做到这一点吗？

4

2 回答 2

10

正如@falsetru 所说，您不能BytesIO()直接流式传输对象；您需要先从中获取一个字节串。这意味着在您调用pass tostream 之前应该已经写入了所有内容。stream.getvalue()process.communicate()

如果您想流式传输而不是一次提供所有输入，那么您可以删除BytesIO()对象并直接写入管道：

from subprocess import Popen, PIPE

process = Popen(['command', 'arg1'], stdin=PIPE, bufsize=-1)
someStreamCreatingProcess(stream=process.stdin) # many `stream.write()` inside
process.stdin.close() # done (no more input)
process.wait()

someStreamCreatingProcess()在完成写入流之前不应返回。如果它立即返回，那么它应该stream.close()在将来的某个时候调用（process.stdin.close()在你的代码中删除）：

from subprocess import Popen, PIPE

process = Popen(['command', 'arg1'], stdin=PIPE, bufsize=-1)
someStreamCreatingProcess(stream=process.stdin) # many `stream.write()` inside
process.wait() # stream.close() is called in `someStreamCreatingProcess`
于 2013-12-02T06:55:34.560 回答
6

根据subprocess.Popen.communicate：

可选的输入参数应该是要发送到子进程的字符串，或者如果不应该向子进程发送数据，则为 None。

要从BytesIO对象获取（字节）字符串值，请使用getvalue：

process.communicate(input=stream.getvalue())
于 2013-12-02T04:55:24.180 回答