diff --git a/modules/contrib/indicator.py b/modules/contrib/indicator.py new file mode 100644 index 0000000..3b57e33 --- /dev/null +++ b/modules/contrib/indicator.py @@ -0,0 +1,51 @@ +#pylint: disable=C0111,R0903 + +"""Displays the indicator status, for numlock, scrolllock and capslock + +Parameters: + * indicator.include: Comma-separated list of interface prefixes to include (defaults to "numlock,capslock") + * indicator.signalstype: If you want the signali type color to be "critical" or "warning" (defaults to "warning") +""" + + +import bumblebee.input +import bumblebee.output +import bumblebee.engine + +class Module(bumblebee.engine.Module): + def __init__(self, engine, config): + widgets = [] + self.status = False + super(Module,self).__init__(engine, config, widgets) + self._include = tuple(filter(len, self.parameter("include", "NumLock,CapsLock").split(","))) + self._signalType = self.parameter("signaltype") if not self.parameter("signaltype") is None else "warning" + + def update(self, widgets): + self._update_widgets(widgets) + + def state(self, widget): + states = [] + if widget.status: + states.append(self._signalType) + elif not widget.status: + states.append("normal") + return states + + def _update_widgets(self, widgets): + status_line = "" + for line in bumblebee.util.execute("xset q").replace(" ", "").split("\n"): + if "capslock" in line.lower(): + status_line = line + break + + for indicator in self._include: + widget = self.widget(indicator) + if not widget: + widget = bumblebee.output.Widget(name=indicator) + widgets.append(widget) + + widget.status = True if indicator.lower()+":on" in status_line.lower() else False + widget.full_text(indicator) + + +# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4