Jak przekazać ciąg znaków do podprocesu.Popen (używając argumentu stdin)?

Wymyśliłem to obejście:

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

Jestem trochę zaskoczony, że nikt nie zasugerował stworzenia rury, co jest moim zdaniem najprostszym sposobem na przekazanie łańcucha na stdin podprocesu:

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

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

Jest piękne rozwiązanie, jeśli używasz Pythona 3.4 lub lepszego. Użyj argumentu input zamiast argumentu stdin, który akceptuje argument bajtów:

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

Używam python3 i dowiedziałem się, że musisz zakodować swój ciąg znaków, zanim będziesz mógł go przekazać na 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)

Najwyraźniej cStringIO.Obiekt StringIO nie kwaknie wystarczająco blisko kaczka plików do podprocesu.Popen

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

Uważaj, że Popen.communicate(input=s) może przysporzyć ci kłopotów, jeślisjest zbyt duży, ponieważ najwyraźniej proces rodzic buforuje go przed rozwidleniem podprocesu potomnego, co oznacza, że potrzebuje "dwa razy więcej" używanej pamięci w tym momencie (przynajmniej zgodnie z wyjaśnieniem "pod maską" i powiązaną dokumentacją znalezioną tutaj). W moim konkretnym przypadku, sbył generatorem, który najpierw został w pełni rozbudowany, a dopiero potem napisany dostdin, więc proces rodzica był ogromny tuż przed narodzeniem dziecka, a pamięć nie została, żeby ją rozwidlić:

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

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

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)

W Pythonie 3.7 + zrób to:

my_data = "whatever you want\nshould match this f"
subprocess.run(["grep", "f"], text=True, input=my_data)

I prawdopodobnie będziesz chciał dodać capture_output=True, aby uzyskać wynik uruchomienia polecenia jako ciąg znaków.

W starszych wersjach Pythona zamień text=True na universal_newlines=True:

subprocess.run(["grep", "f"], universal_newlines=True, input=my_data)