[python] 서브 프로세스에 문자열을 어떻게 전달합니까? (stdin 인수 사용)

PYTHON / 글쓴이

내가 다음을 수행하면 :

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.communicate() 선적 서류 비치:

프로세스의 stdin으로 데이터를 보내려면 stdin = PIPE를 사용하여 Popen 오브젝트를 작성해야합니다. 마찬가지로 결과 튜플에서 None 이외의 값을 얻으려면 stdout = PIPE 및 / 또는 stderr = PIPE도 제공해야합니다.

os.popen * 교체

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

경고 자식 프로세스를 채우고 차단하는 다른 OS 파이프 버퍼로 인한 교착 상태를 피하려면 stdin.write (), stdout.read () 또는 stderr.read () 대신 communi ()을 사용하십시오.

따라서 다음과 같이 예제를 작성할 수 있습니다.

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 이상을 사용하는 경우 아름다운 해결책이 있습니다. bytes 인수를 허용하는 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 객체는 하위 프로세스에 적합하도록 파일 덕에 충분히 근접하지 않습니다.

난 두려워하지. 파이프는 저수준 OS 개념이므로 OS 수준 파일 설명 자로 표시되는 파일 개체가 절대적으로 필요합니다. 해결 방법이 맞습니다.

답변

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