82502c5651
While at it, remove dependency on "parse" module and replace with more standard "re" dependency.
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays APT package update information (<to upgrade>/<to remove >)
|
|
Requires the following packages:
|
|
* aptitude
|
|
|
|
"""
|
|
|
|
import re
|
|
import threading
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.decorators
|
|
|
|
import util.cli
|
|
|
|
PATTERN = '{} packages upgraded, {} newly installed, {} to remove and {} not upgraded.'
|
|
|
|
def parse_result(to_parse):
|
|
# We want to line with the iforamtion about package upgrade
|
|
line_to_parse = to_parse.split('\n')[-4]
|
|
result = re.search('(.+) packages upgraded, (.+) newly installed, (.+) to remove', line_to_parse)
|
|
|
|
return int(result.group(1)), int(result.group(3))
|
|
|
|
def get_apt_check_info(module):
|
|
widget = module.widget()
|
|
try:
|
|
res = util.cli.execute('aptitude full-upgrade --simulate --assume-yes')
|
|
widget.set('error', None)
|
|
except (RuntimeError, FileNotFoundError) as e:
|
|
widget.set('error', 'unable to query APT: {}'.format(e))
|
|
return
|
|
|
|
to_upgrade = 0
|
|
to_remove = 0
|
|
try:
|
|
to_upgrade, to_remove = parse_result(res)
|
|
widget.set('to_upgrade', to_upgrade)
|
|
widget.set('to_remove', to_remove)
|
|
except Exception as e:
|
|
widget.set('error', 'parse error: {}'.format(e))
|
|
|
|
core.event.trigger('update', [ module.id ], redraw_only=True)
|
|
|
|
class Module(core.module.Module):
|
|
@core.decorators.every(minutes=30)
|
|
def __init__(self, config):
|
|
super().__init__(config, core.widget.Widget(self.updates))
|
|
self.__thread = None
|
|
|
|
def updates(self, widget):
|
|
if widget.get('error'):
|
|
return widget.get('error')
|
|
return '{} to upgrade, {} to remove'.format(
|
|
widget.get('to_upgrade', 0), widget.get('to_remove', 0)
|
|
)
|
|
|
|
def update(self):
|
|
if self.__thread and self.__thread.isAlive(): return
|
|
|
|
self.__thread = threading.Thread(target=get_apt_check_info, args=(self,))
|
|
self.__thread.start()
|
|
|
|
def state(self, widget):
|
|
cnt = 0
|
|
ret = 'good'
|
|
for t in ['to_upgrade', 'to_remove']:
|
|
cnt += widget.get(t, 0)
|
|
if cnt > 50:
|
|
ret = 'critical'
|
|
elif cnt > 0:
|
|
ret = 'warning'
|
|
if widget.get('error'):
|
|
ret = 'critical'
|
|
|
|
return ret
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|