Python中运维神器Psutil的用法详解

import psutil

# 获取CPU使用率
cpu_usage = psutil.cpu_percent()
print(f"CPU使用率: {cpu_usage}%")

# 获取内存使用情况
memory_info = psutil.virtual_memory()
print(f"内存总量: {memory_info.total / (1024 ** 3):.2f} GB")
print(f"内存已用: {memory_info.used / (1024 ** 3):.2f} GB")
print(f"内存可用: {memory_info.available / (1024 ** 3):.2f} GB")
print(f"内存使用率: {memory_info.percent}%")

# 获取磁盘使用情况
disk_info = psutil.disk_usage('/')
print(f"磁盘总量: {disk_info.total / (1024 ** 3):.2f} GB")
print(f"磁盘已用: {disk_info.used / (1024 ** 3):.2f} GB")
print(f"磁盘可用: {disk_info.free / (1024 ** 3):.2f} GB")
print(f"磁盘使用率: {disk_info.percent}%")

import psutil

# 获取所有进程信息
processes = psutil.process_iter(['pid', 'name', 'cpu_percent', 'memory_percent'])
for proc in processes:
    print(proc.info)

# 获取特定进程信息（以PID为1的进程为例）
pid = 1
proc = psutil.Process(pid)
print(f"进程ID: {proc.pid}")
print(f"进程名称: {proc.name()}")
print(f"CPU使用率: {proc.cpu_percent(interval=1)}%")
print(f"内存使用率: {proc.memory_percent()}%")

import psutil
import time

# 监控CPU使用率变化
print("开始监控CPU使用率...")
for i in range(10):
    cpu_usage = psutil.cpu_percent(interval=1)
    print(f"CPU使用率: {cpu_usage}%")
    time.sleep(1)

# 查找CPU使用率最高的进程
highest_cpu_proc = max(psutil.process_iter(['pid', 'name', 'cpu_percent']), key=lambda p: p.info['cpu_percent'])
print(f"CPU使用率最高的进程: PID={highest_cpu_proc.info['pid']}, 名称={highest_cpu_proc.info['name']}, 使用率={highest_cpu_proc.info['cpu_percent']}%")

import psutil

# 编写一个自动关闭CPU使用率超过80%的进程的脚本
def close_high_cpu_processes():
    for proc in psutil.process_iter(['pid', 'name', 'cpu_percent']):
        if proc.info['cpu_percent'] > 80:
            print(f"关闭进程: PID={proc.info['pid']}, 名称={proc.info['name']}, 使用率={proc.info['cpu_percent']}%")
            proc.terminate()  # 终止进程
            proc.wait(timeout=3)  # 等待进程终止，超时为3秒
            if proc.is_running():
                proc.kill()  # 强制终止进程close_high_cpu_processes()