Python - 如何将字符串传递给subprocess.Popen(使用stdin参数)?

时间:2008-10-02 17:25:23

标签: python subprocess stdin

如果我执行以下操作:

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对象没有足够接近文件duck以适应subprocess.Popen。我该如何解决这个问题?

12 个答案:

答案 0 :(得分:297)

Popen.communicate()文档:

  

请注意,如果要将数据发送到   你需要的是进程的标准输入   用它创建Popen对象   标准输入=管道。同样,得到任何东西   除了结果元组中的None之外,   你需要给stdout = PIPE和/或   stderr = PIPE。

     

取代os.popen *

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

警告使用communic()而不是   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
# -> 

答案 1 :(得分:42)

我想出了这个解决方法:

>>> 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()

有更好的吗?

答案 2 :(得分:23)

我有点惊讶没有人建议创建一个管道,在我看来,这是将字符串传递给子进程的标准输入的最简单方法:

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

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

答案 3 :(得分:18)

如果您使用的是Python 3.4或更高版本,那么这是一个美丽的解决方案。使用input参数而不是stdin参数,该参数接受字节参数:

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

答案 4 :(得分:14)

我正在使用python3并发现你需要对你的字符串进行编码才能将它传递给stdin:

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)

答案 5 :(得分:13)

“显然cStringIO.StringIO对象没有足够接近文件duck以适应subprocess.Popen”

: - )

我不敢。管道是一个低级操作系统概念,因此它绝对需要一个由操作系统级文件描述符表示的文件对象。你的解决方法是正确的。

答案 6 :(得分:7)

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()

答案 7 :(得分: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()

答案 8 :(得分:5)

请注意,如果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

答案 9 :(得分:3)

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)

答案 10 :(得分:0)

在Python 3.7+上执行以下操作:

UITapGestureRecognizer

,您可能需要添加my_data = "whatever you want\nshould match this f" subprocess.run(["grep", "f"], text=True, input=my_data) 以获得作为字符串运行命令的输出。

在旧版本的Python上,将capture_output=True替换为text=True

universal_newlines=True

答案 11 :(得分:0)

这对 grep 来说太过分了,但通过我的旅程,我了解了 Linux 命令 expect 和 python 库 pexpect

  • expect:与互动程序对话
  • pexpect:用于生成子应用程序的 Python 模块;控制他们;并对其输出中的预期模式做出响应。
import pexpect
child = pexpect.spawn('grep f', timeout=10)
child.sendline('text to match')
print(child.before)

使用 pexpect

可以轻松处理 ftp 等交互式 shell 应用程序
import pexpect
child = pexpect.spawn ('ftp ftp.openbsd.org')
child.expect ('Name .*: ')
child.sendline ('anonymous')
child.expect ('Password:')
child.sendline ('noah@example.com')
child.expect ('ftp> ')
child.sendline ('ls /pub/OpenBSD/')
child.expect ('ftp> ')
print child.before   # Print the result of the ls command.
child.interact()     # Give control of the child to the user.