initial commit

modules/bar/components/stats.py

@@ -0,0 +1,94 @@
+from time import sleep
+import psutil
+from fabric import Fabricator
+from fabric.widgets.box import Box
+from fabric.widgets.button import Button
+from fabric.widgets.circularprogressbar import CircularProgressBar
+from fabric.widgets.label import Label
+from fabric.widgets.overlay import Overlay
+from snippets import MaterialIcon
+
+
+class SystemInfo(Button):
+    ICONS = [
+        ("CPU", "settings_motion_mode"),
+        ("Swap", "swap_horiz"),
+        ("RAM", "memory_alt"),
+        ("Temp", "thermometer"),
+    ]
+
+    def __init__(self):
+        super().__init__(child=self._initialize_content())
+
+        Fabricator(
+            poll_from=self._refresh_system_info,
+            stream=True,
+            on_changed=lambda fab, data: self._update_ui(data),
+        )
+
+    def _initialize_content(self):
+        self.box = Box(orientation="h", name="tray")
+        self.progress_bars = {}
+        self.labels = {}
+        self.icons = {}
+
+        for system, icon_name in self.ICONS:
+            self._create_system_item(system, icon_name)
+
+        return self.box
+
+    def _create_system_item(self, system, icon_name):
+        self.progress_bars[system] = self._create_progress_bar()
+        self.labels[system] = self._create_label()
+        self.icons[system] = self._create_icon(icon_name)
+
+        overlay = Overlay(
+            child=self.progress_bars[system], overlays=[self.icons[system]]
+        )
+        self.box.pack_start(overlay, False, False, 0)
+        self.box.pack_start(self.labels[system], False, False, 0)
+
+    def _create_progress_bar(self):
+        return CircularProgressBar(
+            name="progress", line_style="round", line_width=1, size=24
+        )
+
+    def _create_label(self):
+        return Label(style="font-size:14px; margin: 4px;")
+
+    def _create_icon(self, icon_name):
+        return MaterialIcon(icon_name, size=16)
+
+    def _refresh_system_info(self, fab: Fabricator):
+        while True:
+            yield {
+                "CPU": round(psutil.cpu_percent()),
+                "RAM": round(psutil.virtual_memory().percent),
+                "Swap": round(psutil.swap_memory().percent),
+                "Temp": self._get_device_temperature(),
+            }
+            sleep(1)
+
+    @staticmethod
+    def _get_device_temperature():
+        try:
+            temps = psutil.sensors_temperatures()
+            if "coretemp" in temps:
+                return round(temps["coretemp"][0].current, 1)
+            elif "cpu_thermal" in temps:
+                return round(temps["cpu_thermal"][0].current, 1)
+            return None
+        except Exception:
+            return None
+
+    def _update_ui(self, usages):
+        for system, usage in usages.items():
+            if usage is None:
+                usage = 0
+                self.labels[system].set_label("N/A")
+            else:
+                usage = int(usage)
+
+            self.progress_bars[system].value = usage / 100
+            self.labels[system].set_label(f"{usage}")
+            self.progress_bars[system].set_tooltip_text(f"{system} Usage: {usage}%")