Update pacman.py

Use widget to store parameters instead of using private variables.
This commit is contained in:
yashar-sb-sb 2017-03-10 16:11:59 +03:30 committed by Yashar Shahi
parent bb2ed63528
commit 88e3b8c146

View file

@ -12,53 +12,49 @@ import bumblebee.input
import bumblebee.output import bumblebee.output
import bumblebee.engine import bumblebee.engine
#list of repositories the last one sould always be other
repos = ["community", "core", "extra", "other"]
class Module(bumblebee.engine.Module): class Module(bumblebee.engine.Module):
def __init__(self, engine, config): def __init__(self, engine, config):
super(Module, self).__init__(engine, config, super(Module, self).__init__(engine, config,
bumblebee.output.Widget(full_text=self.updates) bumblebee.output.Widget(full_text=self.updates)
) )
self._count = 0 self._count = 0
self._out = ""
def updates(self, widget): def updates(self, widget):
return self._out return '/'.join(map(lambda x: str(widget.get(x,0)), repos))
def update(self, widgets): def update(self, widgets):
path = os.path.dirname(os.path.abspath(__file__)) path = os.path.dirname(os.path.abspath(__file__))
if self._count == 0: if self._count == 0:
self._out = "?/?/?/?"
try: try:
result = bumblebee.util.execute("{}/../../bin/pacman-updates".format(path)) result = bumblebee.util.execute("{}/../../bin/pacman-updates".format(path))
self._community = 0
self._core = 0 count = len(repos)*[0]
self._extra = 0
self._other = 0
for line in result.splitlines(): for line in result.splitlines():
if line.startswith("http"): if line.startswith("http"):
if "community" in line: for i in range(len(repos)-1):
self._community += 1 if "/" + repos[i] + "/" in line:
continue count[i] += 1
if "core" in line: break
self._core += 1; else:
continue result[-1] += 1
if "extra" in line:
self._extra += 1 for i in range(len(repos)):
continue widgets[0].set(repos[i], count[i])
self._other += 1
self._out = str(self._core)+"/"+str(self._extra)+"/"+str(self._community)+"/"+str(self._other) except BaseException as a:
except RuntimeError: raise a
self._out = "?/?/?/?"
# TODO: improve this waiting mechanism a bit # TODO: improve this waiting mechanism a bit
self._count += 1 self._count += 1
self._count = 0 if self._count > 300 else self._count self._count = 0 if self._count > 300 else self._count
def sumUpdates(self):
return self._core + self._community + self._extra + self._other
def state(self, widget): def state(self, widget):
if self.sumUpdates() > 0: sumUpdates = sum(map(lambda x: widget.get(x,0), repos))
if sumUpdates > 0:
return "critical" return "critical"
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4