Как запустить «ps cax | grep что-то» в Python? - PullRequest
25 голосов
/ 21 июля 2011

Как запустить команду с каналом | в нем?

Модуль подпроцесса кажется сложным ...

Есть ли что-то вроде

output,error = `ps cax | grep something`

как в сценарии оболочки?

Ответы [ 5 ]

45 голосов
/ 21 июля 2011

См. Замена оболочки конвейера :

import subprocess

proc1 = subprocess.Popen(['ps', 'cax'], stdout=subprocess.PIPE)
proc2 = subprocess.Popen(['grep', 'python'], stdin=proc1.stdout,
                         stdout=subprocess.PIPE, stderr=subprocess.PIPE)

proc1.stdout.close() # Allow proc1 to receive a SIGPIPE if proc2 exits.
out, err = proc2.communicate()
print('out: {0}'.format(out))
print('err: {0}'.format(err))

PS.Использование shell=True может быть опасным.См., Например, предупреждение в документации.


Существует также модуль sh , который может сделать скрипты подпроцессов в Python намного более приятными:

import sh
print(sh.grep(sh.ps("cax"), 'something'))
17 голосов
/ 21 июля 2011

Вы уже приняли ответ, но:

Вам действительно нужно , чтобы использовать grep? Я бы написал что-то вроде:

import subprocess
ps = subprocess.Popen(('ps', 'cax'), stdout=subprocess.PIPE)
output = ps.communicate()[0]
for line in output.split('\n'):
    if 'something' in line:
        ...

Это имеет то преимущество, что не включает shell=True и его рискованность, не выделяет отдельный процесс grep и выглядит очень похоже на Python, который вы пишете для обработки объектов, подобных файлу данных.

8 голосов
/ 21 июля 2011
import subprocess

process = subprocess.Popen("ps cax | grep something",
                             shell=True,
                             stdout=subprocess.PIPE,
                           )
stdout_list = process.communicate()[0].split('\n')
7 голосов
/ 22 июля 2011

Отбросьте этот подпроцесс 'ps' и медленно возвращайтесь! :)

Вместо этого используйте модуль psutil .

3 голосов
/ 21 июля 2011
import os

os.system('ps -cax|grep something')

Если вы хотите заменить аргумент grep какой-либо переменной:

os.system('ps -cax|grep '+your_var)
...