Munin contributed plugins

	#!/usr/bin/python

#

#    Copyright (C) 2011  Andreas Thienemann <andreas@bawue.net>

#

#    This program is free software: you can redistribute it and/or modify

#    it under the terms of the GNU General Public License as published by

#    the Free Software Foundation, either version 3 of the License, or

#    (at your option) any later version.

#

#    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 General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with this program.  If not, see <http://www.gnu.org/licenses/>.

#

"""

=head1 NAME

smart_raw__ - Munin plugin to retreive raw SMART values from a disk.

=head1 APPLICABLE SYSTEMS

All machines with a SMART capable disk and smartmontools installed.

This plugin is very useful if you want to need to monitor a single raw S.M.A.R.T.

value reported by a disk. Load Cycle Counts or Pending Sectors come to mind as

these are a good indicator of problems with a disk.

=head1 CONFIGURATION

The plugin should be installed as smart_raw_sda_193 which means that the smart value

numbered 193 will be read from the disk sda.

Basic configuration for every system is that the plugin needs to be called as root

in order to execute smartmontools.

Add the following to your /etc/munin/plugin-conf.d/smart_raw:

 [smart_raw_sda_193]

 user root

=head1 INTERPRETATION

Smart RAW values are provided as is. You need to undertand what you are monitoring...

=head1 MAGIC MARKERS

  #%# family=contrib

  #%# capabilities=

=head1 VERSION

0.0.1

=head1 BUGS

None known.

=head1 AUTHOR

Andreas Thienemann <andreas@bawue.net>

=head1 LICENSE

GPLv3+

=cut

"""

import subprocess

import sys

import os

import re

import pprint

# We are a wildcard plugin, figure out what we are being called for

try:

	disk = sys.argv[0].split('_')[2]

	value = sys.argv[0].split('_')[3]

except IndexError:

	sys.exit(1)

def normalize_name(name):

    name = re.sub("[^a-z0-9A-Z]","_", name)

    return name

# Code sniplet from Philipp Keller

# http://code.pui.ch/2007/02/19/set-timeout-for-a-shell-command-in-python/

def timeout_command(command, timeout):

    """call shell-command and either return its output or kill it

    if it doesn't normally exit within timeout seconds and return None"""

    import subprocess, datetime, os, time, signal

    start = datetime.datetime.now()

    process = subprocess.Popen(command.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    while process.poll() is None:

        time.sleep(0.1)

        now = datetime.datetime.now()

        if (now - start).seconds> timeout:

            os.kill(process.pid, signal.SIGKILL)

            os.waitpid(-1, os.WNOHANG)

            return None

    return process.stdout.read()

def read_smart():

    """Read SMART attributes"""

    out = timeout_command("/usr/sbin/smartctl -A -d ata /dev/%s" % (disk), 2)

    smart_attribs = dict()

    for line in out.split('\n')[7:-2]: # Cut away the header and the footer

        line = line.split()

        smart_attribs[line[0]] = {

            'name' : line[1],

            'label' : normalize_name(line[1]),

            'raw' : line [-1],

        }

    return smart_attribs

def print_config():

    """Return configuration arguments for munin"""

    attribs = read_smart()

    print "graph_title S.M.A.R.T raw value %s for drive %s" % (attribs[value]['name'], disk)

    print "graph_vlabel Raw value"

    print "graph_info RAW smartctl value"

    print "graph_category disk"

    print "graph_args --base 1000 -l 0"

    print "%s.label %s" % (value, attribs[value]['label'])

    sys.exit(0)

def fetch():

    attribs = read_smart()

    print "%s.value %s" % (value, attribs[value]['raw'])

    sys.exit(0)

if "config" in sys.argv[1:]:

    print_config()

elif "autoconf" in sys.argv[1:]:

    pass

elif "suggest" in sys.argv[1:]:

    pass

else:

    fetch()