python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > psutil和subprocess程序监控与管理

Python高级技巧之利用psutil和subprocess实现程序监控与管理

作者：散一世繁华，颠半世琉璃

本文介绍了如何使用Python的psutil和subprocess模块监控程序运行状态,并提供了一个案例脚本,用于监控目标程序并在停止时自动重启,详细介绍了subprocess模块的基本用法和psutil模块的系统信息获取、进程管理及资源监控功能,需要的朋友可以参考下

1.简介

在关键的应用程序中，监听程序的运行状态十分重要，因为其可以确保程序的平稳运行。本文将介绍如何使用Python实现这一功能，这里主要使用psutil和subprocess模块。同时，最后会提供一个完整的脚本案例供参考，用于监控并在目标程序意外停止时重启程序。

2.subprocess模块介绍

subprocess 模块是 Python 中用于创建和管理子进程的标准库模块。它允许你执行系统命令或其他可执行文件，获取它们的输出，以及与其交互。下面是一些 subprocess 模块的基本用法和示例：

1. 运行外部命令

你可以使用 subprocess.run() 函数来运行外部命令，并等待其完成：

import subprocess

# 运行一个简单的命令
result = subprocess.run(['ls', '-l'], capture_output=True, text=True)

# 打印命令输出
print(result.stdout)

2. 后台执行命令

如果你希望命令在后台执行而不阻塞当前进程，可以使用 subprocess.Popen 类：

import subprocess

# 后台执行命令
process = subprocess.Popen(['python', 'target.py'])

# 等待命令完成
process.wait()

3. 获取命令输出

如果需要获取命令的输出，可以使用 subprocess.check_output() 函数：

import subprocess

# 获取命令输出
output = subprocess.check_output(['echo', 'Hello, subprocess!'], text=True)

# 打印输出
print(output)

4. 传递输入给命令

有时候需要将数据传递给命令的标准输入。可以使用 subprocess.Popen.communicate() 方法来实现：

import subprocess

# 启动命令并传递输入
process = subprocess.Popen(['grep', 'Python'], stdin=subprocess.PIPE, stdout=subprocess.PIPE, text=True)

# 向命令输入数据
output, _ = process.communicate('Python is awesome!')

# 打印命令输出
print(output)

5. 处理命令返回值

通过 subprocess.run() 或 subprocess.Popen.wait() 可以获取命令的返回值，通常是命令的退出代码。这可以帮助你判断命令是否成功执行。

3.psutil模块介绍

psutil（Python System and Process Utilities）模块是一个跨平台的库，提供了方便的接口来获取系统信息和管理进程。它可以用于监控系统资源的使用情况、管理进程以及执行系统管理任务。以下是 psutil 模块的一些主要功能和使用示例：

1. 安装 psutil

首先确保你已经安装了 psutil 模块。如果没有安装，可以通过 pip 安装：

pip install psutil

2. 获取系统信息

使用 psutil 可以轻松地获取系统的各种信息，如 CPU 使用率、内存使用情况、磁盘分区信息等。

import psutil

# 获取 CPU 逻辑核心数
cpu_count = psutil.cpu_count(logical=True)
print(f"CPU 逻辑核心数: {cpu_count}")

# 获取内存使用情况
memory = psutil.virtual_memory()
print(f"总内存: {memory.total / 1024 / 1024} MB")
print(f"已使用内存: {memory.used / 1024 / 1024} MB")

# 获取磁盘分区信息
disk_partitions = psutil.disk_partitions()
for partition in disk_partitions:
    print(f"分区设备: {partition.device}, 文件系统: {partition.fstype}")

3. 获取进程信息

psutil 可以用来获取系统中运行的进程信息，包括进程列表、CPU 使用率、内存使用情况等。

import psutil

# 获取所有进程列表
all_processes = list(psutil.process_iter())
for proc in all_processes[:5]:  # 打印前五个进程
    print(f"进程名: {proc.name()}, PID: {proc.pid}")

# 获取指定 PID 进程的详细信息
pid = 1234  # 替换成你要查看的进程 PID
if psutil.pid_exists(pid):
    process = psutil.Process(pid)
    print(f"进程名称: {process.name()}")
    print(f"进程状态: {process.status()}")
    print(f"进程创建时间: {process.create_time()}")
    print(f"进程 CPU 使用率: {process.cpu_percent(interval=1)}%")
    print(f"进程内存使用情况: {process.memory_info().rss / 1024 / 1024} MB")
else:
    print(f"PID {pid} 对应的进程不存在。")

4. 进程管理操作

psutil 还允许对进程进行管理操作，如终止进程、发送信号等。

import psutil

# 终止指定 PID 的进程
pid_to_terminate = 5678  # 替换成你要终止的进程 PID
if psutil.pid_exists(pid_to_terminate):
    process = psutil.Process(pid_to_terminate)
    process.terminate()
    print(f"进程 {pid_to_terminate} 已终止。")
else:
    print(f"PID {pid_to_terminate} 对应的进程不存在。")

5. 实时监控系统资源

使用 psutil 可以实现实时监控系统资源使用情况，例如在一个循环中获取 CPU 使用率：

import psutil

# 实时监控 CPU 使用率
while True:
    cpu_percent = psutil.cpu_percent(interval=1)
    print(f"当前 CPU 使用率: {cpu_percent}%")

这些示例展示了 psutil 模块在系统信息获取、进程管理和系统资源监控方面的基本用法。根据具体需求，你也可以进一步深入使用 psutil 的其他功能来实现更复杂的系统管理和监控任务。

4.实战应用

下面是一个使用subprocess模块来启动和监控一个Python程序的示例代码。后面将使用subprocess.Popen()来启动程序，并通过process.poll()来检查该进程是否仍在运行。如果目标程序停止运行，会获取其返回码，并根据返回码决定是否重新启动该程序。
具体代码如下所示：

import os
import subprocess
import sys
import time
import psutil  # 导入 psutil 模块

def start_target_program(target_script):
    # 显式指定Python解释器和环境变量
    python_executable = sys.executable
    env = dict(os.environ)
    env['PYTHONPATH'] = ":".join(sys.path)  # 将当前Python解释器的路径添加到PYTHONPATH环境变量中
    # 在这里启动你的Python程序
    return subprocess.Popen([python_executable, target_script], env=env)


def is_program_running(process):
    # 检查程序是否在运行
    return process.poll() is None

def bytes_to_readable(size_bytes):
    # 将字节数转换为可读的格式（KB、MB、GB）
    for unit in ['', 'KB', 'MB', 'GB']:
        if size_bytes < 1024.0:
            return f"{size_bytes:.2f} {unit}"
        size_bytes /= 1024.0

def monitor_process(target_script):
    # 启动被监测的程序
    target_program = start_target_program(target_script)
    while True:
        # 检查并打印 CPU 利用率
        cpu_percent = psutil.cpu_percent(interval=1)
        print(f"当前 CPU 利用率：{cpu_percent}%")

        # 获取并打印内存信息
        memory_info = psutil.virtual_memory()
        total_memory = bytes_to_readable(memory_info.total)
        available_memory = bytes_to_readable(memory_info.available)
        print(f"总内存: {total_memory}")
        print(f"可用内存: {available_memory}")

        # 获取并打印磁盘分区信息
        disk_partitions = psutil.disk_partitions()
        print("磁盘分区信息：")
        for partition in disk_partitions:
            print(f"    分区：{partition.device} 挂载点：{partition.mountpoint}")

        # 检查程序是否在运行
        if not is_program_running(target_program):
            # 检查程序结束状态（是否是正常结束）
            if target_program.returncode == 0:
                print("被监测的程序已正常关闭。")
                break
            else:
                print(f"被监测的程序已关闭，返回代码：{target_program.returncode}")
                print("重启中...")
                target_program = start_target_program(target_script)
        else:
            print("被监测的程序正常运行...")

        # 每隔10分钟监听一次
        time.sleep(600)

if __name__ == '__main__':
    # 需要注意被监控脚本的路径和监控程序保持在同一目录下
    target_script = input("请输入你需要监控的脚本名称（xxx.py）:")
    monitor_process(target_script)

通过结合使用 subprocess 和 psutil 模块，可以轻松实现对目标程序的监控和管理。

本文提供了一个简单的示例代码，展示了如何启动、监控和重启一个程序。希望这能帮助你更好地理解和使用 Python 进行系统和进程管理。

总结

到此这篇关于Python高级技巧之利用psutil和subprocess实现程序监控与管理的文章就介绍到这了,更多相关psutil和subprocess程序监控与管理内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:
阅读全文