非常游戏网
如何用Python自动化监控Linux磁盘内存使用率并报警

如何用Python自动化监控Linux磁盘内存使用率并报警

2026-05-29日常编程224507

psutil.disk_usage()通过statvfs()获取实时磁盘使用率,比df -h更准确;需传标准化绝对路径,监控应判连续阈值而非瞬时峰值,并以systemd Type=simple长期运行。

psutil 获取真实磁盘使用率,别信 df -h 的缓存值

Linux 的 df 命令默认显示的是挂载点的统计,但内核可能缓存了部分元数据,导致短时间重复调用结果滞后。而 psutil.disk_usage(path) 底层调用 statvfs(),直接读取当前内核视图,更及时。监控根分区时,传 '/';监控 /data 就传 '/data'。注意:路径必须存在且可访问,否则抛 FileNotFoundError

常见错误是传错路径字符串——比如写成 '/data/'(末尾斜杠)却没做 os.path.normpath() 标准化,在某些挂载配置下会意外失败。建议统一用 os.path.abspath() 处理输入路径。

示例片段:

import psutil
usage = psutil.disk_usage('/')
print(f"已用: {usage.percent:.1f}%")  # 返回 float,非字符串

阈值触发逻辑要区分“瞬时峰值”和“持续越界”

磁盘涨到 95% 可能只是某个日志轮转瞬间的事,立刻发报警反而造成干扰。真正该报警的是“连续 3 次检测(间隔 30 秒)都 ≥92%”。用一个长度为 3 的列表或 deque 记录最近三次的 percent 值,每次新值追加、老值弹出,再用 all(v >= 92.0 for v in history) 判断。

不要用简单 if 判断单次值——线上环境里,du 扫描、rsync 写入、容器镜像拉取都可能引发秒级尖峰,这类毛刺不值得告警。

关键点:

  • psutil.disk_usage() 调用本身很快(微秒级),但频繁调用(如每 5 秒)对 I/O 几乎无压力
  • 避免在循环里反复调用 psutil.disk_partitions() 获取所有挂载点——它涉及 /proc/mounts 解析,开销大;应预先过滤出目标路径再查 usage
  • 若监控多个路径,别用嵌套循环,改用列表推导式批量采集

subprocess 发邮件或调用企业微信机器人,别硬编码 SMTP 密码

明文写密码在脚本里等于裸奔。把邮箱密码、Webhook URL 这类敏感信息抽到外部文件(如 /etc/diskmon/secrets.json),用 json.load() 读取,并确保该文件权限为 600chmod 600 /etc/diskmon/secrets.json)。Python 进程启动用户需有读取权限。

Python 3.14.3

微软官方的 Python 扩展,是 VS Code 安装量最高的扩展(209M+)。集成 IntelliSense(通过 Pylance)、调试(通过 Python Debugger)、代码检查、格式化、重构和单元测试等功能。支持 Jupyter Notebook、虚拟环境管理和多 Python 版本切换。

发邮件推荐走本地 sendmail(无需认证,依赖系统配置),命令形如:['sendmail', '-t'],然后把完整邮件内容(含 headers)喂给 stdin。比用 smtplib 更轻量、更少 TLS 兼容问题。

调用企业微信机器人则用 requests.post(),注意设置 timeout=(5, 10) 防卡死,且检查响应状态码是否为 200resp.json().get('errcode') == 0,否则记录失败日志而非静默忽略。

部署为 systemd service 时,Type=oneshot 是错的

监控必须长期运行,不能用 Type=oneshot(执行完就退出)。正确设为 Type=simple,并配 Restart=alwaysRestartSec=10。否则进程崩溃后不会自拉起,监控就断了。

另外两个易漏点:

  • 在 service 文件里加 Environment="PYTHONPATH=/opt/diskmon",避免因路径问题导入不到自定义模块
  • StandardOutput=journal + StandardError=journal,这样 journalctl -u diskmon.service 就能查日志,别重定向到文件再搞 logrotate
  • 如果脚本里用了 time.sleep(30),记得在 service 中加 StartLimitIntervalSec=0,否则 systemd 可能误判为“频繁崩溃”而拒启

真正难的不是取值或发消息,而是让这个进程在服务器重启、OOM killer 杀掉、磁盘满导致 Python 无法写临时文件等极端情况下,还能被 systemd 拉起来、继续干活——这些边界得靠配置兜底,不是代码能全 cover 的。