Munin contributed plugins

#!/usr/bin/env python

"""=cut

=head1 NAME 

 imapproxy  - Munin Plugin to monitor imapproxy using pimpstat

=head1 CONFIGURATION

 This plugin should require no addition configuration.

=head1 MAGIC MARKERS

  #%# family=auto

  #%# capabilities=autoconf suggest

=head1 Author

Niall Donegan <github@nialldonegan.me>

=head1 LICENSE

GPLv2

=cut"""

import sys

import os

import re

from subprocess import Popen,PIPE

def print_autoconf():

    which = Popen("which pimpstat", shell=True, stdout=PIPE)

    which.communicate()

    if not bool(which.returncode):

        print "yes"

    else:

        print "no (pimpstat not found)"

    sys.exit(0)

def print_config(graph):

    if graph == "cache":

        print "graph_title Cache Statistics For ImapProxy"

        print "graph_args -l 0 --base 1000"

        print "graph_total total"

        print "graph_vlabel Cache Connections / ${graph_period}"

        print "graph_category imapproxy"

        print "cache_hits.draw AREA"

        print "cache_hits.type DERIVE"

        print "cache_hits.label Cache Hits"

        print "cache_hits.min 0"

        print "cache_misses.draw STACK"

        print "cache_misses.type DERIVE"

        print "cache_misses.label Cache Misses"

        print "cache_misses.min 0"

    if graph == "connections":

        print "graph_title Connection Statistics For ImapProxy"

        print "graph_args -l 0 --base 1000"

        print "graph_total total"

        print "graph_vlabel Connections / ${graph_period}"

        print "graph_category imapproxy"

        print "connections_reused.draw AREA"

        print "connections_reused.type DERIVE"

        print "connections_reused.label Reused Connections"

        print "connections_reused.min 0"

        print "connections_created.draw STACK"

        print "connections_created.type DERIVE"

        print "connections_created.label Created Connections"

        print "connections_created.min 0"

    sys.exit(0)

def print_suggest():

    print "cache"

    print "connections"

    sys.exit(0)

def print_fetch(graph):

    if graph == "cache":

        hits = 0

        misses = 0

        cache = Popen(

            "pimpstat -c | egrep 'Cache (Hits|Misses)'", 

            shell=True,

            stdout=PIPE

        )

        for line in cache.stdout:

            if re.search(r'Hits', line):

                hits = line.split()[0]

            if re.search(r'Misses', line):

                misses = line.split()[0]

        print "cache_hits.value %s" % hits

        print "cache_misses.value %s" % misses

    if graph == "connections":

        created = 0

        reused = 0

        connections = Popen(

            "pimpstat -c | egrep 'Total (Reused|Created)'", 

            shell=True,

            stdout=PIPE

        )

        for line in connections.stdout:

            if re.search(r'Created', line):

                created = line.split()[0]

            if re.search(r'Reused', line):

                reused = line.split()[0]

        print "connections_created.value %s" % created

        print "connections_reused.value %s" % resused

    sys.exit(0)

def main():

    basename = "imapproxy_"

    calledname = os.path.basename(sys.argv[0])

    if len(sys.argv) > 1:

        command = sys.argv[1]

    else:

        command = "fetch"

    if calledname == basename and command == "fetch":

        print >> sys.stderr, "Please use either suggest or autoconf"

        sys.exit(1)

    if calledname == basename and command not in ["autoconf","suggest"]:

        print >> sys.stderr, "Command %s not known, please use either autoconf or suggest" % command

        sys.exit(1)

    graph = calledname[calledname.find("_")+1:]

    if calledname != basename and graph not in ["cache","connections"]:

        print >> sys.stderr, "%s is not a known graph" % graph

        sys.exit(1)

    if calledname != basename and command not in ["config", "fetch"]:

        print >> sys.stderr, "Command %s not known, please use either config or fetch" % command

        sys.exit(1)

    if command == "autoconf":

        print_autoconf()

    elif command == "config":

        print_config(graph)

    elif command =="suggest":

        print_suggest()

    else:

        print_fetch(graph)

if __name__ == '__main__':

    main()

# vim:syntax=python