vdstatus/plugins/disk.py

import psutil
import threading
import time


class PluginThread(threading.Thread):
    def __init__(self, section, config, thread_id):
        threading.Thread.__init__(self)
        self.threadID = thread_id
        self.status = dict()
        self.part = config.get(section, 'part')
        if config.has_option(section, 'color'):
            self.status['color'] = config.get(section, 'color')
        self.freq = config.getint(section, 'freq', fallback=30)
        self.hide = False
        self.should_stop = False
        self.problem_value = config.getint(section, 'problem', fallback=70)

    def main(self):
        du_stat = psutil.disk_usage(self.part)
        if du_stat.percent >= self.problem_value:
            self.hide = False
            self.status['urgent'] = True
        else:
            self.hide = True
            self.status['urgent'] = False
        du_free = str(round(du_stat.free / 2**30, 2))
        du = self.part + ': ' + du_free + 'G'
        self.status['full_text'] = du

    def stop(self):
        self.should_stop = True

    def run(self):
        while True:
            if self.should_stop is False:
                self.main()
                time.sleep(self.freq)
            else:
                break