Python - Como faço para passar uma string em subprocess.Popen (usando o argumento stdin)?
-
03-07-2019 - |
Pergunta
Se eu faça o seguinte:
import subprocess
from cStringIO import StringIO
subprocess.Popen(['grep','f'],stdout=subprocess.PIPE,stdin=StringIO('one\ntwo\nthree\nfour\nfive\nsix\n')).communicate()[0]
eu recebo:
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'
Aparentemente um objeto cStringIO.StringIO não quack perto o suficiente para um pato arquivo para atender às subprocess.Popen. Como posso resolver isso?
Solução
Popen.communicate()
documentação:
Note que se você quiser enviar dados para stdin do processo, você precisa criar o objeto Popen com stdin = TUBO. Da mesma forma, para conseguir alguma coisa diferente de Nenhum na tupla resultado, você precisa dar stdout = tubo e / ou stderr = TUBO também.
Substituir * os.popen
pipe = os.popen(cmd, 'w', bufsize)
# ==>
pipe = Popen(cmd, shell=True, bufsize=bufsize, stdin=PIPE).stdin
Aviso Use comunicar () em vez de stdin.write (), stdout.read () ou stderr.read () para evitar bloqueios devido para qualquer um dos outros tampões tubos OS encher e bloqueando a criança processo.
Portanto, o seu exemplo poderia ser escrito da seguinte forma:
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
# ->
Na versão atual do Python 3, você poderia usar subprocess.run
, para passar de entrada como uma string para um comando externo e obter seu status de saída, e sua saída como um back corda em uma chamada:
#!/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
# ->
Outras dicas
Eu descobri esta solução alternativa:
>>> 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()
Existe um melhor?
Estou um pouco surpreso ninguém sugeriu a criação de um tubo, que na minha opinião é a maneira muito mais simples para passar uma string para stdin de um subprocesso:
read, write = os.pipe()
os.write(write, "stdin input here")
os.close(write)
subprocess.check_call(['your-command'], stdin=read)
Há uma solução beatiful se você estiver usando o Python 3.4 ou melhor. Use o argumento input
em vez do argumento stdin
, que aceita um argumento de bytes:
output = subprocess.check_output(
["sed", "s/foo/bar/"],
input=b"foo",
)
Eu estou usando python3 e descobri que você precisa para codificar a seqüência antes que você pode passá-lo para 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)
"Aparentemente um objeto cStringIO.StringIO não quack perto o suficiente para um pato arquivo para atender às subprocess.Popen"
: -)
Eu não estou com medo. O tubo é um conceito OS de baixo nível, por isso absolutamente requer um objeto de arquivo que é representado por um descritor de arquivo de nível OS. Sua solução é o caminho certo.
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()
"""
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()
Tenha em atenção que Popen.communicate(input=s)
may dará ifs
is problemas muito grande, porque, aparentemente, o processo pai vai tamponar ele antes bifurcação subprocesso criança, o que significa que precisa de "duas vezes mais" memória usada naquele momento (pelo menos de acordo com o "sob o capô" explicação e documentação relacionada encontrada aqui ). No meu caso particular, s
was um gerador que foi primeiro completamente expandidas e só então escrito tostdin
de modo que o processo pai era enorme direita antes que a criança foi gerado,
e não há memória foi deixado de garfo-lo:
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
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)