¿Hay alguna manera de 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
por chrome
.
Puede obtener el pid de procesos por nombre utilizando 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 múltiples entradas y convertir 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
Para posix (Linux, BSD, etc … solo es necesario montar el directorio / proc) es más fácil trabajar con archivos OS en / proc. Es puro python, no hay necesidad de llamar progtwigs shell fuera.
Funciona en python 2 y 3 (la única diferencia (2 a 3) es el árbol de Excepciones, por lo tanto, la ” excepción de excepción “, que no me gusta pero que mantengo 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)))
Salida de muestra (funciona como pgrep):
phoemur ~/python $ ./pgrep.py bash 1487 : -bash 1779 : /bin/bash
también puede usar pgrep
, en prgep
también puede dar un patrón para emparejar
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}'
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 atrapar 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
Ejemplo completo basado en la excelente respuesta de @ 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()]
Si su sistema operativo es base Unix use este código:
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
Luego llámelo y pásele un nombre de proceso para obtener todos los PID.
>>> check_process('firefox') ['499', '621', '623', '630', '11733']