81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
import psutil
|
|
from gi.repository import GLib
|
|
from fabric.widgets.box import Box
|
|
from fabric.widgets.label import Label
|
|
from fabric.widgets.image import Image
|
|
from fabric import Fabricator
|
|
|
|
|
|
class BatteryProvider:
|
|
def __init__(self):
|
|
self.bat_percent = 0.0
|
|
self.bat_charging = None
|
|
|
|
self._update()
|
|
GLib.timeout_add_seconds(1, self._update)
|
|
|
|
def _update(self):
|
|
battery = psutil.sensors_battery()
|
|
if battery is None:
|
|
self.bat_percent = 0.0
|
|
self.bat_charging = None
|
|
else:
|
|
self.bat_percent = battery.percent
|
|
self.bat_charging = battery.power_plugged
|
|
|
|
return True
|
|
|
|
def get_battery(self):
|
|
return (self.bat_percent, self.bat_charging)
|
|
|
|
|
|
class Battery(Box):
|
|
def __init__(self, **kwargs):
|
|
super().__init__(name="battery-widget", orientation="h", spacing=4, **kwargs)
|
|
self.bat_provider = BatteryProvider()
|
|
|
|
self.bat_icon = Image(
|
|
name="bat-icon", icon_name="battery-full-symbolic", icon_size=16
|
|
)
|
|
|
|
self.bat_label = Label(name="bat-label", label="100%")
|
|
|
|
self.bat_fabricator = Fabricator(
|
|
poll_from=lambda *_: self.bat_provider.get_battery(),
|
|
on_changed=self.update_battery,
|
|
interval=1000,
|
|
stream=False,
|
|
default_value=(100, False),
|
|
)
|
|
|
|
self.children = [self.bat_icon, self.bat_label]
|
|
self.show_all()
|
|
|
|
GLib.idle_add(self.update_battery, None, self.bat_provider.get_battery())
|
|
|
|
def _icon_lookup(self, bat, charging):
|
|
# Round to nearest 10 for level-based icons
|
|
level = max(10, min(100, round(bat / 10) * 10))
|
|
|
|
if charging:
|
|
return f"battery-level-{level}-charging-symbolic"
|
|
else:
|
|
return f"battery-level-{level}-symbolic"
|
|
|
|
def update_battery(self, sender, battery_data):
|
|
value, charging = battery_data
|
|
|
|
icon_name = self._icon_lookup(value, charging)
|
|
self.bat_icon.set_property("icon-name", icon_name)
|
|
|
|
self.bat_label.set_text(f"{int(value)}%")
|
|
|
|
if value < 20 and not charging:
|
|
self.bat_label.add_style_class("battery-low")
|
|
self.bat_icon.add_style_class("battery-low")
|
|
else:
|
|
self.bat_label.remove_style_class("battery-low")
|
|
self.bat_icon.remove_style_class("battery-low")
|
|
|
|
return True
|