如何通过Python中的进程名称获得PID?

有没有什么办法可以通过Python中的进程名称获得PID?

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 

例如,我需要通过chrome获得3110

您可以通过pidof使用pidof来获取进程的pid:

 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。

处理多个条目并转换为整数:

 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 

你也可以使用pgrep ,在prgep也可以给出匹配模式

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

你也可以像这样用ps来使用awk

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

对于POSIX(Linux,BSD等…只需要安装/ proc目录),在/ proc中使用os文件更容易。 它的纯python,不需要在外面调用shell程序。

工作在Python 2和3(唯一的区别(2to3)是exception树,因此“ 除了exception ”,我不喜欢,但保持兼容性,也可以创build自定义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 

为了改进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 

基于优秀的@ 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()]