python import os linux

¿Cómo obtener el PID por nombre de proceso en Python?

Question

¿Hay alguna forma de que pueda obtener el PID por nombre de proceso en 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

Por ejemplo, necesito obtener 3110 con chrome .

Answer 1

Ejemplo completo basado en la excelente answer @ 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()]

Answer 2

Para mejorar la respuesta de Padraic: cuando check_output devuelve un código distinto de cero, genera un CalledProcessError. Esto sucede cuando el proceso no existe o no se está ejecutando.

Lo que haría para detectar esta excepción es:

#!/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")

La salida:

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

Answer 3

Para posix (Linux, BSD, etc ... solo se necesita el directorio / proc para ser montado) es más fácil trabajar con archivos os en / proc. Es puro pitón, no es necesario llamar a los programas de shell fuera.

Funciona en python 2 y 3 (La única diferencia (2to3) es el árbol de excepciones, por lo tanto, la excepción de excepción , que no me gusta pero mantiene para mantener la compatibilidad. También podría haber creado una excepción personalizada.

#!/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)))

Muestra de salida (funciona como pgrep):

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

Answer 4

Puede obtener el pid de procesos por nombre usando pidof través de 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]) ejecutará el comando como "pidof process_name" . Si el código de retorno no es cero, genera un CalledProcessError.

Para manejar entradas múltiples y lanzar a Ints:

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

En [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]

O pase la bandera -s para obtener un solo pid:

def get_pid(name):
return int(check_output(["pidof","-s",name]))
In [25]: get_pid("chrome")
Out[25]: 27698

Answer 5

también puedes usar pgrep , en prgep también puedes dar el patrón para el partido

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

también puedes usar awk con ps como este

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