From a858cc9083bc534221ceda32979e7c0db8e738ce Mon Sep 17 00:00:00 2001
From: Claudio Maradonna <penguyman@stronzi.org>
Date: Fri, 24 Mar 2023 21:47:45 +0100
Subject: [PATCH] more generic script; can enable zpool or upsc checking only
 executables; now handle low battery situation; now show battery percentage
 near bar

---
 odroid_hc4_oled/screen.py | 68 +++++++++++++++++++++++++++++----------
 1 file changed, 51 insertions(+), 17 deletions(-)

diff --git a/odroid_hc4_oled/screen.py b/odroid_hc4_oled/screen.py
index e266c30..0aafd4b 100644
--- a/odroid_hc4_oled/screen.py
+++ b/odroid_hc4_oled/screen.py
@@ -89,7 +89,7 @@ def zpool_print_status():
 
         draw.text((0, s_height - 10), zpool_status_data["errors"], fill=1)
 
-def ups_print_status():
+def upsc_print_status():
     ups_name_finder = os.popen('upsc -l 127.0.0.1')
     ups_name_finder_output = ups_name_finder.read().split("\n")
 
@@ -108,39 +108,73 @@ def ups_print_status():
 
         battery_charge = upsc_status_data["battery.charge"]
 
-        used_width = draw.textlength("[]%")
+        used_width = draw.textlength("[] "+battery_charge+"%")
         bar_width = (128 - used_width)
         real_width = (int(battery_charge) * bar_width) / 100
 
         draw.text((0,9), "[", fill=1)
         draw.rectangle((5,13, real_width, 17), fill=1)
-        draw.text((bar_width, 9), "]%", fill=1)
+        draw.text((bar_width, 9), "] "+battery_charge+"%", fill=1)
 
-        draw.text((0, 20), upsc_status_data["device.model"], fill=1)
-        draw.text((0, 28), upsc_status_data["outlet.power"] + "W", fill=1)
-        draw.text((0, 36), upsc_status_data["output.voltage"] + "V", fill=1)
+        what_to_show = {
+                'device.model': '%{value}',
+                'outlet.power': '%{value}W',
+                'output.voltage': '%{value}V'
+        }
+
+        starting_height = 19
+        if int(battery_charge) <= int(upsc_status_data['battery.charge.low']):
+            draw.text((0, starting_height + 9),  "*********************", fill=1)
+            draw.text((0, starting_height + 18), "**** BATTERY LOW ****", fill=1)
+            draw.text((0, starting_height + 27), "*********************", fill=1)
+        else:
+            starting_height += 9
+            for key in what_to_show:
+                value = what_to_show[key].replace('%{value}', upsc_status_data[key])
+
+                draw.text((0, starting_height), value, fill=1)
+
+                starting_height += 9
 
 def main():
-    if which('zpool') is None:
-        print("Cannot find `zpool` executable.")
+    modules_enabled = []
+
+    if which('zpool') is not None:
+        modules_enabled.append('zpool')
+
+    if which('upsc') is not None:
+        modules_enabled.append('upsc')
+
+    if not modules_enabled:
+        print("This script cannot run any module; check dependencies")
 
         exit(1)
 
-    if which('upsc') is None:
-        print("Cannot find `upsc` executable.")
-
-        exit(1)
+    possibles = globals().copy()
+    possibles.update(locals())
 
     start_count = 0
+    last_print = ''
     while True:
-        device.clear()
+        module_to_run = modules_enabled[start_count % 2]
+        method = possibles.get(module_to_run + "_print_status")
 
-        if (start_count % 2) == 0:
-            zpool_print_status()
+        if not method:
+            print("Module "+module_to_run+" not implemented")
+
+            exit(1)
         else:
-            ups_print_status()
+            if module_to_run != last_print:
+                device.clear()
+
+            method()
+            last_print = module_to_run
+
+        if start_count == (len(modules_enabled) - 1):
+            start_count = 0
+        else:
+            start_count += 1
 
-        start_count += 1
         time.sleep(5)
 
 if __name__ == "__main__":