使用 Python 根据进程名获取进程根目录

在这一篇文章中，我们将学习如何使用 Python 根据指定的进程名获取进程的根目录。这个操作在处理系统进程相关任务时非常有用。以下是我们将要进行的步骤及其解释。

流程步骤

步骤 描述
1 安装所需的 Python 库
2 导入所需的库
3 获取所有进程信息
4 根据进程名过滤进程
5 打印出找到进程的根目录

每一步详解

第一步：安装所需的 Python 库

在开始编写代码之前，我们需要确保安装了 psutil 库，这个库用于获取系统和进程信息。可以用以下命令进行安装：

pip install psutil

将 psutil 库安装到我们的 Python 环境中。

第二步：导入所需的库

在我们的 Python 文件中，先导入 psutil 库，这样我们就可以使用它提供的功能来获取进程信息。

import psutil

导入 psutil 库，以便后面使用它获取进程信息。

第三步：获取所有进程信息

我们可以使用 psutil.process_iter() 函数来迭代当前系统中的所有进程。在此过程中，我们可以收集进程的进程名和进程 ID。

# 获取所有进程信息，使用 generator 遍历每一个进程
processes = psutil.process_iter(['pid', 'name', 'exe'])

这里，我们调用 process_iter 此方法并传入参数来筛选我们需要的字段：进程ID、名称和执行路径。

第四步：根据进程名过滤进程

接下来，我们需要定义我们希望查找的进程名，并将其与获取的所有进程信息进行比较。

# 设定要查找的进程名
process_name = "your_process_name.exe"  # 请修改为你希望查找的进程名

# 遍历所有进程，查找匹配的进程
for proc in processes:
    if proc.info['name'] == process_name:
        print(f"进程ID: {proc.info['pid']}, 根目录: {proc.info['exe']}")

将 your_process_name.exe 替换为你希望查询的实际进程名。若找到该进程，输出进程ID和根目录。

第五步：打印出找到进程的根目录

在上面我们已通过 print 输出找到的进程的根目录。在实际应用中也可以将返回的信息存储在变量中，供后续操作使用。

代码整合

将上面的代码整合在一起如下：

import psutil

# 设定要查找的进程名
process_name = "your_process_name.exe"  # 请修改为你希望查找的进程名

# 获取所有进程信息，使用 generator 遍历每一个进程
processes = psutil.process_iter(['pid', 'name', 'exe'])

# 遍历所有进程，查找匹配的进程
for proc in processes:
    if proc.info['name'] == process_name:
        print(f"进程ID: {proc.info['pid']}, 根目录: {proc.info['exe']}")

关系图

为了更好理解，下面是我们的进程管理模块和进程迭代之间的关系图：

erDiagram
    Process {
        int pid
        string name
        string exe
    }

    ProcessIter {
        Process[] processes
    }

    ProcessIter --|> Process : iterates

结尾

通过以上几个步骤，我们学习了如何使用 Python 脚本获取特定进程的根目录。整体流程清晰且简洁，适合入门的开发者理解和掌握。希望你能在此基础上继续深入研究 Python 的系统编程与进程管理。若有任何问题，请随时询问，我们一起来解决。