如何通过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()]