netbox.py: Update script to support latest Netbox

Starting with v3.3, several Netbox API fields have been modified, including ones for power outlets (change introduced in netbox-community/netbox#9615). Update script to reflect those changes Signed-off-by: bia1708 <[email protected]>
sdgtt · Oct 18, 2024 · ebe2251 · ebe2251
1 parent 83bf5ee
commit ebe2251
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/nebula/netbox.py b/nebula/netbox.py
@@ -299,12 +299,15 @@ def __init__(  # noqa: C901
         self.data["devices"].update({"power_ports": dict()})
         for pow in pow_raw:
             # get associated oulet
-            outlet_id = pow["connected_endpoint"]["id"]
+            outlet_id = pow["connected_endpoints"][0]["id"]
             outlet = self.nbi.get_power_outlets(id=outlet_id)
-            pow["connected_endpoint"]["outlet"] = outlet[0]["custom_fields"]["outlet"]
+            outlet_number = outlet[0]["custom_fields"]["outlet"]
+            pow["connected_endpoints"][0]["outlet"] = outlet_number
 
             # get ip of pdu
-            pdu_raw = self.nbi.get_devices(id=pow["connected_endpoint"]["device"]["id"])
+            pdu_raw = self.nbi.get_devices(
+                id=pow["connected_endpoints"][0]["device"]["id"]
+            )
             pow["pdus"] = list()
             for pdu in pdu_raw:
                 pow["pdus"].append(pdu)

diff --git a/nebula/resources/template_gen.yaml b/nebula/resources/template_gen.yaml
@@ -113,7 +113,7 @@ pdu-config:
     help: "Outlet number on PDU for dev board"
     optional: True
     type: int
-    netbox_field: devices.power_ports.input.connected_endpoint.outlet
+    netbox_field: devices.power_ports.input.connected_endpoints[0].outlet
   field_4:
     name: username
     default: "username"

diff --git a/nebula/usbmux.py b/nebula/usbmux.py
@@ -83,6 +83,8 @@ def find_muxed_sdcard(self):
         time.sleep(5)
         context = pyudev.Context()
         for device in context.list_devices(subsystem="block"):
+            print("WHAT")
+            print(device.get("ID_SERIAL_SHORT"))
             if device.get("ID_SERIAL_SHORT") == os.path.basename(
                 self._mux_in_use
             ).strip("id-"):