210075fdf9
If querying APT fails for some reason, show an error message and set the state of the widget to critical. fixes #395
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays APT package update information (<toupgrade>/<security>)
|
|
Requires the following debian packages:
|
|
* update-notifier-common
|
|
|
|
"""
|
|
|
|
import threading
|
|
|
|
import bumblebee.util
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
|
|
APT_CHECK_PATH = "/usr/lib/update-notifier/apt_check.py"
|
|
|
|
def get_apt_check_info(widget):
|
|
try:
|
|
res = bumblebee.util.execute(APT_CHECK_PATH)
|
|
widget.set("error", None)
|
|
except (RuntimeError, FileNotFoundError) as e:
|
|
widget.set("error", "unable to query APT: {}".format(e))
|
|
return
|
|
|
|
all_pkg = 0
|
|
security = 0
|
|
|
|
res_array = res.split(';')
|
|
|
|
try:
|
|
s = res_array[0]
|
|
if s.isdigit(): all_pkg = int(s)
|
|
|
|
s = res_array[1]
|
|
if s.isdigit(): security = int(s)
|
|
except:
|
|
pass
|
|
|
|
widget.set("all_pkg", all_pkg)
|
|
widget.set("security", security)
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
widget = bumblebee.output.Widget(full_text=self.updates)
|
|
super(Module, self).__init__(engine, config, widget)
|
|
self.interval(30)
|
|
|
|
def updates(self, widget):
|
|
result = []
|
|
if widget.get("error"):
|
|
return widget.get("error")
|
|
for t in ["all_pkg", "security"]:
|
|
result.append(str(widget.get(t, 0)))
|
|
return "/".join(result)
|
|
|
|
def update(self, widgets):
|
|
thread = threading.Thread(target=get_apt_check_info, args=(widgets[0],))
|
|
thread.start()
|
|
|
|
def state(self, widget):
|
|
cnt = 0
|
|
ret = "good"
|
|
for t in ["all_pkg", "security"]:
|
|
cnt += widget.get(t, 0)
|
|
if cnt > 50 or widget.get("security", 0) > 0:
|
|
ret = "critical"
|
|
elif cnt > 0:
|
|
ret = "warning"
|
|
if widget.get("error"):
|
|
ret = "critical"
|
|
|
|
return ret
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|