Munin contributed plugins

#!/usr/bin/env python3

"""

=head1 NAME

exodus_ - Exodus wildcard-plugin to monitor an L<Exodus|https://github.com/Exodus-Privacy/exodus/>

instance.

This wildcard plugin provides the suffixes C<applications>, C<reports> and C<trackers>.

=head1 INSTALLATION

- Copy this plugin in your munin plugins directory

=over 2

    ln -s /usr/share/munin/plugins/exodus_ /etc/munin/plugins/exodus_applications

    ln -s /usr/share/munin/plugins/exodus_ /etc/munin/plugins/exodus_reports

    ln -s /usr/share/munin/plugins/exodus_ /etc/munin/plugins/exodus_trackers

=back

After the installation you need to restart your munin-node service.

=head1 CONFIGURATION

You need to create a file named exodus placed in the directory

/etc/munin/plugin-conf.d/ with the following config:

=over 2

    [exodus_*]

    env.exodus_url https://reports.exodus-privacy.eu.org

    env.exodus_token your-api-token

=back

=head1 AUTHORS

Codimp <contact@lithio.fr>

=head1 LICENSE

GPLv3

=head1 MAGIC MARKERS

 #%# family=manual

 #%# capabilities=suggest

=cut

"""

import os

import sys

import json

import urllib.request

def print_count(element):

    endpoint = os.getenv("exodus_url") + "/api/" + element + "/count"

    headers = {"Authorization": "Token " + os.getenv("exodus_token")}

    request = urllib.request.Request(endpoint, method="POST", headers=headers)

    with urllib.request.urlopen(request) as connection:

        if connection.getcode() == 200:

            count = json.loads(connection.read())["count"]

            print(element + ".value", count)

def main():

    try:

        mode = sys.argv[1]

    except IndexError:

        mode = ""

    wildcard = sys.argv[0].split("exodus_")[1]

    if mode == "suggest":

        print("applications")

        print("reports")

        print("trackers")

        exit(0)

    if wildcard == "applications":

        if mode == "config":

            print("graph_title Exodus applications")

            print("graph_vlabel applications")

            print("graph_category exodus")

            print("applications.label Applications")

        else:

            print_count("applications")

    elif wildcard == "reports":

        if mode == "config":

            print("graph_title Exodus reports")

            print("graph_vlabel reports")

            print("graph_category exodus")

            print("reports.label Reports")

        else:

            print_count("reports")

    elif wildcard == "trackers":

        if mode == "config":

            print("graph_title Exodus trackers")

            print("graph_vlabel trackers")

            print("graph_category exodus")

            print("trackers.label Trackers")

        else:

            print_count("trackers")

if __name__ == "__main__":

    main()