Как получить PID по имени процесса?

Ответ 1

Вы можете получить pid процессов по имени, используя pidof через subprocess.check_output:

from subprocess import check_output
def get_pid(name):
    return check_output(["pidof",name])


In [5]: get_pid("java")
Out[5]: '23366\n'

check_output(["pidof",name]) выполнит команду как "pidof process_name", если код возврата был ненулевым, он вызывает CalledProcessError.

Для обработки нескольких записей и их перевода в ints:

from subprocess import check_output
def get_pid(name):
    return map(int,check_output(["pidof",name]).split())

В [21]: get_pid ( "chrome" )

Out[21]: 
[27698, 27678, 27665, 27649, 27540, 27530, 27517, 14884, 14719, 13849, 13708, 7713, 7310, 7291, 7217, 7208, 7204, 7189, 7180, 7175, 7166, 7151, 7138, 7127, 7117, 7114, 7107, 7095, 7091, 7087, 7083, 7073, 7065, 7056, 7048, 7028, 7011, 6997]

Или pas флаг -s для получения одного pid:

def get_pid(name):
    return int(check_output(["pidof","-s",name]))

In [25]: get_pid("chrome")
Out[25]: 27698

Ответ 2

вы также можете использовать pgrep, в prgep вы также можете указать шаблон для соответствия

import subprocess
child = subprocess.Popen(['pgrep','program_name'], stdout=subprocess.PIPE, shell=True)
result = child.communicate()[0]

вы также можете использовать awk с ps как

ps aux | awk '/name/{print $2}'

Ответ 3

Для posix (Linux, BSD и т.д.) нужен только каталог /proc для сборки), проще работать с os файлами в /proc. Его чистый питон, не нужно вызывать программы оболочки снаружи.

Работает на python 2 и 3 (Единственное различие (2to3) - это дерево исключений, поэтому "исключение исключение", которое мне не нравится, но поддерживается для поддержания совместимости. Также возможно создать настраиваемое исключение.)

#!/usr/bin/env python

import os
import sys


for dirname in os.listdir('/proc'):
    if dirname == 'curproc':
        continue

    try:
        with open('/proc/{}/cmdline'.format(dirname), mode='rb') as fd:
            content = fd.read().decode().split('\x00')
    except Exception:
        continue

    for i in sys.argv[1:]:
        if i in content[0]:
            print('{0:<12} : {1}'.format(dirname, ' '.join(content)))

Пример вывода (он работает как pgrep):

phoemur ~/python $ ./pgrep.py bash
1487         : -bash 
1779         : /bin/bash

Ответ 4

Чтобы улучшить ответ Padraic: когда check_output возвращает ненулевой код, он вызывает CalledProcessError. Это происходит, когда процесс не существует или не работает.

Что бы я сделал, чтобы поймать это исключение:

#!/usr/bin/python

from subprocess import check_output, CalledProcessError

def getPIDs(process):
    try:
        pidlist = map(int, check_output(["pidof", process]).split())
    except  CalledProcessError:
        pidlist = []
    print 'list of PIDs = ' + ', '.join(str(e) for e in pidlist)

if __name__ == '__main__':
    getPIDs("chrome")

Выход:

$ python pidproc.py
list of PIDS = 31840, 31841, 41942

Ответ 5

Полный пример, основанный на отличном ответе @Hackaholic :

def get_process_id(name):
    """Return process ids found by (partial) name or regex.

    >>> get_process_id('kthreadd')
    [2]
    >>> get_process_id('watchdog')
    [10, 11, 16, 21, 26, 31, 36, 41, 46, 51, 56, 61]  # ymmv
    >>> get_process_id('non-existent process')
    []
    """
    child = subprocess.Popen(['pgrep', '-f', name], stdout=subprocess.PIPE, shell=False)
    response = child.communicate()[0]
    return [int(pid) for pid in response.split()]

Ответ 6

Если ваша ОС на базе Unix, используйте этот код:

import os
def check_process(name):
    output = []
    cmd = "ps -aef | grep -i '%s' | grep -v 'grep' | awk '{ print $2 }' > /tmp/out"
    os.system(cmd % name)
    with open('/tmp/out', 'r') as f:
        line = f.readline()
        while line:
            output.append(line.strip())
            line = f.readline()
            if line.strip():
                output.append(line.strip())

    return output

Затем вызовите его и передайте имя процесса, чтобы получить все PID.

>>> check_process('firefox')
['499', '621', '623', '630', '11733']

Ответ 7

Начиная с Python 3.5, subprocess.run() рекомендуется вместо subprocess.check_output():

>>> int(subprocess.run(["pidof", "-s", "your_process"], stdout=subprocess.PIPE).stdout)

Кроме того, начиная с Python 3.7, вы можете использовать параметр capture_output=true для захвата stdout и stderr:

>>> int(subprocess.run(["pidof", "-s", "your process"], capture_output=True).stdout)