120 lines
4.3 KiB
Python
120 lines
4.3 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays battery status, remaining percentage and charging information.
|
|
|
|
Parameters:
|
|
* battery.device : Comma-separated list of battery devices to read information from (defaults to auto for auto-detection)
|
|
* battery.warning : Warning threshold in % of remaining charge (defaults to 20)
|
|
* battery.critical : Critical threshold in % of remaining charge (defaults to 10)
|
|
* battery.showdevice : If set to "true", add the device name to the widget (defaults to False)
|
|
"""
|
|
|
|
import os
|
|
import glob
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
import bumblebee.util
|
|
|
|
try:
|
|
import power
|
|
except ImportError:
|
|
pass
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
widgets = []
|
|
super(Module, self).__init__(engine, config, widgets)
|
|
self._batteries = self.parameter("device", "auto").split(",")
|
|
if self._batteries[0] == "auto":
|
|
self._batteries = glob.glob("/sys/class/power_supply/BAT*")
|
|
else:
|
|
self._batteries = ["/sys/class/power_supply/{}".format(b) for b in self._batteries]
|
|
if len(self._batteries) == 0:
|
|
self._batteries = ["/sys/class/power_supply/BAT0"]
|
|
self.update(widgets)
|
|
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
|
|
cmd="gnome-power-statistics")
|
|
|
|
def update(self, widgets):
|
|
new_widgets = []
|
|
for path in self._batteries:
|
|
widget = self.widget(path)
|
|
if not widget:
|
|
widget = bumblebee.output.Widget(full_text=self.capacity, name=path)
|
|
new_widgets.append(widget)
|
|
self.capacity(widget)
|
|
while len(widgets) > 0: del widgets[0]
|
|
for widget in new_widgets:
|
|
widgets.append(widget)
|
|
self._widgets = widgets
|
|
|
|
def remaining(self):
|
|
estimate = 0.0
|
|
try:
|
|
estimate = power.PowerManagement().get_time_remaining_estimate()
|
|
# do not show remaining if on AC
|
|
if estimate == power.common.TIME_REMAINING_UNLIMITED:
|
|
return None
|
|
if estimate == power.common.TIME_REMAINING_UNKNOWN:
|
|
return ""
|
|
except Exception:
|
|
return ""
|
|
return bumblebee.util.durationfmt(estimate*60, shorten=True, suffix=True) # estimate is in minutes
|
|
|
|
def capacity(self, widget):
|
|
widget.set("capacity", -1)
|
|
widget.set("ac", False)
|
|
if not os.path.exists(widget.name):
|
|
widget.set("capacity", 100)
|
|
widget.set("ac", True)
|
|
return "ac"
|
|
capacity = 100
|
|
try:
|
|
with open("{}/capacity".format(widget.name)) as f:
|
|
capacity = int(f.read())
|
|
except IOError:
|
|
return "n/a"
|
|
capacity = capacity if capacity < 100 else 100
|
|
widget.set("capacity", capacity)
|
|
if bumblebee.util.asbool(self.parameter("showdevice", False)):
|
|
widget.set("theme.minwidth", "100% ({})".format(os.path.basename(widget.name)))
|
|
return "{}% ({})".format(capacity, os.path.basename(widget.name))
|
|
widget.set("theme.minwidth", "100%")
|
|
|
|
remaining = None
|
|
if bumblebee.util.asbool(self.parameter("showremaining", True)):
|
|
remaining = self.remaining()
|
|
|
|
return "{}%{}".format(capacity, "" if not remaining else " ({})".format(remaining))
|
|
|
|
def state(self, widget):
|
|
state = []
|
|
capacity = widget.get("capacity")
|
|
|
|
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 = ""
|
|
with open("{}/status".format(widget.name)) as f:
|
|
charge = f.read().strip()
|
|
if charge == "Discharging":
|
|
state.append("discharging-{}".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
|