[output] Allow modules to pass back a single widget

Since "single widget" is the 99% use-case, make it easier for a module
to return a single widget.
This commit is contained in:
Tobias Witek 2016-11-05 13:12:30 +01:00
parent 2cfb0997a0
commit 286aff2aa0
5 changed files with 9 additions and 14 deletions

View file

@ -22,9 +22,7 @@ class Module(bumblebee.module.Module):
self._capacity = int(f.read())
self._capacity = self._capacity if self._capacity < 100 else 100
return [
bumblebee.output.Widget(self,"{:02d}%".format(self._capacity))
]
return bumblebee.output.Widget(self,"{:02d}%".format(self._capacity))
def warning(self):
return self._capacity < self._config.parameter("battery.warning", 20)

View file

@ -21,11 +21,7 @@ class Module(bumblebee.module.Module):
def widgets(self):
self._perc = psutil.cpu_percent(percpu=False)
return [
bumblebee.output.Widget(self,
"{:05.02f}%".format(self._perc)
)
]
return bumblebee.output.Widget(self, "{:05.02f}%".format(self._perc))
def warning(self):
return self._perc > self._config.parameter("cpu.warning", 70)

View file

@ -32,10 +32,6 @@ class Module(bumblebee.module.Module):
self._fmt = config.parameter(param_name, default)
def widgets(self):
return [
bumblebee.output.Widget(self,
datetime.datetime.now().strftime(self._fmt)
)
]
return bumblebee.output.Widget(self, datetime.datetime.now().strftime(self._fmt))
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

View file

@ -66,6 +66,11 @@ class Output(object):
pass
def draw(self, widgets, theme):
if not type(widgets) is list:
widgets = [ widgets ]
self._draw(widgets, theme)
def _draw(self, widgets, theme):
pass
def flush(self):

View file

@ -42,7 +42,7 @@ class Output(bumblebee.output.Output):
def start(self):
print json.dumps({ "version": 1, "click_events": True }) + "["
def draw(self, widgets, theme):
def _draw(self, widgets, theme):
for widget in widgets:
if theme.separator(widget):
self._data.append({