app.py

from collections import namedtuple

from prometheus_client import start_http_server, Gauge

from apt_check import run
from apscheduler.schedulers.blocking import BlockingScheduler

Options = namedtuple('Options', ['security_updates_unattended', 'show_package_names', 'readable_output'])


scheduler = BlockingScheduler()

NUM_UPDATES = Gauge('apt_updates', 'Number of available apt updates')
NUM_SECURITY_UPDATES = Gauge('apt_security_updates', 'Number of available apt security updates')


def check_apt():
    options = Options(False, False, False)
    (num_updates, num_security_updates) = run(options)
    NUM_UPDATES.set(num_updates)
    NUM_SECURITY_UPDATES.set(num_security_updates)


def run_app():
    check_apt()

    # Start up the server to expose the metrics.
    start_http_server(8000)

    scheduler.add_job(check_apt, 'interval', hours=12)
    scheduler.start()