Bagaimana cara mengirim string ke subprocess.Popen (menggunakan argumen stdin)?

Jika saya melakukan hal berikut:

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

Saya mendapat:

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'

Rupanya objek cStringIO.StringIO tidak berdetak cukup dekat ke file bebek yang sesuai dengan subprocess.Popen. Bagaimana saya mengatasi ini?

Popen.communicate() dokumentasi:

Perhatikan bahwa jika Anda ingin mengirim data ke stdin proses, Anda harus membuat objek Popen dengan stdin = PIPE. Demikian pula, untuk mendapatkan apa pun selain Tidak Ada dalam tuple hasil, Anda perlu memberikan stdout = PIPE dan / atau stderr = PIPE juga.

Mengganti os.popen *

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

Peringatan Gunakan berkomunikasi () daripada stdin.write (), stdout.read () atau stderr.read () untuk menghindari kebuntuan karena salah satu penyangga pipa OS lainnya mengisi dan memblokir proses anak.

Jadi contoh Anda dapat ditulis sebagai berikut:

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
# ->

Pada versi Python 3 saat ini, Anda bisa menggunakan subprocess.run, untuk meneruskan input sebagai string ke perintah eksternal dan mendapatkan status keluarnya, dan hasilnya sebagai string kembali dalam satu panggilan:

#!/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
# -> 

Saya menemukan solusi ini:

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

Apakah ada yang lebih baik?

Saya agak terkejut tidak ada yang menyarankan membuat pipa, yang menurut saya cara paling sederhana untuk meneruskan string ke stdin dari subproses:

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

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

Ada solusi yang bagus jika Anda menggunakan Python 3.4 atau lebih baik. Gunakan inputargumen alih-alih stdinargumen, yang menerima argumen byte:

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

Ini bekerja untuk check_outputdan run, tetapi tidak callatau check_calluntuk beberapa alasan.

Saya menggunakan python3 dan menemukan bahwa Anda perlu menyandikan string Anda sebelum Anda dapat meneruskannya ke 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)

Rupanya objek cStringIO.StringIO tidak dukun cukup dekat ke file bebek sesuai subprocess.Popen

Aku takut tidak. Pipa adalah konsep OS tingkat rendah, sehingga benar-benar membutuhkan objek file yang diwakili oleh deskriptor file tingkat OS. Solusi Anda adalah yang benar.

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

Berhati-hatilah yang Popen.communicate(input=s)mungkin memberi Anda masalah jika sterlalu besar, karena tampaknya proses induk akan buffer sebelum forking subproses anak, yang berarti perlu "digunakan memori" dua kali lebih banyak pada saat itu (setidaknya sesuai dengan penjelasan "di bawah tenda" dan dokumentasi terkait ditemukan di sini ). Dalam kasus khusus saya, sadalah generator yang pertama kali sepenuhnya diperluas dan baru kemudian ditulis untuk stdinsehingga proses induk sangat besar sebelum anak itu lahir, dan tidak ada memori yang tersisa untuk memotongnya:

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)

Pada Python 3.7+ lakukan ini:

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

dan Anda mungkin ingin menambahkan capture_output=Trueuntuk mendapatkan output dari menjalankan perintah sebagai string.

Pada versi Python yang lebih lama, ganti text=Truedengan universal_newlines=True:

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