40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
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
|