refactor(battery): update status class based on threshold for icons

amnweb · Oct 29, 2024 · 6695b01 · 6695b01
1 parent 5351ac0
commit 6695b01
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/core/widgets/yasb/battery.py b/src/core/widgets/yasb/battery.py
@@ -176,6 +176,11 @@ def _update_label(self):
                     # Ensure the icon is correctly set
                     icon = re.sub(r'<span.*?>|</span>', '', battery_status).strip()
                     active_widgets[widget_index].setText(icon)
+                    icon = re.sub(r'<span.*?>|</span>', '', battery_status).strip()
+                    active_widgets[widget_index].setText(icon)
+                    existing_classes = active_widgets[widget_index].property("class")
+                    new_classes = re.sub(r'status-\w+', '', existing_classes).strip()
+                    active_widgets[widget_index].setProperty("class", f"{new_classes} status-{threshold}")
                 else:
                     alt_class = "alt" if self._show_alt_label else ""
                     formatted_text = battery_status.format(battery_status)