Cómo obtener la salida del comando externo combinar con Pipe
Pregunta
Tengo un comando como este.
wmctrl -lp | awk '/gedit/ { print $1 }'
Y quiero su salida dentro del script de Python, probé este código
>>> import subprocess
>>> proc = subprocess.Popen(["wmctrl -lp", "|","awk '/gedit/ {print $1}"], shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
>>> proc.stdout.readline()
'0x0160001b -1 6504 beer-laptop x-nautilus-desktop\n'
>>> proc.stdout.readline()
'0x0352f117 0 6963 beer-laptop How to get output from external command combine with Pipe - Stack Overflow - Chromium\n'
>>> proc.stdout.readline()
'0x01400003 -1 6503 beer-laptop Bottom Expanded Edge Panel\n'
>>>
Parece que mi código es incorrecto, solo se ejecutó wmctrl -lp
y | awk '{print $ 1}'
se omite
Mi resultado esperado sería 0x03800081
$ wmctrl -lp | awk '/gedit/ {print $1}'
0x03800081
¿Alguien ayuda?
Solución
Con shell = True
, debe usar una sola línea de comando en lugar de una matriz, de lo contrario, sus argumentos adicionales se interpretarán como argumentos de shell. Del subprocess
documentación :
En Unix, con shell = True: si args es una cadena, especifica la cadena de comando que se ejecutará a través del shell. Si args es una secuencia, el primer elemento especifica la cadena de comando y cualquier elemento adicional se tratará como argumentos de shell adicionales.
Entonces su llamada debería ser:
subprocess.Popen("wmctrl -lp | sed /gedit/ '{print $1}'", shell=True, ...
Creo que también puede tener una cita única desequilibrada allí.
Otros consejos
Debido a que está pasando una secuencia para el programa, cree que la tubería es un argumento para wmcrtrl
, como si lo hiciera
wmctrl -lp "|"
y, por lo tanto, se pierde la operación real de tubería .
Convertirlo en una sola cadena debería darte el resultado correcto:
>>> import subprocess as s
>>> proc = s.Popen("echo hello | grep e", shell=True, stdout=s.PIPE, stderr=s.PIPE)
>>> proc.stdout.readline()
'hello\n'
>>> proc.stdout.readline()
''
Después de investigar un poco, tengo el siguiente código que funciona muy bien para mí. Básicamente imprime stdout y stderr en tiempo real. Espero que ayude a alguien más que lo necesita.
stdout_result = 1
stderr_result = 1
def stdout_thread(pipe):
global stdout_result
while True:
out = pipe.stdout.read(1)
stdout_result = pipe.poll()
if out == '' and stdout_result is not None:
break
if out != '':
sys.stdout.write(out)
sys.stdout.flush()
def stderr_thread(pipe):
global stderr_result
while True:
err = pipe.stderr.read(1)
stderr_result = pipe.poll()
if err == '' and stderr_result is not None:
break
if err != '':
sys.stdout.write(err)
sys.stdout.flush()
def exec_command(command, cwd=None):
if cwd is not None:
print '[' + ' '.join(command) + '] in ' + cwd
else:
print '[' + ' '.join(command) + ']'
p = subprocess.Popen(
command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd
)
out_thread = threading.Thread(name='stdout_thread', target=stdout_thread, args=(p,))
err_thread = threading.Thread(name='stderr_thread', target=stderr_thread, args=(p,))
err_thread.start()
out_thread.start()
out_thread.join()
err_thread.join()
return stdout_result + stderr_result
Cuando es necesario, creo que es fácil recopilar el resultado o error en una cadena y devolverlo.