Netbox Monitoring UPS by SNMP for Power utilization feature. Script

[JHarrison712]

I made an SNMP script that sites on the netbox server and pushes every so often. Since the way our datacenter and other places are design we have Panel A > Power Feed A > UPS > PDU > Servers.

---

from pysnmp.hlapi import *
from pynetbox import api

# NetBox API credentials and UPS device ID
NETBOX_URL = 'http://netbox.example.com/api/'
NETBOX_TOKEN = '[REDACTED]'
UPS_DEVICE_ID = 6 # Replace with the actual UPS device ID in NetBox

# SNMP parameters for the UPS device (SNMPv1)
SNMP_COMMUNITY = 'public'  # SNMP community string
SNMP_DEVICE = ('[REDACTED]', 161)  # Replace with the UPS IP address and SNMP port
# SNMP OID for UPS load
OID_UPS_LOAD = '1.3.6.1.2.1.33.1.4.4.1.5.1'

# Function to fetch UPS load using SNMP
def get_ups_load():
    errorIndication, errorStatus, errorIndex, varBinds = next(
        getCmd(SnmpEngine(),
               CommunityData(SNMP_COMMUNITY),
               UdpTransportTarget(SNMP_DEVICE, timeout=1, retries=5),  # Increase timeout and retries if needed
               ContextData(),
               ObjectType(ObjectIdentity(OID_UPS_LOAD)))
    )

    if errorIndication:
        print("SNMP Error:", errorIndication)
        return None
    elif errorStatus:
        print("SNMP Error:", errorStatus.prettyPrint())
        return None
    else:
        for varBind in varBinds:
            return int(varBind[1])

# Function to update UPS load in NetBox
def update_ups_load_in_netbox(upl):
    try:
        nb = api(url=NETBOX_URL, token=NETBOX_TOKEN)
        device = nb.dcim.devices.get(UPS_DEVICE_ID)
        device.custom_fields['ups_load'] = upl
        device.save()
        print("UPS load updated in NetBox:", upl)
    except Exception as e:
        print("Error updating UPS load in NetBox:", e)

# Main function
def main():
    ups_load = get_ups_load()
    if ups_load is not None:
        update_ups_load_in_netbox(ups_load)

if __name__ == "__main__":
    main()

---

I worked out all the kinks and the only issues I have now is "Error: Device not found in NetBox."

Even tho when I go to my Netbox that UPS it shows DCIM.Devices.6

[markkuleinio]

Please separate your whole codeblock from other text with three backticks (```) to format it nicely. (Use the preview tab to verify before posting)

I'm confused because I don't think that code could ever print error "Error: Device not found in NetBox." Did I misread the code (due to uncomfortable formatting) or did you mean something else?

[JHarrison712]

I get this now.

[markkuleinio]

Now that's a real error, meaning that nb.dcim.devices.get() returned None.

Does the user (whose token it is) have permission to view that device?

[JHarrison712]

After I modified these >

I now get

[JHarrison712]

Yes I'm just using the admin account I have for right now.

[markkuleinio]

I still don't think that error is coming from either NetBox or pynetbox, so your environment is using something that you haven't revealed yet.

What are your NetBox and pynetbox versions?

How about removing try-except as I suggested?