ฉันจะส่งผ่านสตริงไปยัง subprocess.Popen (โดยใช้อาร์กิวเมนต์ 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 นั้นไม่ได้อยู่ใกล้กับเป็ดพอที่จะให้เหมาะสมกับ subprocess.Popen ฉันจะแก้ไขสิ่งนี้ได้อย่างไร

Popen.communicate() เอกสาร:

โปรดทราบว่าถ้าคุณต้องการส่งข้อมูลไปยัง stdin ของกระบวนการคุณต้องสร้างวัตถุ Popen ด้วย stdin = PIPE ในการรับสิ่งอื่นที่ไม่ใช่ None ในผลลัพธ์ tuple คุณต้องให้ 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 () เพื่อหลีกเลี่ยงการหยุดชะงักเนื่องจากบัฟเฟอร์ท่อระบบปฏิบัติการอื่นใดที่เติมและบล็อกกระบวนการลูก

ดังนั้นตัวอย่างของคุณสามารถเขียนได้ดังนี้:

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

ฉันพบวิธีแก้ปัญหานี้:

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

มีดีกว่าไหม

ฉันประหลาดใจเล็กน้อยที่ไม่มีใครแนะนำให้สร้างไพพ์ซึ่งฉันคิดว่าเป็นวิธีที่ง่ายที่สุดในการส่งสตริงไปยัง stdin ของกระบวนการย่อย:

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

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

มีวิธีแก้ปัญหาที่สวยงามถ้าคุณใช้ Python 3.4 หรือดีกว่า ใช้inputอาร์กิวเมนต์แทนstdinอาร์กิวเมนต์ซึ่งยอมรับอาร์กิวเมนต์ไบต์:

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

มันใช้งานได้check_outputและrunแต่ไม่ใช่callหรือcheck_callด้วยเหตุผลบางอย่าง

ฉันใช้ 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)

เห็นได้ชัดว่าวัตถุ cStringIO.StringIO นั้นไม่ได้อยู่ใกล้กับเป็ดพอที่จะให้เหมาะสมกับกระบวนการย่อยเปิด

ฉันเกรงว่าจะไม่. ไพพ์เป็นแนวคิดระบบปฏิบัติการระดับต่ำดังนั้นจึงต้องการวัตถุไฟล์ที่แสดงโดยตัวอธิบายไฟล์ระดับระบบปฏิบัติการ วิธีแก้ปัญหาของคุณคือสิ่งที่ถูกต้อง

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

ระวังที่Popen.communicate(input=s)อาจทำให้คุณมีปัญหาหากsมีขนาดใหญ่เกินไปเพราะเห็นได้ชัดว่ากระบวนการหลักจะบัฟเฟอร์มันก่อนฟอร์ก subprocess เด็กซึ่งหมายความว่าต้องมีหน่วยความจำที่ใช้ "สองเท่า" ที่จุดนั้น (อย่างน้อยตามคำอธิบาย "ภายใต้ประทุน") และเอกสารที่เชื่อมโยงพบได้ที่นี่ ) ในกรณีเฉพาะของฉัน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

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

บน Python 3.7+ ทำสิ่งนี้:

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

และคุณอาจต้องการเพิ่ม capture_output=Trueเพื่อให้ได้ผลลัพธ์ของการรันคำสั่งเป็นสตริง

ใน Python เวอร์ชันเก่าให้แทนที่text=Trueด้วยuniversal_newlines=True:

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