python脚本监控Linux磁盘空间使用率

脚本内容如下:

#cat check_snmp_storage.py

#!/usr/bin/python2.4
#_*_ coding: utf-8 _*_

'''
Last Modify: 2014/05/09
Version: 1.2
Description: Monitor Disk usage
Author: Victor
QQ:1409175531
'''

from sys import argv,exit
from decimal import Decimal
from optparse import OptionParser
from netsnmp import snmpwalk

parser = OptionParser('%prog -C  -H  -d  -w  -c ')

parser.add_option('-H', '--HostAddr', action='store', dest='hostaddress', help='Specify HostAddress')
parser.add_option('-C', '--community', action='store', dest='community', help='Specify host community')
parser.add_option('-d', '--device', action='store', dest='device', help='Specify a device')
parser.add_option('-w', '--warning', action='store', dest='warning', help='set the warning value')
parser.add_option('-c', '--critical', action='store', dest='critical', help='set the critical value')

(options, argvs) = parser.parse_args(argv)

desc_list = ['hrStorageDescr', 'hrStorageSize', 'hrStorageUsed', 'hrStorageAllocationUnits']

#依次获取desc_list对应的返回值，并分别存入列表
try:
    for mibdsc in desc_list:
        r1 = list(snmpwalk(mibdsc,Version=2,Community=options.community,DestHost=options.hostaddress))
        if mibdsc == 'hrStorageDescr':
            dev = r1
        elif mibdsc == 'hrStorageSize':
            size = r1
        elif mibdsc == 'hrStorageUsed':
            used = r1
        elif mibdsc == 'hrStorageAllocationUnits':
            units = r1
except TypeError:
    parser.print_help()
    exit(3)

size_list = {}
used_list = {}
per_list = {}
for devname in dev:
    l = dev.index(devname) #依次获取该元素在列表的位置
    dev_size = int(size[l]) * int(units[l]) / 1024 / 1024 #将磁盘容量单位转为M
    dev_used = int(used[l]) * int(units[l]) / 1024 / 1024
    used_percent = round(float(dev_used) / float(dev_size),2) * 100

    size_list[devname] = dev_size
    used_list[devname] = dev_used
    per_list[devname] = used_percent

if float(options.warning) > float(options.critical):
    print "-w <阀值> 必须等于或小于 -c <阀值>"
    exit(3)
try:
    if float(per_list[options.device]) < float(options.warning):
        print 'OK - "%s" 使用了:%s%% | 总容量:%s MB,已使用:%s MB (w:%s%%, c:%s%%)' % (options.device, per_list[options.device], size_list[options.device], used_list[options.device], options.warning, options.critical)
        exit(0)
    elif float(options.critical) > float(per_list[options.device]) >= float(options.warning):
        print 'Warning - "%s" 使用了:%s%% | 总容量:%s MB , 已使用:%s MB (w:%s%%, c:%s%%)' % (options.device, per_list[options.device], size_list[options.device], used_list[options.device], options.warning, options.critical)
        exit(1)
    elif float(per_list[options.device]) >= float(options.critical):
        print 'Critical - "%s" 使用了:%s%% | 总容量:%s MB , 已使用:%s MB (w:%s%%, c:%s%%)' % (options.device, per_list[options.device], size_list[options.device], used_list[options.device], options.warning, options.critical)
        exit(2)
except KeyError:
    print '无响应或指定的文件系统不存在'
    exit(3)

将脚本check_snmp_storage.py放到/usr/local/nagios/libexec目录下,给执行权限。直接执行脚本可以看到脚本的用法：

#python check_snmp_storage.py -h

usage: check_snmp_storage.py -C -H -d -w -c

options:

  -h, --help            show this help message and exit

  -H HOSTADDRESS, --HostAddr=HOSTADDRESS

                        Specify HostAddress

  -C COMMUNITY, --community=COMMUNITY

                        Specify host community

  -d DEVICE, --device=DEVICE

                        Specify a device

  -w WARNING, --warning=WARNING

                        set the warning value

  -c CRITICAL, --critical=CRITICAL

                        set the critical value

     设备名，如“Physical memory”、“Swap space”、 “/usr/local”等，如果不确定设备名，可先使用snmpdf查看，如：

 需要与上图Description列的名字吻合，大小写必须一致，否则脚本通过snmp去取数据时会找不到设备。

    和是报警阀值，如80表示80%，90表示90%。

举例：

commands.cfg配置如下：

define command{

  command_name check_snmp_storage

  command_line $USER1$/check_snmp_storage.py $ARG1$ -H $HOSTADDRESS$ $ARG2$ $ARG3$ $ARG4$

 

}

services.cfg配置如下：

define service{

        use                             service02

        host_name                       test

        service_description             /data       

        check_command                   check_snmp_storage!-C public!-d /data!-w 85!-c 90

       }

以上完成后重启nagios服务，效果图如下：

本文乃何秋平原创文章，请勿转载。如须转载请详细标明转载出处。