linux-cpu-usage.py

#!/usr/bin/env python
import sys
import time
from optparse import OptionParser

###  Global Identifiers  ###
cpu_stat_var_array = ('user', 'nice', 'system', 'idle', 'iowait', 'irq', 'softirq', 'steal_time') 

###   Main code   ###
# Command Line Arguments Parser
cmd_parser = OptionParser(version="%prog 0.1")
cmd_parser.add_option("-C", "--CPU", action="store", type="string", dest="cpu_name", help="Which CPU to be Check", metavar="cpu or cpu0 or cpu1")
cmd_parser.add_option("-w", "--warning", type="int", action="store", dest="warning_per", help="Exit with WARNING status if higher than the PERCENT of CPU Usage", metavar="Warning Percentage")
cmd_parser.add_option("-c", "--critical", type="int", action="store", dest="critical_per", help="Exit with CRITICAL status if higher than the PERCENT of CPU Usage", metavar="Critical Percentage")
cmd_parser.add_option("-d", "--debug", action="store_true", dest="debug", default=False, help="enable debug")

(cmd_options, cmd_args) = cmd_parser.parse_args()
# Check the Command syntax
if not (cmd_options.cpu_name and cmd_options.warning_per and cmd_options.critical_per):
    cmd_parser.print_help()
    sys.exit(3)

# Collect CPU Statistic Object 
class CollectStat:
    """Object to Collect CPU Statistic Data"""
    def __init__(self,cpu_name):
        
        self.total = 0 
        self.cpu_stat_dict = {}
        
        for line in open("/proc/stat"):
            line = line.strip()
        
            if line.startswith(cpu_name):
                cpustat=line.split()
                cpustat.pop(0)              # Remove the First Array of the Line 'cpu'

                # Remove the unwanted data from the array
                # only retain first 8 field on the file
                while len(cpustat) > 8:
                    cpustat.pop()

                if cmd_options.debug:
                    print "DEBUG : cpustat array %s" % cpustat
                        
                cpustat=map(float, cpustat)     # Convert the Array to Float

                for i in range(len(cpustat)):
                    self.cpu_stat_dict[cpu_stat_var_array[i]] = cpustat[i]

                # Calculate the total utilization
                for i in cpustat:
                    self.total += i 

                break

        if cmd_options.debug:
            print "DEBUG : cpu statistic dictionary %s" % self.cpu_stat_dict
            print "DEBUG : total statistics %s" % self.total

# Get Sample CPU Statistics 
initial_stat = CollectStat(cmd_options.cpu_name)
time.sleep(5)
final_stat = CollectStat(cmd_options.cpu_name)

cpu_total_stat = final_stat.total - initial_stat.total

if cmd_options.debug:
    print "DEBUG : diff total stat %f" % cpu_total_stat

for cpu_stat_var,cpu_stat in final_stat.cpu_stat_dict.items():
    globals()['cpu_%s_usage_percent' % cpu_stat_var] = ((final_stat.cpu_stat_dict[cpu_stat_var] - initial_stat.cpu_stat_dict[cpu_stat_var])/cpu_total_stat)*100  

cpu_usage_percent = cpu_user_usage_percent + cpu_nice_usage_percent + cpu_system_usage_percent + cpu_iowait_usage_percent + cpu_irq_usage_percent + cpu_softirq_usage_percent + cpu_steal_time_usage_percent

# Check if CPU Usage is Critical/Warning/OK
if cpu_usage_percent >= cmd_options.critical_per:
    print cmd_options.cpu_name +' STATISTICS CRITICAL : user=%.2f%% system=%.2f%% iowait=%.2f%% steal=%.2f%% | user=%.2f system=%.2f iowait=%.2f steal=%.2f warn=%d crit=%d' % (cpu_user_usage_percent, cpu_system_usage_percent, cpu_iowait_usage_percent, cpu_steal_time_usage_percent, cpu_user_usage_percent, cpu_system_usage_percent, cpu_iowait_usage_percent, cpu_steal_time_usage_percent, cmd_options.warning_per, cmd_options.critical_per)
    sys.exit(2)
elif  cpu_usage_percent >= cmd_options.warning_per:
    print cmd_options.cpu_name +' STATISTICS WARNING : user=%.2f%% system=%.2f%% iowait=%.2f%% steal=%.2f%% | user=%.2f system=%.2f iowait=%.2f steal=%.2f warn=%d crit=%d' % (cpu_user_usage_percent, cpu_system_usage_percent, cpu_iowait_usage_percent, cpu_steal_time_usage_percent, cpu_user_usage_percent, cpu_system_usage_percent, cpu_iowait_usage_percent, cpu_steal_time_usage_percent, cmd_options.warning_per, cmd_options.critical_per)
    sys.exit(1)
else:
    print cmd_options.cpu_name +' STATISTICS OK : user=%.2f%% system=%.2f%% iowait=%.2f%% steal=%.2f%% | user=%.2f system=%.2f iowait=%.2f steal=%.2f warn=%d crit=%d' % (cpu_user_usage_percent, cpu_system_usage_percent, cpu_iowait_usage_percent, cpu_steal_time_usage_percent, cpu_user_usage_percent, cpu_system_usage_percent, cpu_iowait_usage_percent, cpu_steal_time_usage_percent, cmd_options.warning_per, cmd_options.critical_per)
    sys.exit(0)