345 lines
13 KiB
Python
345 lines
13 KiB
Python
# UPowerManger Class Copyright (C) 2017 Oscar Svensson (wogscpar) under MIT licence from upower-python
|
|
|
|
"""Displays battery status, remaining percentage and charging information.
|
|
|
|
Parameters:
|
|
* battery-upower.warning : Warning threshold in % of remaining charge (defaults to 20)
|
|
* battery-upower.critical : Critical threshold in % of remaining charge (defaults to 10)
|
|
* battery-upower.showremaining : If set to true (default) shows the remaining time until the batteries are completely discharged
|
|
|
|
contributed by `martindoublem <https://github.com/martindoublem>`_ - many thanks!
|
|
"""
|
|
|
|
import dbus
|
|
import logging
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
|
|
import util.format
|
|
|
|
|
|
class UPowerManager:
|
|
def __init__(self):
|
|
self.UPOWER_NAME = "org.freedesktop.UPower"
|
|
self.UPOWER_PATH = "/org/freedesktop/UPower"
|
|
|
|
self.DBUS_PROPERTIES = "org.freedesktop.DBus.Properties"
|
|
self.bus = dbus.SystemBus()
|
|
|
|
def detect_devices(self):
|
|
upower_proxy = self.bus.get_object(self.UPOWER_NAME, self.UPOWER_PATH)
|
|
upower_interface = dbus.Interface(upower_proxy, self.UPOWER_NAME)
|
|
|
|
devices = upower_interface.EnumerateDevices()
|
|
return devices
|
|
|
|
def get_display_device(self):
|
|
upower_proxy = self.bus.get_object(self.UPOWER_NAME, self.UPOWER_PATH)
|
|
upower_interface = dbus.Interface(upower_proxy, self.UPOWER_NAME)
|
|
|
|
dispdev = upower_interface.GetDisplayDevice()
|
|
return dispdev
|
|
|
|
def get_critical_action(self):
|
|
upower_proxy = self.bus.get_object(self.UPOWER_NAME, self.UPOWER_PATH)
|
|
upower_interface = dbus.Interface(upower_proxy, self.UPOWER_NAME)
|
|
|
|
critical_action = upower_interface.GetCriticalAction()
|
|
return critical_action
|
|
|
|
def get_device_percentage(self, battery):
|
|
battery_proxy = self.bus.get_object(self.UPOWER_NAME, battery)
|
|
battery_proxy_interface = dbus.Interface(battery_proxy, self.DBUS_PROPERTIES)
|
|
|
|
return battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Percentage")
|
|
|
|
def get_full_device_information(self, battery):
|
|
battery_proxy = self.bus.get_object(self.UPOWER_NAME, battery)
|
|
battery_proxy_interface = dbus.Interface(battery_proxy, self.DBUS_PROPERTIES)
|
|
|
|
hasHistory = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "HasHistory"
|
|
)
|
|
hasStatistics = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "HasStatistics"
|
|
)
|
|
isPresent = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "IsPresent"
|
|
)
|
|
isRechargeable = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "IsRechargeable"
|
|
)
|
|
online = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Online")
|
|
powersupply = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "PowerSupply"
|
|
)
|
|
capacity = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Capacity")
|
|
energy = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Energy")
|
|
energyempty = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "EnergyEmpty"
|
|
)
|
|
energyfull = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "EnergyFull"
|
|
)
|
|
energyfulldesign = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "EnergyFullDesign"
|
|
)
|
|
energyrate = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "EnergyRate"
|
|
)
|
|
luminosity = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "Luminosity"
|
|
)
|
|
percentage = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "Percentage"
|
|
)
|
|
temperature = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "Temperature"
|
|
)
|
|
voltage = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Voltage")
|
|
timetoempty = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "TimeToEmpty"
|
|
)
|
|
timetofull = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "TimeToFull"
|
|
)
|
|
iconname = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "IconName")
|
|
model = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Model")
|
|
nativepath = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "NativePath"
|
|
)
|
|
serial = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Serial")
|
|
vendor = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Vendor")
|
|
state = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "State")
|
|
technology = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "Technology"
|
|
)
|
|
battype = battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "Type")
|
|
warninglevel = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "WarningLevel"
|
|
)
|
|
updatetime = battery_proxy_interface.Get(
|
|
self.UPOWER_NAME + ".Device", "UpdateTime"
|
|
)
|
|
|
|
information_table = {
|
|
"HasHistory": hasHistory,
|
|
"HasStatistics": hasStatistics,
|
|
"IsPresent": isPresent,
|
|
"IsRechargeable": isRechargeable,
|
|
"Online": online,
|
|
"PowerSupply": powersupply,
|
|
"Capacity": capacity,
|
|
"Energy": energy,
|
|
"EnergyEmpty": energyempty,
|
|
"EnergyFull": energyfull,
|
|
"EnergyFullDesign": energyfulldesign,
|
|
"EnergyRate": energyrate,
|
|
"Luminosity": luminosity,
|
|
"Percentage": percentage,
|
|
"Temperature": temperature,
|
|
"Voltage": voltage,
|
|
"TimeToEmpty": timetoempty,
|
|
"TimeToFull": timetofull,
|
|
"IconName": iconname,
|
|
"Model": model,
|
|
"NativePath": nativepath,
|
|
"Serial": serial,
|
|
"Vendor": vendor,
|
|
"State": state,
|
|
"Technology": technology,
|
|
"Type": battype,
|
|
"WarningLevel": warninglevel,
|
|
"UpdateTime": updatetime,
|
|
}
|
|
|
|
return information_table
|
|
|
|
def is_lid_present(self):
|
|
upower_proxy = self.bus.get_object(self.UPOWER_NAME, self.UPOWER_PATH)
|
|
upower_interface = dbus.Interface(upower_proxy, self.DBUS_PROPERTIES)
|
|
|
|
is_lid_present = bool(upower_interface.Get(self.UPOWER_NAME, "LidIsPresent"))
|
|
return is_lid_present
|
|
|
|
def is_lid_closed(self):
|
|
upower_proxy = self.bus.get_object(self.UPOWER_NAME, self.UPOWER_PATH)
|
|
upower_interface = dbus.Interface(upower_proxy, self.DBUS_PROPERTIES)
|
|
|
|
is_lid_closed = bool(upower_interface.Get(self.UPOWER_NAME, "LidIsClosed"))
|
|
return is_lid_closed
|
|
|
|
def on_battery(self):
|
|
upower_proxy = self.bus.get_object(self.UPOWER_NAME, self.UPOWER_PATH)
|
|
upower_interface = dbus.Interface(upower_proxy, self.DBUS_PROPERTIES)
|
|
|
|
on_battery = bool(upower_interface.Get(self.UPOWER_NAME, "OnBattery"))
|
|
return on_battery
|
|
|
|
def has_wakeup_capabilities(self):
|
|
upower_proxy = self.bus.get_object(
|
|
self.UPOWER_NAME, self.UPOWER_PATH + "/Wakeups"
|
|
)
|
|
upower_interface = dbus.Interface(upower_proxy, self.DBUS_PROPERTIES)
|
|
|
|
has_wakeup_capabilities = bool(
|
|
upower_interface.Get(self.UPOWER_NAME + ".Wakeups", "HasCapability")
|
|
)
|
|
return has_wakeup_capabilities
|
|
|
|
def get_wakeups_data(self):
|
|
upower_proxy = self.bus.get_object(
|
|
self.UPOWER_NAME, self.UPOWER_PATH + "/Wakeups"
|
|
)
|
|
upower_interface = dbus.Interface(upower_proxy, self.UPOWER_NAME + ".Wakeups")
|
|
|
|
data = upower_interface.GetData()
|
|
return data
|
|
|
|
def get_wakeups_total(self):
|
|
upower_proxy = self.bus.get_object(
|
|
self.UPOWER_NAME, self.UPOWER_PATH + "/Wakeups"
|
|
)
|
|
upower_interface = dbus.Interface(upower_proxy, self.UPOWER_NAME + ".Wakeups")
|
|
|
|
data = upower_interface.GetTotal()
|
|
return data
|
|
|
|
def is_battery_present(self, battery):
|
|
battery_proxy = self.bus.get_object(self.UPOWER_NAME, battery)
|
|
battery_proxy_interface = dbus.Interface(battery_proxy, self.DBUS_PROPERTIES)
|
|
|
|
return bool(battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "IsPresent"))
|
|
|
|
def is_loading(self, battery):
|
|
battery_proxy = self.bus.get_object(self.UPOWER_NAME, battery)
|
|
battery_proxy_interface = dbus.Interface(battery_proxy, self.DBUS_PROPERTIES)
|
|
|
|
state = int(battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "State"))
|
|
|
|
if state == 1:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def get_state(self, battery):
|
|
battery_proxy = self.bus.get_object(self.UPOWER_NAME, battery)
|
|
battery_proxy_interface = dbus.Interface(battery_proxy, self.DBUS_PROPERTIES)
|
|
|
|
state = int(battery_proxy_interface.Get(self.UPOWER_NAME + ".Device", "State"))
|
|
|
|
if state == 0:
|
|
return "Unknown"
|
|
elif state == 1:
|
|
return "Loading"
|
|
elif state == 2:
|
|
return "Discharging"
|
|
elif state == 3:
|
|
return "Empty"
|
|
elif state == 4:
|
|
return "Fully charged"
|
|
elif state == 5:
|
|
return "Pending charge"
|
|
elif state == 6:
|
|
return "Pending discharge"
|
|
|
|
|
|
class Module(core.module.Module):
|
|
def __init__(self, config, theme):
|
|
super().__init__(config, theme, core.widget.Widget(self.capacity))
|
|
|
|
try:
|
|
self.power = UPowerManager()
|
|
self.device = self.power.get_display_device()
|
|
except Exception as e:
|
|
logging.exception("unable to get battery display device: {}".format(str(e)))
|
|
core.input.register(
|
|
self, button=core.input.LEFT_MOUSE, cmd="gnome-power-statistics"
|
|
)
|
|
|
|
self._showremaining = util.format.asbool(self.parameter("showremaining", True))
|
|
|
|
def capacity(self, widget):
|
|
widget.set("capacity", -1)
|
|
widget.set("ac", False)
|
|
output = "n/a"
|
|
if not self.power.is_battery_present(self.device):
|
|
widget.set("ac", True)
|
|
widget.set("capacity", 100)
|
|
output = "ac"
|
|
return output
|
|
try:
|
|
capacity = int(self.power.get_device_percentage(self.device))
|
|
capacity = capacity if capacity < 100 else 100
|
|
widget.set("capacity", capacity)
|
|
output = "{}%".format(capacity)
|
|
widget.set("theme.minwidth", "100%")
|
|
except Exception as e:
|
|
logging.exception("unable to get battery capacity: {}".format(str(e)))
|
|
|
|
if self._showremaining:
|
|
try:
|
|
p = self.power # an alias to make each line of code shorter
|
|
proxy = p.bus.get_object(p.UPOWER_NAME, self.device)
|
|
interface = dbus.Interface(proxy, p.DBUS_PROPERTIES)
|
|
state = int(interface.Get(p.UPOWER_NAME + ".Device", "State"))
|
|
# state: 1 => charging, 2 => discharging, other => don't care
|
|
remain = int(
|
|
interface.Get(
|
|
p.UPOWER_NAME + ".Device",
|
|
["TimeToFull", "TimeToEmpty"][state - 1],
|
|
)
|
|
)
|
|
remain = util.format.duration(remain, compact=True, unit=True)
|
|
output = "{} {}".format(output, remain)
|
|
except IndexError:
|
|
pass
|
|
except Exception as e:
|
|
logging.exception(
|
|
"unable to get battery remaining time: {}".format(str(e))
|
|
)
|
|
|
|
return output
|
|
|
|
def state(self, widget):
|
|
state = []
|
|
capacity = widget.get("capacity", -1)
|
|
if capacity < 0:
|
|
return ["critical", "unknown"]
|
|
|
|
if capacity < int(self.parameter("critical", 10)):
|
|
state.append("critical")
|
|
elif capacity < int(self.parameter("warning", 20)):
|
|
state.append("warning")
|
|
|
|
if widget.get("ac"):
|
|
state.append("AC")
|
|
else:
|
|
charge = "Unknown"
|
|
try:
|
|
charge = self.power.get_state(self.device)
|
|
except Exception as e:
|
|
logging.exception("unable to get charge value: {}".format(str(e)))
|
|
if charge == "Discharging":
|
|
state.append(
|
|
"discharging-{}".format(
|
|
min([10, 25, 50, 80, 100], key=lambda i: abs(i - capacity))
|
|
)
|
|
)
|
|
elif charge == "Unknown":
|
|
state.append(
|
|
"unknown-{}".format(
|
|
min([10, 25, 50, 80, 100], key=lambda i: abs(i - capacity))
|
|
)
|
|
)
|
|
else:
|
|
if capacity > 95:
|
|
state.append("charged")
|
|
else:
|
|
state.append("charging")
|
|
return state
|
|
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|