121 lines
3 KiB
Python
121 lines
3 KiB
Python
import os
|
|
import inspect
|
|
import threading
|
|
import bumblebee.util
|
|
|
|
def output(args):
|
|
import bumblebee.outputs.i3
|
|
return bumblebee.outputs.i3.Output(args)
|
|
|
|
class Widget(object):
|
|
def __init__(self, obj, text, instance=None):
|
|
self._obj = obj
|
|
self._text = text
|
|
self._store = {}
|
|
self._instance = instance
|
|
|
|
obj._output.register_widget(self.instance(), self)
|
|
|
|
def set(self, key, value):
|
|
self._store[key] = value
|
|
|
|
def get(self, key, default=None):
|
|
return self._store.get(key, default)
|
|
|
|
def state(self):
|
|
return self._obj.state(self)
|
|
|
|
def warning(self):
|
|
return self._obj.warning(self)
|
|
|
|
def critical(self):
|
|
return self._obj.critical(self)
|
|
|
|
def module(self):
|
|
return self._obj.__module__.split(".")[-1]
|
|
|
|
def instance(self):
|
|
return self._instance if self._instance else getattr(self._obj, "instance")(self)
|
|
|
|
def text(self):
|
|
return self._text
|
|
|
|
class Command(object):
|
|
def __init__(self, command, event, widget):
|
|
self._command = command
|
|
self._event = event
|
|
self._widget = widget
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
if not isinstance(self._command, list):
|
|
self._command = [ self._command ]
|
|
|
|
for cmd in self._command:
|
|
if not cmd: continue
|
|
if inspect.ismethod(cmd):
|
|
cmd(self._event, self._widget)
|
|
else:
|
|
c = cmd.format(*args, **kwargs)
|
|
bumblebee.util.execute(c, False)
|
|
|
|
class Output(object):
|
|
def __init__(self, config):
|
|
self._config = config
|
|
self._callbacks = {}
|
|
self._wait = threading.Condition()
|
|
self._wait.acquire()
|
|
self._widgets = {}
|
|
|
|
def register_widget(self, identity, widget):
|
|
self._widgets[identity] = widget
|
|
|
|
def redraw(self):
|
|
self._wait.acquire()
|
|
self._wait.notify()
|
|
self._wait.release()
|
|
|
|
def add_callback(self, cmd, button, module=None):
|
|
if module:
|
|
module = module.replace("bumblebee.modules.", "")
|
|
|
|
if self._callbacks.get((button, module)): return
|
|
|
|
self._callbacks[(
|
|
button,
|
|
module,
|
|
)] = cmd
|
|
|
|
def callback(self, event):
|
|
cb = self._callbacks.get((
|
|
event.get("button", -1),
|
|
None,
|
|
), None)
|
|
cb = self._callbacks.get((
|
|
event.get("button", -1),
|
|
event.get("instance", event.get("module", None)),
|
|
), cb)
|
|
|
|
identity = event.get("instance", event.get("module", None))
|
|
return Command(cb, event, self._widgets.get(identity, None))
|
|
|
|
def wait(self):
|
|
self._wait.wait(self._config.parameter("interval", 1))
|
|
|
|
def draw(self, widgets, theme):
|
|
if not type(widgets) is list:
|
|
widgets = [ widgets ]
|
|
self._draw(widgets, theme)
|
|
|
|
def start(self):
|
|
pass
|
|
|
|
def _draw(self, widgets, theme):
|
|
pass
|
|
|
|
def flush(self):
|
|
pass
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|