100 lines
2.9 KiB
Python
100 lines
2.9 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
|
|
|
|
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):
|
|
if charging:
|
|
if bat >= 100:
|
|
return "battery-full-charging-symbolic"
|
|
elif bat > 70:
|
|
return "battery-good-charging-symbolic"
|
|
elif bat > 40:
|
|
return "battery-medium-charging-symbolic"
|
|
elif bat > 20:
|
|
return "battery-low-charging-symbolic"
|
|
else:
|
|
return "battery-caution-charging-symbolic"
|
|
else:
|
|
if bat >= 100:
|
|
return "battery-full-symbolic"
|
|
elif bat > 70:
|
|
return "battery-good-symbolic"
|
|
elif bat > 40:
|
|
return "battery-medium-symbolic"
|
|
elif bat > 20:
|
|
return "battery-low-symbolic"
|
|
else:
|
|
return "battery-caution-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_css_class("battery-low")
|
|
self.bat_icon.add_css_class("battery-low")
|
|
else:
|
|
self.bat_label.remove_css_class("battery-low")
|
|
self.bat_icon.remove_css_class("battery-low")
|
|
|
|
return True
|