root/bacula_job

#!/usr/bin/python

# Copyright (C) 2009 Andreas Thienemann <andreas@bawue.net>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Library General Public License as published by
# the Free Software Foundation; version 2 only
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU Library General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#

#
# Munin Plugin to get job throughput for Bacula by parsing the bconsole
# output.
#
# Parameters:
#
#       config   (required)
#       autoconf (optional - only used by munin-config)
#

# Magic markers (optional - only used by munin-config and some
# installation scripts):
#
#%# family=contrib
#%# capabilities=autoconf

import subprocess
import time
import sys
import re
import os

def parse_running_jobs():
        """ Parse the bconsole output once to get the running jobs """

        bconsole = subprocess.Popen("bconsole", stdin=subprocess.PIPE, stdout=subprocess.PIPE)
        stdout, stderr = bconsole.communicate("status\n1\nstatus\n3\n.")

        jobs = []
        clients = []
        clientlist = False

        # Hold the line numbers for devices
        dev_line = []
        input = stdout.split("\n")

        for line, i in zip(input, range(0, len(input))):
                if line.startswith("Connecting to Director "):
                        hostname = line.split()[-1].split(":")[0]

                if line.endswith(" is running"):
                        jobs.append(line.split()[2].split(".")[0])

                # Parse the clientlist, warning, order of statements is important
                if line.startswith("Select Client (File daemon) resource"):
                        clientlist = False

                if clientlist is True:
                        client_id, client_name = line.split()
                        client_clean = re.sub("^[^A-Za-z_]", "_", client_name, 1)
                        client_clean = re.sub("[^A-Za-z0-9_]", "_", client_clean, 0)
                        clients.append((client_name, client_clean, client_id[:-1]))

                if line.startswith("The defined Client resources are:"):
                        clientlist = True

        return hostname, jobs, clients


def parse(clients):
        """ Parse the bconsole output """

        query_str = ""
        for client in clients:
                query_str = query_str + "status\n3\n" + client[1] + "\n"
        query_str = query_str + "quit"

        bconsole = subprocess.Popen("bconsole", stdin=subprocess.PIPE, stdout=subprocess.PIPE)
        stdout, stderr = bconsole.communicate(query_str)

        input = stdout.split("\n")

        jobstats = []

        for line, pos in zip(input, range(0, len(input))):

                # Get the client name
                if line.startswith("Connecting to Client "):
                        # client_name = input[pos].split()[3].split(".")[0]
                        client_name = line.split()[3]
                        client_clean = re.sub("^[^A-Za-z_]", "_", client_name, 1)
                        client_clean = re.sub("[^A-Za-z0-9_]", "_", client_clean, 0)

                # Get the current bytes
                if line.endswith(" is running."):
                        bytes = long(input[pos+2].split()[1].split("=")[1].replace(",", ""))
                        jobstats.append([client_name, client_clean, bytes])

        job_dict = {}
        for job in jobstats:
                job_dict[job[0].split("-")[0]] = job

        return job_dict


def print_config():
        hostname, jobs, clients = parse_running_jobs()
        print "graph_title Bacula Job throughput"
        print "graph_vlabel bytes per ${graph_period}"
        print "graph_args --base 1024 -l 0"
        print "graph_scale yes"
        print "graph_info Bacula Job measurement."
        print "graph_category Bacula"
        print "graph_order", 
        for fd in clients:
                print fd,
        print
        if os.getenv("report_hostname") is not None and \
                os.getenv("report_hostname").upper() in ["YES", "TRUE", "1", "Y"]:
                print "host_name", hostname
        for client in clients:
                print "%s.label %s" % (client[1], client[0])
                print "%s.type DERIVE" % (client[1])
                print "%s.min 0" % (client[1])
#               print "%s.max %s" % (client[1], str(1024*1024*1024*16))
#               print "%s.cdef up,8,*" (client[1])
        sys.exit(0)


if "config" in sys.argv[1:]:
        print_config()
elif "autoconf" in sys.argv[1:]:
        for dir in os.getenv("PATH").split(":"): 
                for root, dirs, files in os.walk(dir):
                        if "bconsole" in files:
                                print "yes"
                                sys.exit(0)
        print "no"
        sys.exit(1)
elif "suggest" in sys.argv[1:]:
        sys.exit(1)
else:
        hostname, jobs, clients = parse_running_jobs()
        str = []
        for client in clients:
                if client[0].split("-")[0] in jobs:
                        str.append((client[0], client[2]))

        client_values = parse(str)

        for client in clients:
                client_name_short =  client[0].split("-")[0]
                if client_name_short in client_values:
                        print "%s.value %s" % (client_values[client_name_short][1], client_values[client_name_short][2])
                else:
                        print "%s.value %s" % (client[1], "0")

        sys.exit(0)