92 lines
2.7 KiB
Python
92 lines
2.7 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
|
|
stat_length = len(old_stat) > len(stat) and len(old_stat) or 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
|