# 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 bumblebee.util
import bumblebee.input
import bumblebee.output
import bumblebee.engine

def get_dnf_info(widget):
    try:
        res = bumblebee.util.execute('dnf updateinfo')
    except RuntimeError:
        pass

    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)

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_factor(60)
        self.interval(30)

    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, widgets):
        thread = threading.Thread(target=get_dnf_info, args=(widgets[0],))
        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