#!/usr/bin/python3

import json
import SNMPUtil
import argparse

### Monitoring iDRAC Servers - Memory Performance

### It uses snmpwalk command to get the hadrware data from the iDRAC Servers.
### SNMPUtil.py is used to get the snmp raw data and parsed to get the output json
### Download and install the latest version of Site24x7 Linux Agent. The agent will execute the plugin and push the data to the Site24x7 server
### 
### Author: Anita, Zoho Corp
### Language : Python
### Tested in Ubuntu
### Tested for snmp version 2c

### iDRAC Server Configuration Details

OIDREPLACE = "1.3.6.1.4.1"
SMISTR="SNMPv2-SMI::enterprises"
ISOSTR="iso.3.6.1.4.1" 

### OIDS for Getting Memory Details
OIDS = {'memory' : ['1.3.6.1.4.1.674.10892.5.4.1100.50.1']}
### OID Attributes
hardware = {'memory' : ['1.3.6.1.4.1.674.10892.5.4.1100.50.1.4','1.3.6.1.4.1.674.10892.5.4.1100.50.1.5','    1.3.6.1.4.1.674.10892.5.4.1100.50.1.7','1.3.6.1.4.1.674.10892.5.4.1100.50.1.14','1.3.6.1.4.1.674.10892.5.4.1100.50.1.15']}
### Output Keys and their units
names = {'memory' : ['state','status','type',{'size': 'KB'},{'speed':'nanosec'}]}


class HardwareParser:
    def __init__(self, hostname, snmp_version, snmp_community_str, mib_location):
        self.hostname = hostname
        self.snmp_version = snmp_version
        self.snmp_community_str = snmp_community_str
        self.mib_location = mib_location
    
        self.hardware = ''
        self.oids = ''
        self.pattern = ''
        
    def getData(self):
        output_data = {}
        output_data['data'] = {}
        output_data['units'] = {}
        for _ in OIDS:
            self.hardware = _
            self.oids = OIDS[self.hardware]
            
            for _ in self.oids:
                ### SNMPUtil module is used to get the snmp output for the input OIDS
                snmpdata = SNMPUtil.SNMPPARSER('snmpwalk',self.hostname, self.snmp_version, self.snmp_community_str,_, self.mib_location, hardware[self.hardware])
                ### get Raw SNMP Output as a dict
                self.snmp_data = snmpdata.getRawData()
                ### Method to parse the SNMP command output data
                output_data = self.parseSNMPData(output_data)
                
        return output_data
    
    
    ### Method to parse the SNMP command output data
    def parseSNMPData(self,output_data):
        jsondata = output_data['data'] 
        unitdata = output_data['units'] 
        
        for _ in self.snmp_data:
            if ( not _.startswith(OIDREPLACE) and _.startswith(SMISTR) ):
                _ = _.replace(SMISTR, OIDREPLACE)
            elif ( not _.startswith(OIDREPLACE) and _.startswith(ISOSTR) ):
                _ = _.replace(ISOSTR, OIDREPLACE)
            #print(_)
            
            for index, __ in enumerate(hardware[self.hardware]) :
                if __ in _:        
                    
                    _ = _.replace('\n','').replace('\r','').replace('"','')
                    name = _.split(' ')[0]
                    elementname = name[len(name)-1]

                    l = _.split(' ')
                    l.pop(0)
                    value = ' '.join(l)
                    
                    
                    if ':' in value:
                        val = value.split(':')[1:] 
                        value = val[len(val)-1]
                   
                    elem = names[self.hardware][index]
                    attribute = ''  # Attribute Name
                    unit = ''       # Attribute Value
                    
                    if type(elem) is str:   # Attributes with no units specified
                        attribute = elem
                    elif type(elem) is dict:    # Attributes with units
                        attribute = list(elem.keys())[0]
                        unit = elem[list(elem.keys())[0]]
                        
                    key = (attribute +'_'+elementname).replace(' ','')
                    jsondata[key] = value
                    
                    if unit!='':
                        unitdata[key] = unit 
                 
        output_data['data'] = jsondata
        output_data['units'] = unitdata
        
        return output_data



if __name__ == '__main__':
    
    result = {}
    parser = argparse.ArgumentParser()
    parser.add_argument('--hostname', help='hostname', nargs='?', default='localhost')
    parser.add_argument('--snmp_version', help='snmp version', type=str, nargs='?', default="2c")
    parser.add_argument('--snmp_community_str', help='snmp community version',  nargs='?', default='public')
    parser.add_argument('--idrac_mib_file_locn', help='idrac mib file location',  nargs='?', default='')
    parser.add_argument('--plugin_version', help='plugin template version',  nargs='?', default='1')
    parser.add_argument('--heartbeat_required', help='Enable heartbeat for monitoring',  nargs='?', default="true")
    args = parser.parse_args()
    
    try:
        parser = HardwareParser(args.hostname, args.snmp_version, args.snmp_community_str, args.idrac_mib_file_locn)
        output = parser.getData()
        result = output['data']
        result['units'] = output['units']
    except ValueError as e:
        result['msg'] = str(e)
    
    result['plugin_version'] = args.plugin_version
    result['heartbeat_required'] = args.heartbeat_required
    print(json.dumps(result, indent=2, sort_keys=True))