#!/usr/bin/env python

# Copyright (c) 2008, Net Easy, Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of Net Easy, Inc. nor the
#       names of its contributors may be used to endorse or promote products
#       derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY Net Easy, Inc. ''AS IS'' AND ANY
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL Net Easy, Inc. BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

import re, os
from sys import argv

class checker(object):
    def __init__(self):
        'parsed_data will have a tuple of the percentage of disk usage and the mount point for each disk partition'
        self.parsed_data = {}

    def __repr__(self):
        thisdata = ''
        for item in self.parsed_data.keys():
            thisdata = '%s%s.value %s\n' % (thisdata, item, self.parsed_data[item][0])
        return thisdata

    def config(self):
        thisdata = """graph_title Filesystem usage (in %)
graph_args --lower-limit 0
graph_vlabel %
graph_category disk
graph_info Filesystem usage
"""
        for item in self.parsed_data.keys():
            thisdata = "%s%s.label %s\n%s.warning 92\n%s.critical 98\n" % (thisdata, item, self.parsed_data[item][1], item, item)
        return thisdata

    def get_data(self):
        rawdata = os.popen('df -P -l').readlines()
        for i in range(1,len(rawdata)):
            dataline=rawdata[i].split()
            self.parsed_data[re.sub('/', '_', dataline[0])] = (re.sub('%', '', dataline[4]), dataline[5])

if __name__ == "__main__":
    processor = checker()
    processor.get_data()
    if len(argv) > 1 and argv[1] == 'config':
        print processor.config()
    else:
        print processor