Comment obtenir PID par nom de processus en Python?
y a-t-il un moyen d'obtenir le PID par nom de processus 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
par exemple, je dois obtenir 3110 par chrome .
5 réponses
vous pouvez obtenir le pid des processus par leur nom en utilisant pidof par sous-processus.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]) exécutera la commande comme "pidof process_name" , si le code de retour était non-zéro il soulève un Processesserror appelé.
pour traiter les entrées multiples et moulé à l'envers:
from subprocess import check_output
def get_pid(name):
return map(int,check_output(["pidof",name]).split())
In [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]
ou pas le drapeau -s pour obtenir un seul pid:
def get_pid(name):
return int(check_output(["pidof","-s",name]))
In [25]: get_pid("chrome")
Out[25]: 27698
vous pouvez aussi utiliser pgrep , dans prgep vous pouvez aussi donner le motif pour match
import subprocess
child = subprocess.Popen(['pgrep','program_name'], stdout=subprocess.PIPE, shell=True)
result = child.communicate()[0]
vous pouvez également utiliser awk avec ps comme ceci
ps aux | awk '/name/{print }'
pour posix (Linux, BSD, etc... il est plus facile de travailler avec des fichiers os dans /proc. Son python pur, pas besoin d'appeler les programmes shell à l'extérieur.
fonctionne sur python 2 et 3 ( la seule différence (2to3) est L'arbre D'Exception, donc le " sauf Exception ", que je n'aime pas mais que j'ai gardé pour maintenir la compatibilité. J'aurais aussi pu créer une exception personnalisée.)
#!/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)))
Sortie D'Échantillon (cela fonctionne comme pgrep):
phoemur ~/python $ ./pgrep.py bash
1487 : -bash
1779 : /bin/bash
pour améliorer la réponse du Padraic: quand check_output renvoie un code non-zéro, il soulève un Processerror appelé. Cela se produit lorsque le processus n'existe pas ou ne fonctionne pas.
ce que je ferais pour attraper cette exception est:
#!/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 sortie:
$ python pidproc.py
list of PIDS = 31840, 31841, 41942
exemple complet basé sur l'excellente réponse de @Hackaholic 151930920":
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()]