gistfile1.txt

#!/usr/bin/env python
# Author: Nick Anderson <[email protected]>
# Rack PDU AMP Load Aggregation
# Designed for use with APC 7840
# Scinario: You have 2 legs of power A and B, each rack has 1 PDU on each leg
#           and you need to monitor the aggregate and individual AMP Load

import subprocess
from optparse import *



def display(aggregate, results):
    '''Output load results in zenoss parsable format'''
    print "aggregate=%s" %aggregate,
    count = 0
    for device in results:
        if count == 0:
            print "A=%s" %(results[device]),
        else:
            print "B=%s" %(results[device]),
        count += 1


def collect_data(options, devices, oid):
    '''Fetch data at given OID from specified devices'''
    results = {}
    for device in devices:
        cmd = "snmpget -v1 -c public %s %s" %(device, oid)
        # Get the numeric value from snmp
        value = subprocess.Popen(cmd,
                                shell=True,
                                stdout=subprocess.PIPE).communicate()[0].split()[-1]
        results[device] = value
    return results
    
def convert_values(results):
    '''Likely specific to the OID data but for APC 7840 this converts
    the string into a float and converts it to AMP Decimal'''
    for device in results:
        results[device] = round(float(results[device])/10,2)
    return results
    
if __name__ == "__main__":

    op = OptionParser("Usage: %prog [-c communitystring] PDUA PDUB OID")
    op.add_option('-c',
            dest='community_string',
            help='snmp v1 community string (default: %default)')

    op.set_defaults(community_string='public')

    (options, args) = op.parse_args()

    if not len(args) == 3:
        op.error("PDUA PDUB and OID are required")

    # Add PDUA and PDUB to device list
    devices = [ args[0], args[1] ]
    oid = args[2]

    # Collect and convert results
    results = collect_data(options, devices, oid)
    results = convert_values(results)

    # Tabulate aggrigate value for devices
    aggregate = 0
    for device in results:
       aggregate += results[device]

    # Display output
    display(aggregate,results)