bumblebee-status/modules/contrib/indicator.py

#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 core.module
import core.widget

import util.cli
import util.format

class Module(core.module.Module):
    def __init__(self, config, theme):
        super().__init__(config, theme, [])

        self.__include = tuple(filter(len, util.format.aslist(self.parameter('include', 'NumLock,CapsLock'))))
        self.__signalType = self.parameter('signaltype') if not self.parameter('signaltype') is None else 'warning'

    def update(self):
        status_line = '' 
        for line in util.cli.execute('xset q', ignore_errors=True).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 = core.widget.Widget(name=indicator, module=self)
                self.widgets().append(widget)

            widget.set('status', True if '{}:on'.format(indicator.lower()) in status_line.lower() else False)
            widget.full_text(indicator)

    def state(self, widget):
        states = []
        if widget.get('status', False):
            states.append(self.__signalType)
        else:
            states.append('normal')
        return states

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4