nagios-plugin(mem)

人气旺

#!/usr/bin/env python

import sys
from optparse import OptionParser
import string

OK=0
WARNING=1
CRITICAL=2
UNKNOWN=3

unit = {'t':2**40,'g':2**30,'m':2**20,'k':2**10,'b':1}

def opt():
    parser = OptionParser(usage="usage: %prog -w WARNING -c CRITICAL")
    parser.add_option("-c", default="100M", action="store", type="string", dest="critical")
    parser.add_option("-w", default="500M", action="store", type="string", dest="warning")
    return parser.parse_args()


def convertUnit(s):
    s = s.lower()
    lastchar = s[-1]
    num = int(s[:-1])
    if lastchar in unit:
        return num*unit[lastchar]
    else:
        return int(s)

def scaleUnit(d):
    for k,v in unit.items():
        num = d / float(v)
        if (0 < num < 2**10):
            return num,k

def getFreeMemory():
    with open('/proc/meminfo','r') as fd:
        for line in fd.readlines():
            if line.startswith('MemFree'):
                k, v, u =  line.split()
                return int(v)*1024

def main():
    opts, args = opt()

    w = convertUnit(opts.warning)
    c = convertUnit(opts.critical)

    free_mem =  getFreeMemory()
     
    h_free, h_unit = scaleUnit(free_mem)

    if free_mem > w:
        print "OK, free:%s %s" %  (h_free, h_unit)
        sys.exit(OK)
    elif w >= free_mem > c:
        print "WARNING, free: %s %s" % (h_free, h_unit)
        sys.exit(WARNING)
    elif free_mem <= c:
        print "CRITICAL, free: %s %s" % (h_free, h_unit)
        sys.exit(CRITICAL)
    else:
        print "UNKNONW, free:%s %s" %  (h_free, h_unit)
        sys.exit(UNKNOWN)

if __name__ == "__main__":
    main()