#!/usr/bin/python
#
# Copyright 2014, Tomas Edwardsson 
#
# 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/>.
"""
Gathers information on cpu usage on a linux machine
"""

from pynag.Plugins import PluginHelper, ok, unknown


def main():
    """The main logic of the program"""

    # See man proc and search for /proc/stat
    stat_fields = ['cpu', 'user', 'nice', 'system', 'idle', 'iowait', 'irq',
                   'softirq', 'steal', 'guest', 'guest_nice']

    pyplug = PluginHelper()

    old_stat = get_saved_stat()
    stat = get_stat()

    # Find shortest stat
    if old_stat:
        stat_length = len(old_stat) > len(stat) and len(old_stat) or len(stat)
    else:
        stat_length = len(stat)

    if old_stat:
        diff = []
        for pos in range(stat_length):
            diff.append(stat[pos] - old_stat[pos])
        averages =  calc_avg(diff)
    else:
        averages =  calc_avg(stat)

    for pos in range(stat_length):
        pyplug.add_metric(label=stat_fields[pos+1], value=averages[pos],
                          uom="%")

    try:
        save_stat(stat)
    except IOError, err:
        pyplug.exit(unknown, "Unable to save cpu statistics: %s" % err)

    pyplug.add_status(ok)
    pyplug.add_summary("CPU load %.2f%%" % (100.0 - averages[3]))
    pyplug.exit()

def calc_avg(stat):
    """Calculates the average of each int in array"""
    stat_sum = sum(stat)
    avg = []
    for pos in range(len(stat)):
        avg.append(float(stat[pos]) / stat_sum * 100)
    return avg

def get_stat():
    """Fetches the first line from /proc/stat and returns the numbers"""
    stat_fh = open("/proc/stat")
    return [int(pos) for pos in stat_fh.readline().split()[1:]]

def get_saved_stat():
    """Fetches information about the last run to be able to do averages"""
    try:
        stat_fh = open("/var/lib/nagios-okplugin/check_cpu")
    except IOError:
        return None
    return [int(pos) for pos in stat_fh.readline().split()[1:]]

def save_stat(stat):
    """Save statistics to compare to in next run"""
    stat_fh = open("/var/lib/nagios-okplugin/check_cpu", "w")

    stat_fh.write("cpu " + " ".join([str(pos) for pos in stat]) + "\n")
    stat_fh.close()

if __name__ == "__main__":
    main()

# vim: sts=4 expandtab