netbox

1 files changed, 45 insertions, 14 deletions

diff --git a/ansible/netbox/sync-unifi.py b/ansible/netbox/sync-unifi.py
index 4ee9c84..1df5e35 100644
--- a/ansible/netbox/sync-unifi.py
+++ b/ansible/netbox/sync-unifi.py
@@ -52,7 +52,7 @@ class NetboxCache():
         self.device_types[slug] = dt
         return dt
 
-    def get_or_create_ip_address(self, addr: str, vrf: Record | None):
+    def get_or_create_ip_address(self, addr: str, vrf: Record | None, data):
         vrf_id = vrf.id if vrf is not None else None
         key = (addr, vrf_id)
         ip = self.ip_addresses.get(key)
@@ -61,7 +61,9 @@ class NetboxCache():
 
         ip = self.nb.ipam.ip_addresses.get(address=addr, vrf_id=vrf_id)
         if ip is not None:
-            print(f"Found IP address {ip.id}")
+            print(f"Found IP address {ip.id} address={ip.address}, vrf={ip.vrf}")
+            ip.update(data)
+            ip = self.nb.ipam.ip_addresses.get(address=addr, vrf_id=vrf_id)
             self.ip_addresses[key] = ip
             return ip
 
@@ -69,27 +71,56 @@ class NetboxCache():
         self.ip_addresses[key] = ip
         return ip
 
-def process_switch(d: UnifiDevice, db: Db, nb: NetboxCache, site, vrf):
-    ip = nb.get_or_create_ip_address(f"{d.ip}/32", vrf)
+    def create_or_update_device(self, d):
+        device = self.nb.dcim.devices.get(name = d["name"])
+        if device is None:
+            device = self.nb.dcim.devices.create(d)
+            print(f"Created device id={device.id}, name={device.name}")
+
+            return device
+
+        print(f"Updating device id={device.id}, name={device.name}")
+        device.update(d)
+
+        return device
+
+    def create_or_update_interface(self, i):
+        iface = self.nb.dcim.interfaces.get(device_id=i["device"], name=i["name"])
+        if iface is None:
+            iface = self.nb.dcim.interfaces.create(i)
+            print(f"Created interface id={iface.id}, name={iface.name}")
 
-    d = {
+            return iface
+
+        print(f"Updating interface id={iface.id}, name={iface.name}")
+        iface.update(i)
+
+        return iface
+
+def process_switch(d: UnifiDevice, db: Db, nb: NetboxCache, site, vrf):
+    device = nb.create_or_update_device({
         "name": d.name,
         "device_type": nb.get_device_type("ubiquiti-us-8-150w").id,
         "role": nb.get_device_role("switch").id,
-#        "mac": d.mac,
         "serial": d.serial,
         "site": site.id,
+    })
+
+    i = {
+        "device": device.id,
+        "name": "switch0",
+#        "mac": d.mac, TODO: create the ipam.mac first
+        "type": "virtual",
     }
+    iface = nb.create_or_update_interface(i)
 
-    device = nb.nb.dcim.devices.get(name = d["name"])
-    if device is None:
-        device = nb.nb.dcim.devices.create(d)
-        print(f"Created device id={device.id}, name={device.name}")
-    else:
-        print(f"Updating device id={device.id}, name={device.name}")
-        device.update(d)
+    ip = nb.get_or_create_ip_address(f"{d.ip}/32", vrf, {
+        "assigned_object": iface,
+        "assigned_object_id": iface.id,
+        "assigned_object_type": "dcim.interface",
+    })
 
-#    pprint(device.serialize())
+    pprint(ip.serialize())
 
 def main():
     unifi_url=os.getenv("UNIFI_URL")