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)

# 給檢查任務設(shè)個id,方便任務的取消
sched.add_job(spaceMonitorJob, trigger=intervalTrigger, id='id_space_monitor')
sched.start()

# 禁止apscheduler相關(guān)信息屏幕輸出
logging.getLogger('apscheduler.executors.default').propagate = False

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.')

  # 移除任務，病關(guān)閉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)