From b91d5ff62a8932fcaa431b69e2477873f8884c6a Mon Sep 17 00:00:00 2001
From: Ace <saadnaseer63@yahoo.de>
Date: Sat, 21 Sep 2024 00:21:00 +0200
Subject: [PATCH] Show battery level of PiSugar when PiSugar support is enabled

---
 inkycal/main.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/inkycal/main.py b/inkycal/main.py
index 6238082..52e738f 100644
--- a/inkycal/main.py
+++ b/inkycal/main.py
@@ -345,8 +345,12 @@ class Inkycal:
                 logger.info("All images generated successfully!")
             del errors
 
-            if self.battery_capacity < 20:
-                self.info += "Low battery!  "
+            if self.use_pi_sugar:
+                self.battery_capacity = self.pisugar.get_battery() or 0
+                if self.battery_capacity < 20:
+                    self.info += f"Low battery! ({self.battery_capacity})% "
+                else:
+                    self.info += f"Battery: {self.battery_capacity}% "
 
             # Assemble image from each module - add info section if specified
             self._assemble()