disk check to auto hide as well

2 files changed, 12 insertions, 6 deletions

diff --git a/plugins/disk.py b/plugins/disk.py
index 0f508c0..4f54241 100644
--- a/plugins/disk.py
+++ b/plugins/disk.py
@@ -13,10 +13,16 @@ class PluginThread(threading.Thread):
             self.status['color'] = config.get(section, 'color')
         self.freq = config.getint(section, 'freq', fallback=30)
         self.hide = False
+        self.problem_value = config.getint(section, 'problem', fallback=70)
 
     def main(self):
         du_stat = psutil.disk_usage(self.part)
-        # du_perc = str(du_stat.percent)
+        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
diff --git a/vdstatus b/vdstatus
index 18abc9a..4be025f 100755
--- a/vdstatus
+++ b/vdstatus
@@ -1,8 +1,8 @@
 #!/usr/bin/python3 -u
-# TODO:
-# * handle SIGINT properly
-# * add documentation / comments
-# * add a dummy plugin to use as a starting point
+# TODO: handle SIGINT properly
+# TODO: add documentation / comments
+# TODO: add a dummy plugin to use as a starting point
+# TODO: interactivity support
 from sys import argv
 import argparse
 import configparser
@@ -70,7 +70,7 @@ def run_plugins(config_file=DEFAULT_CONFIG):
     while True:
         outputs = list()
         for plugin in plugins_l:
-            if plugin.hide == False:
+            if not plugin.hide:
                 outputs.append(plugin.status)
         print(format_outputs(outputs))
         time.sleep(1)