人气旺 发表于 2014-9-28 09:19:42

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
    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()
页: [1]
查看完整版本: nagios-plugin(mem)