Jak używać polecenia `subprocess` z potokami

Chcę używać subprocess.check_output()z ps -A | grep 'process_name'. Próbowałem różnych rozwiązań, ale jak dotąd nic nie działało. Czy ktoś może mi pomóc, jak to zrobić?

Aby użyć rury z subprocessmodułem, musisz przejść shell=True.

Nie jest to jednak zalecane z różnych powodów, między innymi z uwagi na bezpieczeństwo. Zamiast utworzyć psi grepprocesami oddzielnie, wyjście z rury jedna w drugą, tak jak poniżej:

ps = subprocess.Popen(('ps', '-A'), stdout=subprocess.PIPE)
output = subprocess.check_output(('grep', 'process_name'), stdin=ps.stdout)
ps.wait()

Jednak w twoim konkretnym przypadku prostym rozwiązaniem jest wywołanie, subprocess.check_output(('ps', '-A'))a następnie str.findwyjście.

Lub zawsze możesz użyć metody komunikacji na obiektach podprocesu.

cmd = "ps -A|grep 'process_name'"
ps = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
output = ps.communicate()[0]
print(output)

Metoda komunikacji zwraca krotkę standardowego wyjścia i standardowy błąd.

Zobacz dokumentację dotyczącą konfigurowania potoku za pomocą podprocesu: http://docs.python.org/2/library/subprocess.html#replacing-shell-pipeline

Nie testowałem następującego przykładu kodu, ale powinien on być mniej więcej taki, jak chcesz:

query = "process_name"
ps_process = Popen(["ps", "-A"], stdout=PIPE)
grep_process = Popen(["grep", query], stdin=ps_process.stdout, stdout=PIPE)
ps_process.stdout.close()  # Allow ps_process to receive a SIGPIPE if grep_process exits.
output = grep_process.communicate()[0]

Rozwiązanie JKALAVIS jest dobre, ale dodałbym ulepszenie do używania shlex zamiast SHELL = TRUE. poniżej przedstawiam czasy zapytania

#!/bin/python
import subprocess
import shlex

cmd = "dig @8.8.4.4 +notcp www.google.com|grep 'Query'"
ps = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
output = ps.communicate()[0]
print(output)

Spróbuj także użyć 'pgrep'polecenia zamiast'ps -A | grep 'process_name'

Możesz wypróbować funkcjonalność potoku w sh.py :

import sh
print sh.grep(sh.ps("-ax"), "process_name")

Po Python 3.5 możesz także używać:

    import subprocess

    f = open('test.txt', 'w')
    process = subprocess.run(['ls', '-la'], stdout=subprocess.PIPE, universal_newlines=True)
    f.write(process.stdout)
    f.close()

Wykonanie polecenia blokuje, a dane wyjściowe będą przetwarzane . Proces .