switchcat.py

#!/usr/bin/env python
#encoding=utf-8

import subprocess
import time
import urllib2

monitor = [{'group':'switch',
            'type':'cisco',
            'public':'dianpingP@ssword',
            'version':'2c',
            'ip':'172.25.21.88',
            'port':[10101,10102,10601,10602]
          }]

CISCO_IN_TRAFFIC_OID = '.1.3.6.1.2.1.2.2.1.10.'
CISCO_IN_COUNT_OID = '.1.3.6.1.2.1.2.2.1.11.'
CISCO_OUT_TRAFFIC_OID = '.1.3.6.1.2.1.2.2.1.16.'
CISCO_OUT_COUNT_OID = '.1.3.6.1.2.1.2.2.1.17.'
CISCO_NAME_OID = '.1.3.6.1.2.1.1.5'
CISCO_PORT_NAME_OID = '.1.3.6.1.2.1.2.2.1.2.'
SLEEP_TIME = 8 #s

SERVER = '10.128.120.38:2281'

if __name__ == '__main__':
    while True:
        for unit in monitor:
            if not unit.get('name'):
                try:
                    cmd = 'snmpwalk -c '+unit['public']+' -v '+unit['version']+' '+unit['ip']+' '+CISCO_NAME_OID
                    p = subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)
                    p.wait()
                    unit['name'] = p.stdout.read().split(': ')[1].strip()
                except Exception, e:
                    print str(e)
                    unit['name'] = 'unknown'
            cmd = 'snmpget -c '+unit['public']+' -v '+unit['version']+' '+unit['ip']+' '
            if unit['type'] == 'cisco':
                in_traffic_oid = CISCO_IN_TRAFFIC_OID
                out_traffic_oid = CISCO_OUT_TRAFFIC_OID
                in_count_oid = CISCO_IN_COUNT_OID
                out_count_oid = CISCO_OUT_COUNT_OID
            else:
                print 'The unit ' + unit['name'] + '\'s type can not resolved!! [' + unit['type'] + ']'
                continue
            if not unit.get('portname'):
                try:
                    cmd = 'snmpget -c '+unit['public']+' -v '+unit['version']+' '+unit['ip']+' '
                    if unit['type'] == 'cisco':
                        for p in unit['port']:
                            cmd += CISCO_PORT_NAME_OID + str(p) + ' '
                    else:
                        raise Exception('unknown device type!')
                    p = subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)
                    p.wait()
                    unit['portname'] = p.stdout.read().split(': ')[1].strip()
                except Exception, e:
                    print str(e)
                    unit['portname'] = unit['port']
            for port in unit['port']:
                if not port:
                    continue
                cmd += in_traffic_oid + str(port) + ' '
                cmd += out_traffic_oid + str(port) + ' '
                cmd += in_count_oid + str(port) + ' '
                cmd += out_count_oid + str(port) + ' '
            p = subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)
            p.wait()
            result = [r.split(': ')[1] for r in p.stdout.read().split('\n') if r]
            if not unit.get('last'):
                unit['last'] = [int(r) for r in result]
            else:
                now = [int(r) for r in result]
                diff = []
                i = 0
                while i < len(now):
                    d = now[i]-unit['last'][i]
                    if d < 0:
                        d += 4294967296
                    diff.append(d)
                    i += 1
                unit['last'] = now
                print unit['last']
                i = 0
                while i < len(unit['port']):
                    try:
                        #in
                        print 'http://'+SERVER+'/cat/r/systemMonitor?group='+unit['group']+'&domain='+ \
                                unit['name']+'&key='+str(unit['portname'][i])+'-in&op=sum&sum='+str(diff[::4][i])
                        #urllib2.urlopen('http://'+SERVER+'/cat/r/systemMonitor?group='+unit['group']+'&domain='+
                        #            unit['name']+'&key='+unit['name']+'-'+group['key']+'/in-traffic'+'&op=sum&sum='+str(data),timeout=0)
                        #out
                        print 'http://'+SERVER+'/cat/r/systemMonitor?group='+unit['group']+'&domain='+ \
                                unit['name']+'&key='+str(unit['portname'][i])+'-in&op=sum&sum='+str(diff[1::4][i])
                        #urllib2.urlopen('http://'+SERVER+'/cat/r/systemMonitor?group='+unit['group']+'&domain='+
                        #            unit['name']+'&key='+unit['name']+'-'+group['key']+'/in-traffic'+'&op=sum&sum='+str(data),timeout=0)
                    except Exception, e:
                        print str(e)
                    i += 1

        time.sleep(SLEEP_TIME)