Python - 如何将字符串传递到subprocess.Popen（使用stdin参数）？

Question

如果我做到以下几点：

import subprocess 
from cStringIO import StringIO 
subprocess.Popen(['grep','f'],stdout=subprocess.PIPE,stdin=StringIO('one\ntwo\nthree\nfour\nfive\nsix\n')).communicate()[0] 

我得到：

Traceback (most recent call last): 
    File "<stdin>", line 1, in ? 
    File "/build/toolchain/mac32/python-2.4.3/lib/python2.4/subprocess.py", line 533, in __init__ 
    (p2cread, p2cwrite, 
    File "/build/toolchain/mac32/python-2.4.3/lib/python2.4/subprocess.py", line 830, in _get_handles 
    p2cread = stdin.fileno() 
AttributeError: 'cStringIO.StringI' object has no attribute 'fileno' 

显然一个cStringIO.StringIO对象不呱足够接近到一个文件中的鸭子，以满足subprocess.Popen。我如何解决这个问题？

Answer 1

Popen.communicate()文档：

需要注意的是，如果你想将数据发送到 进程的标准输入，你需要 与 标准输入= PIPE创建POPEN对象。同样，要在结果元组 中获得除None以外的任何其他值 ，您还需要stdout = PIPE和/或 stderr = PIPE。

更换os.popen *

pipe = os.popen(cmd, 'w', bufsize) 
    # ==> 
    pipe = Popen(cmd, shell=True, bufsize=bufsize, stdin=PIPE).stdin 

警告使用通信（），而不是 stdin.write（），stdout.read（）或 stderr.read（）以避免由于 到其他任何OS管道缓冲区 填充和阻塞子进程而导致的死锁。

所以，你的例子可以写成如下：

from subprocess import Popen, PIPE, STDOUT 

p = Popen(['grep', 'f'], stdout=PIPE, stdin=PIPE, stderr=STDOUT)  
grep_stdout = p.communicate(input=b'one\ntwo\nthree\nfour\nfive\nsix\n')[0] 
print(grep_stdout.decode()) 
# -> four 
# -> five 
# -> 

---

在当前的Python 3版本，你可以使用subprocess.run，通过输入一个字符串到一个外部命令，并得到其退出状态，以及它作为一个字符串在一个调用输出回：

#!/usr/bin/env python3 
from subprocess import run, PIPE 

p = run(['grep', 'f'], stdout=PIPE, 
     input='one\ntwo\nthree\nfour\nfive\nsix\n', encoding='ascii') 
print(p.returncode) 
# -> 0 
print(p.stdout) 
# -> four 
# -> five 
# -> 

Answer 2

我想通了以下解决方法：

>>> p = subprocess.Popen(['grep','f'],stdout=subprocess.PIPE,stdin=subprocess.PIPE) 
>>> p.stdin.write(b'one\ntwo\nthree\nfour\nfive\nsix\n') #expects a bytes type object 
>>> p.communicate()[0] 
'four\nfive\n' 
>>> p.stdin.close() 

是否还有更好的吗？

Answer 3

“显然，一个cStringIO.StringIO对象不呱足够接近到一个文件中的鸭子，以满足subprocess.Popen”

:-)

恐怕不行。管道是一个低级别的操作系统概念，所以它绝对需要一个由操作系统级文件描述符表示的文件对象。你的解决方法是正确的。

Answer 4

p = Popen(['grep', 'f'], stdout=PIPE, stdin=PIPE, stderr=STDOUT)  
p.stdin.write('one\n') 
time.sleep(0.5) 
p.stdin.write('two\n') 
time.sleep(0.5) 
p.stdin.write('three\n') 
time.sleep(0.5) 
testresult = p.communicate()[0] 
time.sleep(0.5) 
print(testresult) 

Answer 5

from subprocess import Popen, PIPE 
from tempfile import SpooledTemporaryFile as tempfile 
f = tempfile() 
f.write('one\ntwo\nthree\nfour\nfive\nsix\n') 
f.seek(0) 
print Popen(['/bin/grep','f'],stdout=PIPE,stdin=f).stdout.read() 
f.close() 

Answer 6

""" 
Ex: Dialog (2-way) with a Popen() 
""" 

p = subprocess.Popen('Your Command Here', 
       stdout=subprocess.PIPE, 
       stderr=subprocess.STDOUT, 
       stdin=PIPE, 
       shell=True, 
       bufsize=0) 
p.stdin.write('START\n') 
out = p.stdout.readline() 
while out: 
    line = out 
    line = line.rstrip("\n") 

    if "WHATEVER1" in line: 
     pr = 1 
     p.stdin.write('DO 1\n') 
     out = p.stdout.readline() 
     continue 

    if "WHATEVER2" in line: 
     pr = 2 
     p.stdin.write('DO 2\n') 
     out = p.stdout.readline() 
     continue 
""" 
.......... 
""" 

out = p.stdout.readline() 

p.wait() 

Answer 7

谨防Popen.communicate(input=s)可能会给你带来麻烦，如果s太大，因为很明显的父进程将缓冲它之前分叉子子，这意味着它需要“两倍”使用的内存在这一点上（至少根据“引擎盖下”的解释和链接文档发现here）。在我的具体情况，s是发电机，其最早是完全展开，然后才写入stdin所以父进程是孩子催生之前巨大的权利， 和无记忆留到餐桌吧：

File "/opt/local/stow/python-2.7.2/lib/python2.7/subprocess.py", line 1130, in _execute_child self.pid = os.fork() OSError: [Errno 12] Cannot allocate memory

Answer 8

我使用python3，并发现你需要你的编码字符串，然后才能将它传递到标准输入：

p = Popen(['grep', 'f'], stdout=PIPE, stdin=PIPE, stderr=PIPE) 
out, err = p.communicate(input='one\ntwo\nthree\nfour\nfive\nsix\n'.encode()) 
print(out) 

Answer 9

我有点惊讶没有人建议设立一个管道，这是在我看来远将字符串传递给stdin的最简单方法一个子进程：

read, write = os.pipe() 
os.write(write, "stdin input here") 
os.close(write) 

subprocess.check_call(['your-command'], stdin=read) 

Answer 10

如果您使用的是Python 3.4或更高版本，那么有一个美丽的解决方案。使用input参数而不是stdin参数，它接受一个字节的说法：

output = subprocess.check_output(
    ["sed", "s/foo/bar/"], 
    input=b"foo", 
)