Complete restructure to modules.

Using modules and imports makes it much more modular
and managable. There are no 600 row files now

cont/idrac/idrac9.py

@@ -0,0 +1,168 @@
+import os, asyncio
+from models.idracModel import snmpPyIDRACData
+from snmp.walker import walk_column_v3
+from typing import Annotated, Final
+
+# SNMP
+from pysnmp.hlapi.v3arch.asyncio import *
+
+# SNMP ENV-------------------------------------------
+ROUND_PREC: Final[int] = int(os.getenv("ROUND_PREC", 2))
+
+SNMPUSER: Final[str] = os.getenv("SNMPUSER", None)
+SNMPPRIVKEY: Final[str] = os.getenv("SNMPPRIVKEY", None)
+SNMPAUTHKEY: Final[str] = os.getenv("SNMPAUTHKEY", None)
+# Right now I'll only use SHA
+# SNMPAUTHPROTO: Final[str] = os.getenv("SNMPAUTHPROTO", "SHA")
+# SNMPPRIVPROTO: Final[str] = os.getenv("SNMPPRIVPROTO", "SHA")
+SNMPORT: Final[int] = int(os.getenv("SNMPORT", 161))
+
+
+# Check if SNMP ENV are empty
+if not SNMPUSER or not SNMPPRIVKEY or not SNMPAUTHKEY:
+    raise Exception("No SNMP user or/and PrivAuth passed")
+
+
+# SNMP
+USMUSRDATA = UsmUserData(
+            userName=SNMPUSER,
+            authKey=SNMPAUTHKEY,
+            privKey=SNMPPRIVKEY,
+            authProtocol=usmHMACSHAAuthProtocol,
+            privProtocol=usmAesCfb128Protocol,
+        )
+
+
+
+
+
+
+# IDRAC get data for snmpPyIDRACData class
+async def idrac9PoolRemote_v3(remoteIP: str, queueToInsrt: asyncio.Queue):
+    print("starting work on ", remoteIP)
+    snmpEngine = SnmpEngine()
+
+    iterator = get_cmd(
+        snmpEngine,
+        USMUSRDATA,
+        await UdpTransportTarget.create((remoteIP, SNMPORT)),
+        ContextData(),
+        # Hostname (sysName.0)
+        ObjectType(ObjectIdentity(".1.3.6.1.2.1.1.5.0")),                            #0
+        # Uptime in seconds
+        ObjectType(ObjectIdentity(".1.3.6.1.4.1.674.10892.5.2.5.0")),    #1
+
+    )
+    errorIndication, errorStatus, errorIndex, mainVarBinds = await iterator
+
+    if errorIndication:
+        print(errorIndication)
+    elif errorStatus:
+        print(
+            "{} at {}".format(
+                errorStatus.prettyPrint(),
+                errorIndex and mainVarBinds[int(errorIndex) - 1][0] or "?",
+            )
+        )
+    else:
+        for oid, val in mainVarBinds:
+            print(f"{oid.prettyPrint()} = {val.prettyPrint()}")
+
+
+
+    # Get PSU1 current, [PSU2 current], total board power draw in watts
+    currentsCurrentResults = await walk_column_v3(snmpEngine, remoteIP, ".1.3.6.1.4.1.674.10892.5.4.600.30.1.6")
+    # PSU1 OID name, [PSU2 OID name], system board power draw 
+    currentsNamesResults = await walk_column_v3(snmpEngine, remoteIP, ".1.3.6.1.4.1.674.10892.5.4.600.30.1.8")
+
+    # CPU1 OID name .1, [CPU2 OID name], Inlet temp OID name, [Exhaust temp OID name]
+    sensorNamesResults = await walk_column_v3(snmpEngine, remoteIP, ".1.3.6.1.4.1.674.10892.5.4.700.20.1.8")
+    # CPU1 temp, [CPU2 temp], Inlet temp, [Exhaust temp]
+    sensorValuesResults = await walk_column_v3(snmpEngine, remoteIP, ".1.3.6.1.4.1.674.10892.5.4.700.20.1.6")
+
+    # .1.3.6.1.4.1.674.10892.5.4.600.20.1.6.1
+    voltResult = await walk_column_v3(snmpEngine, remoteIP, ".1.3.6.1.4.1.674.10892.5.4.600.20.1.6.1")
+
+    snmpEngine.close_dispatcher()
+
+    voltList = []
+    for thing in voltResult:
+        voltList.append(voltResult[thing])
+
+    hasExhaust = any("exhaust" in str(thing).lower() for thing in sensorNamesResults.values())
+    hasCPU2 = any("cpu2" in str(thing).lower() for thing in sensorNamesResults.values())
+    hasPSU2 = any("ps2" in str(thing).lower() for thing in currentsNamesResults.values())
+
+    # PSU2 values and board power
+    psu2PowerDraw = None
+    psu2Amperage = None
+    psu2Voltage = None
+    psu1PowerDraw = None
+    psu1Amperage = None
+    psu1Voltage = None
+
+    if hasPSU2:
+        psu2PowerDraw = round((int(currentsCurrentResults[2]) / 10) * (int(voltList[1]) / 1000), ROUND_PREC)
+        psu2Amperage = int(currentsCurrentResults[2]) / 10
+        psu2Voltage = round((int(voltList[1]) / 1000), ROUND_PREC)
+        boardPowerDraw = int(currentsCurrentResults[3])
+    else:
+        boardPowerDraw = int(currentsCurrentResults[2])
+
+    # Exhaust CPU1 and CPU2 temp
+
+    exhaustTemp = None
+    inletTemp = None
+    cpu2Temp = None
+    if hasExhaust and hasCPU2:
+        exhaustTemp = int(sensorValuesResults[4]) / 10
+        inletTemp = int(sensorValuesResults[3]) / 10
+        cpu2Temp = int(sensorValuesResults[2]) / 10
+    elif hasExhaust and not hasCPU2:
+        exhaustTemp = int(sensorValuesResults[3]) / 10
+        inletTemp = int(sensorValuesResults[2]) / 10
+    elif hasCPU2 and not hasExhaust:
+        inletTemp = int(sensorValuesResults[3]) / 10
+        cpu2Temp = int(sensorValuesResults[2]) / 10
+    else: 
+        inletTemp = int(sensorValuesResults[2]) / 10
+
+    returnObj = snmpPyIDRACData(
+        hostname=mainVarBinds[0][-1], 
+        # Hostname
+
+        powerDrawPSU1=round((int(currentsCurrentResults[1]) / 10) * (int(voltList[0]) / 1000), ROUND_PREC),
+        powerDrawPSU2=psu2PowerDraw,
+        # PSU1 and PSU2 power draw in Watts
+
+        # board power draw
+        powerDrawBoard=boardPowerDraw,
+
+        voltagePSU1=round((int(voltList[0]) / 1000), ROUND_PREC),
+        voltagePSU2=psu2Voltage,
+        # PSU1 and PSU2 voltages
+
+        inletTemp=inletTemp,
+        exhaustTemp=exhaustTemp,
+        # Inlet and Exhaust temp
+
+        cpu1Temp=int(sensorValuesResults[1]) / 10,
+        cpu2Temp=cpu2Temp,
+        # CPU1 and CPU2 temp
+
+        uptimeS=int(mainVarBinds[1][-1]),
+        # seconds
+        uptimeH=round(((int(mainVarBinds[1][-1]) / 60) / 60), ROUND_PREC),
+        # seconds->minutes->hours
+        uptimeD=round((((int(mainVarBinds[1][-1]) / 60) / 60) / 24), ROUND_PREC)
+        # seconds->minutes->hours->days
+    )
+
+    returnDict = {
+        "source": "IDRAC",
+        "value": returnObj,
+        "type": "snmpPyIDRACData"
+    }
+    # return returnObj
+    await queueToInsrt.put(returnDict)
+