欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python3实现磁盘空间监控

程序员文章站 2022-10-29 16:13:04
本文实例为大家分享了python3磁盘空间监控的具体代码,供大家参考,具体内容如下 软硬件环境 python3 apscheduler 前言 在做频繁操作...

本文实例为大家分享了python3磁盘空间监控的具体代码,供大家参考,具体内容如下

软硬件环境

python3
apscheduler

前言

在做频繁操作磁盘的python项目时,经常会碰到磁盘空间不足的情况,这个时候,工程应该要有自己的处理模块,当磁盘利用率到达某个点时,发出警告并停止程序的运行。本文就利用python3中的apscheduler模块来处理这样的问题。

代码实践

import os
import sys
import signal
import logging

from apscheduler.schedulers.background import backgroundscheduler
from apscheduler.triggers.interval import intervaltrigger

# 开启磁盘空间检测
sched = backgroundscheduler()

# 间隔5分钟开启一个检查
intervaltrigger = intervaltrigger(minutes=5)

# 给检查任务设个id,方便任务的取消
sched.add_job(spacemonitorjob, trigger=intervaltrigger, id='id_space_monitor')
sched.start()

# 禁止apscheduler相关信息屏幕输出
logging.getlogger('apscheduler.executors.default').propagate = false

方法spacemonitorjob代码如下

def spacemonitorjob():
 '''
 当磁盘(切片存储的目录)利用率超过90%,程序退出
 :return:
 '''

 try:
  st = os.statvfs('/')
  total = st.f_blocks * st.f_frsize
  used = (st.f_blocks - st.f_bfree) * st.f_frsize
 except filenotfounderror:
  print('check webroot space error.')
  logger.error('check webroot space error.')

  # 移除任务,病关闭sched任务
  sched.remove_job(job_id='id_space_monitor')
  sched.shutdown(wait=false)
  sys.exit(-3)

 if used / total > 0.9:
  print('no enough space.')
  logger.debug('no enough space.')
  sched.remove_job(job_id='id_space_monitor')
  sched.shutdown(wait=false)

  # 杀掉进程
  os.killpg(os.getpgid(os.getpid()), signal.sigkill)

  # 退出
  exit(-3)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。