ad330ad70e
I'm not a major programmer but would this fix the issue that we had? Instead of going straight to BAT0 it would go to any BAT that exists? Correct me if I'm wrong.
94 lines
3.4 KiB
Python
94 lines
3.4 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
|
|
"""
|
|
|
|
import os
|
|
import glob
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
|
|
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/BAT*" ]
|
|
self.update(widgets)
|
|
|
|
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 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 self.parameter("showdevice") == "true":
|
|
widget.set("theme.minwidth", "100% ({})".format(os.path.basename(widget.name)))
|
|
return "{}% ({})".format(capacity, os.path.basename(widget.name))
|
|
widget.set("theme.minwidth", "100%")
|
|
return "{}%".format(capacity)
|
|
|
|
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
|