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}''