initial commit

modules/bar/components/battery.py

@@ -0,0 +1,68 @@
+import psutil
+from fabric import Fabricator
+from fabric.widgets.box import Box
+from fabric.widgets.overlay import Overlay
+from fabric.widgets.circularprogressbar import CircularProgressBar
+from snippets import MaterialIcon
+
+
+class BatteryLabel(Box):
+    ICONS_CHARGING = [
+        "battery_charging_20",
+        "battery_charging_20",
+        "battery_charging_20",
+        "battery_charging_30",
+        "battery_charging_30",
+        "battery_charging_50",
+        "battery_charging_60",
+        "battery_charging_80",
+        "battery_charging_80",
+        "battery_charging_90",
+        "battery_charging_full",
+    ]
+    ICONS_NOT_CHARGING = [
+        "battery_alert",
+        "battery_1_bar",
+        "battery_1_bar",
+        "battery_2_bar",
+        "battery_2_bar",
+        "battery_3_bar",
+        "battery_4_bar",
+        "battery_4_bar",
+        "battery_5_bar",
+        "battery_6_bar",
+        "battery_full",
+    ]
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        Fabricator(interval=1000, poll_from=self.update_battery_status)
+
+    def update_battery_status(self, *_):
+        battery = psutil.sensors_battery()
+        if battery is None:
+            self.hide()
+            return
+        is_charging = battery.power_plugged if battery else False
+        icons = self.ICONS_CHARGING if is_charging else self.ICONS_NOT_CHARGING
+        battery_percent = round(battery.percent)
+        index = min(max(battery_percent // 10, 0), 10)
+        battery_icon = MaterialIcon(icons[index], size=16)
+
+        self.progress_bar = CircularProgressBar(
+            name="progress", line_style="round", line_width=1, size=24
+        )
+
+        self.battery_overlay = Overlay(child=self.progress_bar, overlays=battery_icon)
+
+        self.children = self.battery_overlay
+
+        battery_percent = round(battery.percent)
+        self.progress_bar.value = battery_percent / 100
+        # self.show() if battery_percent < 100 else self.hide()
+
+        self.set_tooltip_text(
+            f"Battery: {battery_percent}%\nStatus: {'Charging' if is_charging else 'Discharging'}"
+        )
+
+        return True