9cd9ff626d
first, this fixes #607 also, i think it slightly simplifies code to make "simple" stuff like names, etc. attributes instead of methods all the time. so, expect this to be extended to other components, as well.
114 lines
3.4 KiB
Python
114 lines
3.4 KiB
Python
import importlib
|
|
import logging
|
|
|
|
import core.input
|
|
import core.widget
|
|
import core.decorators
|
|
|
|
try:
|
|
error = ModuleNotFoundError('')
|
|
except Exception as e:
|
|
ModuleNotFoundError = Exception
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
def load(module_name, config=core.config.Config([]), theme=None):
|
|
error = None
|
|
module_short, alias = (module_name.split(':') + [module_name])[0:2]
|
|
config.set('__alias__', alias)
|
|
for namespace in [ 'core', 'contrib' ]:
|
|
try:
|
|
mod = importlib.import_module('modules.{}.{}'.format(namespace, module_short))
|
|
return getattr(mod, 'Module')(config, theme)
|
|
except ImportError as e:
|
|
log.fatal('failed to import {}: {}'.format(module_short, e))
|
|
if not error or module_short in error:
|
|
error = str(e)
|
|
log.fatal('failed to import {}: {}'.format(module_name, error))
|
|
return Error(config=config, module=module_name, error=error)
|
|
|
|
class Module(core.input.Object):
|
|
def __init__(self, config=core.config.Config([]), theme=None, widgets=[]):
|
|
super().__init__()
|
|
self.__config = config
|
|
self.__widgets = widgets if isinstance(widgets, list) else [ widgets ]
|
|
for widget in self.__widgets:
|
|
widget.module(self)
|
|
self.__name = None
|
|
self.alias = self.__config.get('__alias__', None)
|
|
self.next_update = None
|
|
|
|
self.theme = theme
|
|
|
|
def hidden(self):
|
|
return False
|
|
|
|
def parameter(self, key, default=None):
|
|
value = default
|
|
|
|
for prefix in [ self.name(), self.module_name(), self.alias ]:
|
|
value = self.__config.get('{}.{}'.format(prefix, key), value)
|
|
# TODO retrieve from config file
|
|
return value
|
|
|
|
def set(self, key, value):
|
|
self.__config.set('{}.{}'.format(self.name(), key), value)
|
|
|
|
def update(self):
|
|
pass
|
|
|
|
def update_wrapper(self):
|
|
try:
|
|
self.update()
|
|
except Exception as e:
|
|
self.set('interval', 1)
|
|
module = Error(config=self.__config, module='error', error=str(e))
|
|
self.__widgets = [module.widget()]
|
|
self.update = module.update
|
|
|
|
def name(self):
|
|
return self.__name if self.__name else self.module_name()
|
|
|
|
def module_name(self):
|
|
return self.__module__.split('.')[-1]
|
|
|
|
def widgets(self, widgets=None):
|
|
if widgets:
|
|
self.__widgets = widgets
|
|
return self.__widgets
|
|
|
|
def widget(self, name=None):
|
|
if not name: return self.widgets()[0]
|
|
|
|
for w in self.widgets():
|
|
if w.name == name: return w
|
|
return None
|
|
|
|
def state(self, widget):
|
|
return []
|
|
|
|
def threshold_state(self, value, warn, crit):
|
|
if value > float(self.parameter('critical', crit)):
|
|
return 'critical'
|
|
if value > float(self.parameter('warning', warn)):
|
|
return 'warning'
|
|
return None
|
|
|
|
class Error(Module):
|
|
def __init__(self, module, error, config=core.config.Config([]), theme=None):
|
|
super().__init__(config, theme, core.widget.Widget(self.full_text))
|
|
self.__module = module
|
|
self.__error = error
|
|
|
|
self.set('scrolling.bounce', False)
|
|
self.set('scrolling.speed', 2)
|
|
self.set('width', 15)
|
|
|
|
@core.decorators.scrollable
|
|
def full_text(self, widget):
|
|
return '{}: {}'.format(self.__module, self.__error)
|
|
|
|
def state(self, widget):
|
|
return ['critical']
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|