bumblebee-status/bumblebee_status/modules/contrib/dnf.py

# pylint: disable=C0111,R0903

"""Displays DNF package update information (<security>/<bugfixes>/<enhancements>/<other>)

Requires the following executable:
    * dnf

Parameters:
    * dnf.interval: Time in minutes between two consecutive update checks (defaults to 30 minutes)

"""

import threading

import core.event
import core.module
import core.widget
import core.decorators

import util.cli


def get_dnf_info(widget):
    res = util.cli.execute("dnf updateinfo", ignore_errors=True)

    security = 0
    bugfixes = 0
    enhancements = 0
    other = 0
    for line in res.split("\n"):
        if not line.startswith(" "):
            continue
        elif "ecurity" in line:
            for s in line.split():
                if s.isdigit():
                    security += int(s)
        elif "ugfix" in line:
            for s in line.split():
                if s.isdigit():
                    bugfixes += int(s)
        elif "hancement" in line:
            for s in line.split():
                if s.isdigit():
                    enhancements += int(s)
        else:
            for s in line.split():
                if s.isdigit():
                    other += int(s)

    widget.set("security", security)
    widget.set("bugfixes", bugfixes)
    widget.set("enhancements", enhancements)
    widget.set("other", other)

    core.event.trigger("update", [widget.module.id], redraw_only=True)


class Module(core.module.Module):
    @core.decorators.every(minutes=30)
    def __init__(self, config, theme):
        super().__init__(config, theme, core.widget.Widget(self.updates))

    def updates(self, widget):
        result = []
        for t in ["security", "bugfixes", "enhancements", "other"]:
            result.append(str(widget.get(t, 0)))
        return "/".join(result)

    def update(self):
        thread = threading.Thread(target=get_dnf_info, args=(self.widget(),))
        thread.start()

    def state(self, widget):
        cnt = 0
        for t in ["security", "bugfixes", "enhancements", "other"]:
            cnt += widget.get(t, 0)
        if cnt == 0:
            return "good"
        if cnt > 50 or widget.get("security", 0) > 0:
            return "critical"


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4