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

Има ли някакъв начин да получа PID по име на процес в Python?

  PID USER      PR  NI  VIRT  RES  SHR S  %CPU %MEM    TIME+  COMMAND                                                                                        
 3110 meysam    20   0  971m 286m  63m S  14.0  7.9  14:24.50 chrome 

Например трябва да получа 3110 от chrome.


person B Faley    schedule 01.11.2014    source източник
comment
Може да има повече процеси с едно и също име и различни PID   -  person Adam Jenča    schedule 01.07.2021


Отговори (10)


Можете да получите 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]

Или подайте флага -s, за да получите един pid:

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

In [25]: get_pid("chrome")
Out[25]: 27698
person Padraic Cunningham    schedule 01.11.2014
comment
+1 изглежда като перфектен отговор. бихте ли обяснили това return check_output(["pidof",name]) - person Avinash Raj; 01.11.2014
comment
@AvinashRaj, добави обяснение, надяваме се да го направи малко по-ясно - person Padraic Cunningham; 01.11.2014
comment
Защо не преминете през /proc записи, вместо да извиквате външни инструменти? Въпреки че е често срещано в bash скриптовете, обикновено не е много чисто да се прави това в скриптовете на python. Освен това, какво ще стане, ако има няколко процеса с това име? Бих поне splitlines() изхода и бих скрил pids в ints. - person ThiefMaster; 01.11.2014
comment
@ThiefMaster, какво не е чисто в използването на командата pidof? - person Padraic Cunningham; 01.11.2014
comment
Дори наличен ли е по подразбиране на всички/повечето Linux системи? - person ThiefMaster; 01.11.2014
comment
@ThiefMaster, доколкото знам, да, така е. Също така използвах split with map за преобразуване в int и обработка на случая ако има множество процеси с това име. Все още не разбирам проблема ви с използването на командата. анализирането на горния изход също ще даде множество идентификатори на процеси - person Padraic Cunningham; 01.11.2014
comment
Имайте предвид, че check_output() не е наличен в Python 2.6 и по-стари версии. - person Jaime M.; 23.02.2016
comment
@PadraicCunningham Какво означава, когато получа CalledProcessError и как да го поправя, за да получа идентификатора на процеса. Опитвам се да направя скрипта и да го стартирам в OpenWRT. - person Vraj Solanki; 22.06.2016
comment
@AvinashRaj Тъй като плакатът на този отговор не е активен, се обръщам към вас. Когато използвам първия кодов фрагмент, той извежда FileNotFoundError. Защо е това? - person Voldemort's Wrath; 19.05.2019
comment
@Voldemort'sWrath, ако стартирате фрагмента на устройство без команда pidof като Windows, командата няма да бъде намерена. Това може да е една от причините. - person Brian Hannay; 10.04.2021

Можете да използвате пакет psutil:

Инсталиране

pip install psutil

Употреба:

import psutil

process_name = "chrome"
pid = None

for proc in psutil.process_iter():
    if process_name in proc.name():
       pid = proc.pid

person rhoitjadhav    schedule 27.11.2019

можете също да използвате 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}'
person Hackaholic    schedule 01.11.2014
comment
Можете да използвате една от помощните функции в модула на подпроцеса... много по-четливо. Също така, изпълнението му в обвивка е наистина лоша идея! - person ThiefMaster; 01.11.2014
comment
да, това е само концепция, която съм дал на OP - person Hackaholic; 01.11.2014
comment
по-добре е да работите с shell=False. - person sobolevn; 11.11.2015
comment
Страхотен отговор! Написах пълен пример въз основа на него тук: stackoverflow.com/a/44712205/304209 - person Dennis Golomazov; 23.06.2017

За posix (Linux, BSD и т.н... трябва само /proc директория да бъде монтирана) е по-лесно да работите с os файлове в /proc. Това е чист Python, няма нужда да извиквате shell програми отвън.

Работи на Python 2 и 3 (Единствената разлика (2to3) е дървото на изключенията, следователно except Exception, което не харесвам, но запазих, за да поддържам съвместимост. Също така можех да създам персонализирано изключение.)

#!/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
person Fernando    schedule 13.08.2015
comment
В семейството на Solaris /proc е двоичен, а не текстов. - person ; 27.11.2019

Пълен пример, базиран на отличния отговор на @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()]
person Dennis Golomazov    schedule 23.06.2017

За да подобрите отговора на 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
person Alejandro Blasco    schedule 11.03.2016
comment
В този случай, тъй като не правим нищо полезно в try/catch, можем също да използваме subprocess.call вместо subprocess.check_output с блока try/catch. - person Rajan Ponnappan; 02.03.2017
comment
@RajanPonnappan С subprocess.call получавате само кода за връщане ($?), но subprocess.check_output връща това, което наистина искате: изхода на командата (в този случай списъка с PID) - person Alejandro Blasco; 03.03.2017
comment
Ти си прав. Обърках се между поведението на check_call и check_output. - person Rajan Ponnappan; 03.03.2017

От 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)
person Gohu    schedule 18.04.2019

Ако вашата операционна система е базирана на 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']
person Ali Hallaji    schedule 30.01.2019

ако използвате Windows, можете да получите PID на процес/приложение с името на изображението с този код:

from subprocess import Popen, PIPE

def get_pid_of_app(app_image_name):
    final_list = []
    command = Popen(['tasklist', '/FI', f'IMAGENAME eq {app_image_name}', '/fo', 'CSV'], stdout=PIPE, shell=False)
    msg = command.communicate()
    output = str(msg[0])
    if 'INFO' not in output:
        output_list = output.split(app_image_name)
        for i in range(1, len(output_list)):
            j = int(output_list[i].replace("\"", '')[1:].split(',')[0])
            if j not in final_list:
                final_list.append(j)

    return final_list

ще ви върне всички PID на приложение като firefox или chrome, напр.

>>> get_pid_of_app("firefox.exe")
[10908, 4324, 1272, 6936, 1412, 2824, 6388, 1884]

уведомете ме дали е помогнало

person Ali Alavizadeh    schedule 01.06.2021

В Unix можете да използвате пакета pyproc2.

Инсталация
pip install pyproc2
Използване
import pyproc2
chrome_pid=pyproc2.find("chrome").pid #Returns PID of first process with name "chrome"
person Adam Jenča    schedule 01.07.2021