71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
import psutil
|
|
from fabric import Fabricator
|
|
from fabric.widgets.box import Box
|
|
from fabric.widgets.overlay import Overlay
|
|
from fabric.widgets.label import Label
|
|
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.battery_label = Label(label=f"{battery_percent}%", name="battery-label")
|
|
|
|
self.children = self.battery_label, 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
|