如何通过进程名获取PID？

Posted 2023-02-23

【中文标题】如何通过进程名获取PID？

【英文标题】：How to get PID by process name?

【发布时间】：2014-12-28 14:46:21

【问题描述】：

有什么方法可以在 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。

【问题讨论】：

可以有多个同名不同PID的进程

【参考方案1】：

在 Unix 上，您可以使用 pyproc2 包。

安装

pip install pyproc2

用法

import pyproc2
chrome_pid=pyproc2.find("chrome").pid #Returns PID of first process with name "chrome"

【参考方案2】：

如果您使用的是 Windows， 您可以使用以下代码获取进程/应用程序的 PID 及其图像名称：

from subprocess import Popen, PIPE

def get_pid_of_app(app_image_name):
    final_list = []
    command = Popen(['tasklist', '/FI', f'IMAGENAME eq app_image_name', '/fo', 'CSV'], stdout=PIPE, shell=False)
    msg = command.communicate()
    output = str(msg[0])
    if 'INFO' not in output:
        output_list = output.split(app_image_name)
        for i in range(1, len(output_list)):
            j = int(output_list[i].replace("\"", '')[1:].split(',')[0])
            if j not in final_list:
                final_list.append(j)

    return final_list

它将返回您应用程序的所有 PID，例如 firefox 或 chrome，例如

>>> get_pid_of_app("firefox.exe")
[10908, 4324, 1272, 6936, 1412, 2824, 6388, 1884]

如果有帮助请告诉我

【参考方案3】：

对于 posix（Linux、BSD 等...只需要挂载 /proc 目录），使用 /proc 中的 os 文件更容易。 纯python，不需要在外面调用shell程序。

适用于 python 2 和 3（唯一的区别 (2to3) 是异常树，因此是“except 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

【讨论】：

在 Solaris 系列上 /proc 是二进制的，而不是文本的。

【参考方案4】：

你可以使用psutil包：

安装

pip install psutil

用法：

import psutil

process_name = "chrome"
pid = None

for proc in psutil.process_iter():
    if process_name in proc.name():
       pid = proc.pid

【参考方案5】：

从 Python 3.5 开始，推荐使用 subprocess.run() 而非 subprocess.check_output()：

>>> int(subprocess.run(["pidof", "-s", "your_process"], stdout=subprocess.PIPE).stdout)

另外，从 Python 3.7 开始，您可以使用capture_output=true 参数来捕获标准输出和标准错误：

>>> int(subprocess.run(["pidof", "-s", "your process"], capture_output=True).stdout)

【参考方案6】：

如果您的操作系统是基于 Unix 的，请使用以下代码：

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

然后调用它并传递一个进程名称以获取所有 PID。

>>> check_process('firefox')
['499', '621', '623', '630', '11733']

【参考方案7】：

基于优秀的@Hackaholic 的answer 的完整示例：

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()]

【参考方案8】：

您可以使用pidof 到subprocess.check_output 按名称获取进程的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

【讨论】：

+1 似乎是一个完美的答案。你能解释一下吗return check_output(["pidof",name])

为什么不通过 /proc 条目而不是调用外部工具？虽然在 bash 脚本中很常见，但在 python 脚本中这样做通常不是很干净。此外，如果有多个具有该名称的进程怎么办？我至少会 splitlines() 输出并将 pid 转换为整数。

@ThiefMaster，使用 pidof 命令有什么不干净的地方？

@JaimeM.，实现gist.github.com/edufelipe/1027906

非常简单

@AvinashRaj 由于此答案的发布者尚未活跃，因此我求助于您。当我使用第一个 sn-p 代码时，它会抛出一个FileNotFoundError。这是为什么呢？

【参考方案9】：

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

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

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

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

【讨论】：

您可以使用 subprocess 模块中的辅助函数之一.. 更具可读性。此外，在 shell 中执行它是一个非常糟糕的主意！

是的，它只是一个概念，我已经给了 OP

最好用shell=False运行。

很好的答案！我在这里写了一个基于它的完整示例：***.com/a/44712205/304209

【参考方案10】：

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

【讨论】：

在这种情况下，由于我们不会在 try/catch 中进行任何有用的操作，因此我们也可以在 try/catch 块中使用 subprocess.call 而不是 subprocess.check_output。

@RajanPonnappan 使用subprocess.call，您只能获得返回码（$？），但subprocess.check_output 返回您真正想要的：命令输出（在本例中为 PID 列表）跨度>

你是对的。我对check_call 和check_output 的行为感到困惑。