1
0
mirror of https://github.com/opinkerfi/nagios-plugins.git synced 2024-12-24 18:13:55 +01:00
nagios-plugins/check_cpu.py/okplugin_check_cpu
2014-01-20 11:41:11 +00:00

95 lines
2.8 KiB
Python

#!/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