Python - Как мне передать строку в подпроцесс.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 недостаточно близко подходит к утке файла, чтобы удовлетворить подпроцесс.Popen.Как мне обойти это?

Answer 1

Popen.communicate () документация:

  

Обратите внимание, что если вы хотите отправить данные в   процесс, вам нужно   создать объект Popen с   STDIN = PIPE. Точно так же, чтобы получить что-нибудь   кроме None в кортеже результата,   вам нужно дать стандартный вывод = ТРУБА и / или   stderr = ТРУБА тоже.

     

Замена os.popen *

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

  

Предупреждение . Используйте функцию connect () вместо   stdin.write (), stdout.read () или   stderr.read (), чтобы избежать взаимных блокировок   к любому из других буферов OS 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 вы можете использовать подпроцесс .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

Я немного удивлен, что никто не предложил создать канал, который, на мой взгляд, является самым простым способом передачи строки в стандартный поток подпроцесса:

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

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

Answer 4

Есть прекрасное решение, если вы используете Python 3.4 или выше. Используйте аргумент input вместо аргумента stdin , который принимает аргумент в байтах:

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

Answer 5

Я использую 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 6

"Очевидно, cStringIO.Объект StringIO недостаточно близко подходит к утке файла, чтобы удовлетворить подпроцесс.Popen"

:-)

Боюсь, что нет.Канал - это концепция низкоуровневой операционной системы, поэтому для него абсолютно необходим файловый объект, представленный файловым дескриптором уровня операционной системы.Ваш обходной путь - правильный.

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

Answer 8

"""
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 9

Имейте в виду, что Popen.communicate (input = s) может доставить вам неприятности, если s слишком велик, поскольку, очевидно, родительский процесс будет буферизовать его перед разветвление дочернего подпроцесса, что означает, что ему нужно "вдвое больше" в этот момент использовалась память (по крайней мере, в соответствии с пояснением «под капотом» и соответствующей документацией, найденной здесь ). В моем конкретном случае, s был генератором, который сначала был полностью раскрыт и только потом записан в stdin , поэтому родительский процесс был огромным прямо перед порождением потомка, и не осталось памяти, чтобы его разветвлять:

Файл " /opt/local/stow/python-2.7.2/lib/python2.7/subprocess.py" ;, строка 1130, в _execute_child     self.pid = os.fork () Ошибка OSE: [Errno 12] Невозможно выделить память

Answer 10

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)