Làm cách nào để tôi chuyển một chuỗi vào quy trình con.Popen (sử dụng đối số stdin)?

Nếu tôi làm như sau:

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

Tôi có:

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'

Rõ ràng là một đối tượng cStringIO.StringIO không quấy đủ gần với một con vịt tập tin để phù hợp với quy trình con.Popen. Làm thế nào để tôi làm việc xung quanh này?

Popen.communicate() tài liệu:

Lưu ý rằng nếu bạn muốn gửi dữ liệu đến stdin của tiến trình, bạn cần tạo đối tượng Popen với stdin = PIPE. Tương tự, để có được bất cứ thứ gì khác ngoài Không có trong bộ kết quả, bạn cần cung cấp stdout = PIPE và / hoặc stderr = PIPE.

Thay thế os.popen *

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

Cảnh báo Sử dụng giao tiếp () thay vì stdin.write (), stdout.read () hoặc stderr.read () để tránh bế tắc do bất kỳ bộ đệm ống hệ điều hành nào khác lấp đầy và chặn quá trình con.

Vì vậy, ví dụ của bạn có thể được viết như sau:

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

Trên phiên bản Python 3 hiện tại, bạn có thể sử dụng subprocess.run, để chuyển đầu vào dưới dạng chuỗi sang lệnh bên ngoài và nhận trạng thái thoát của nó và đầu ra của chuỗi dưới dạng chuỗi trở lại trong một cuộc gọi:

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

Tôi đã tìm ra cách giải quyết này:

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

Có một cái tốt hơn?

Tôi hơi ngạc nhiên khi không ai đề xuất việc tạo một đường ống, theo ý kiến ​​của tôi là cách đơn giản nhất để truyền một chuỗi cho stdin của một quy trình con:

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

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

Có một giải pháp tuyệt vời nếu bạn đang sử dụng Python 3.4 trở lên. Sử dụng inputđối số thay vì stdinđối số chấp nhận đối số byte:

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

Điều này làm việc cho check_outputvà run, nhưng không callhoặc check_callvì một số lý do.

Tôi đang sử dụng python3 và phát hiện ra rằng bạn cần mã hóa chuỗi của mình trước khi bạn có thể chuyển nó vào 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)

Rõ ràng một đối tượng cStringIO.StringIO không quấy đủ gần với một con vịt tập tin để phù hợp với quy trình con.Popen

Tôi không sợ. Đường ống là một khái niệm hệ điều hành cấp thấp, vì vậy nó hoàn toàn yêu cầu một đối tượng tệp được đại diện bởi một bộ mô tả tệp cấp độ hệ điều hành. Cách giải quyết của bạn là đúng.

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

Coi chừng điều đó Popen.communicate(input=s)có thể gây rắc rối cho bạn nếu squá lớn, vì rõ ràng quy trình cha mẹ sẽ đệm nó trước khi hủy bỏ quy trình con, nghĩa là nó cần bộ nhớ được sử dụng "gấp đôi" tại thời điểm đó (ít nhất là theo giải thích "dưới mui xe" và tài liệu liên kết được tìm thấy ở đây ). Trong trường hợp cụ thể của tôi, slà một trình tạo đầu tiên được mở rộng hoàn toàn và chỉ sau đó được viết cho stdinnên quá trình cha mẹ là rất lớn ngay trước khi đứa trẻ được sinh ra, và không còn bộ nhớ nào để rẽ nhánh:

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)

Trên Python 3.7+ làm điều này:

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

và có lẽ bạn sẽ muốn thêm capture_output=Trueđể có được đầu ra của việc chạy lệnh dưới dạng một chuỗi.

Trên các phiên bản cũ hơn của Python, thay thế text=Truebằng universal_newlines=True:

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