ipfabric_netbox 3.1.2__py3-none-any.whl

ipfabric_netbox/utilities/nbutils.py

@@ -0,0 +1,105 @@
+from collections import Counter
+from copy import deepcopy
+
+from dcim.models import Device
+from dcim.models import InventoryItem
+from dcim.models import Manufacturer
+
+DEFAULT_DEVICE_ROLE = "Network Device"
+device_serial_max_length = Device._meta.get_field("serial").max_length
+
+
+def order_members(members):
+    devices = {}
+
+    for member in members:
+        master_serial = member.get("sn")
+        if master_serial and member.get("memberSn"):
+            if master_serial in devices:
+                devices[master_serial].append(member)
+            else:
+                devices[master_serial] = [member]
+
+    return devices
+
+
+def order_devices(devices, members):
+    hostnames = [d["hostname"] for d in devices]
+    counter = Counter(hostnames)
+
+    new_devices = []
+
+    for device in devices:
+        if counter[device["hostname"]] > 1:
+            device["hostname"] = f"{device['hostname']} - ({device['sn']})"
+        if child_members := members.get(device.get("sn")):
+            for child_member in child_members:
+                if device.get("sn") != child_member.get("memberSn"):
+                    new_device = deepcopy(device)
+                    new_device[
+                        "hostname"
+                    ] = f"{device['hostname']}/{child_member.get('member')}"
+                    new_device["model"] = child_member.get("pn")
+                    new_device["sn"] = child_member.get("memberSn")
+                    new_device["virtual_chassis"] = child_member
+                    new_devices.append(new_device)
+                else:
+                    device["virtual_chassis"] = child_member
+            hostnames = [d["hostname"] for d in devices]
+            counter = Counter(hostnames)
+
+    devices.extend(new_devices)
+
+    return devices
+
+
+def order_vrf(vrfs):
+    vrf_dict = {}
+
+    for vrf in vrfs:
+        sn = vrf["sn"]
+        if sn in vrf_dict:
+            vrf_dict[sn].append(vrf)
+        else:
+            vrf_dict[sn] = [vrf]
+
+    return vrf_dict
+
+
+def order_pn(part_numbers):
+    pn_dict = {}
+
+    for pn in part_numbers:
+        device_sn = pn["deviceSn"]
+        if device_sn in pn_dict:
+            pn_dict[device_sn].append(pn)
+        else:
+            pn_dict[device_sn] = [pn]
+
+    return pn_dict
+
+
+def create_inventory_items(device: Device, parts: list, manufacturer: Manufacturer):
+    for part in parts:
+        name = part.get("name", "")
+        if len(name) > InventoryItem._meta.get_field("name").max_length:
+            if part.get("dscr"):
+                name = part.get("dscr")
+            else:
+                name = part.get("sn")
+
+        defaults = {
+            "name": name,
+            "manufacturer": manufacturer,
+            "serial": part.get("sn", ""),
+            # "description": part.get('dscr', "123"),
+            "part_id": part.get("pid", ""),
+            "device": device,
+            "lft": device.pk,
+        }
+        if part.get("dscr"):
+            defaults["description"] = part.get("dscr")
+
+        inventory_object, _ = InventoryItem.objects.update_or_create(
+            serial=part.get("sn", ""), defaults=defaults
+        )

ipfabric_netbox/utilities/transform_map.py

@@ -0,0 +1,35 @@
+from django.contrib.contenttypes.models import ContentType
+
+from ipfabric_netbox.models import IPFabricRelationshipField
+from ipfabric_netbox.models import IPFabricTransformField
+from ipfabric_netbox.models import IPFabricTransformMap
+
+
+def BuildField(data):
+    if "target_model" in data:
+        ct = ContentType.objects.get(
+            app_label=data["target_model"]["app_label"],
+            model=data["target_model"]["model"],
+        )
+        data["target_model"] = ct
+    elif "source_model" in data:
+        ct = ContentType.objects.get(
+            app_label=data["source_model"]["app_label"],
+            model=data["source_model"]["model"],
+        )
+        data["source_model"] = ct
+    return data
+
+
+def BuildTransformMaps(data):
+    for tm in data:
+        field_data = BuildField(tm["data"])
+        tm_obj = IPFabricTransformMap.objects.create(**field_data)
+        for fm in tm["field_maps"]:
+            field_data = BuildField(fm)
+            IPFabricTransformField.objects.create(transform_map=tm_obj, **field_data)
+        for rm in tm["relationship_maps"]:
+            relationship_data = BuildField(rm)
+            IPFabricRelationshipField.objects.create(
+                transform_map=tm_obj, **relationship_data
+            )