add parent/child support for loading device roles (#127)

* NetBox has supported a parent/child structure for device roles since 4.3. This adds parent support to loading of device roles.

* update device_roles.yml with example for parent/child support (#127)

Author: mmguero

src/netbox_initializers/initializers/device_roles.py

@@ -1,5 +1,6 @@
 from dcim.models import DeviceRole
 from netbox.choices import ColorChoices
+from django.core.exceptions import ObjectDoesNotExist
 
 from netbox_initializers.initializers.base import BaseInitializer, register_initializer
 
@@ -11,9 +12,26 @@ def load_data(self):
         device_roles = self.load_yaml()
         if device_roles is None:
             return
+
         for params in device_roles:
             tags = params.pop("tags", None)
 
+            # Resolve parent role (accept slug or name; try slug first, then name)
+            if "parent" in params and params["parent"] is not None:
+                parent_value = params.pop("parent")
+
+                parent_obj = None
+                try:
+                    parent_obj = DeviceRole.objects.get(slug=parent_value)
+                except ObjectDoesNotExist:
+                    try:
+                        parent_obj = DeviceRole.objects.get(name=parent_value)
+                    except ObjectDoesNotExist:
+                        raise ValueError(f"DeviceRole parent '{parent_value}' not found by slug or name")
+
+                if parent_obj:
+                    params["parent"] = parent_obj
+
             if "color" in params:
                 color = params.pop("color")
 
@@ -22,9 +40,7 @@ def load_data(self):
                         params["color"] = color_tpl[0]
 
             matching_params, defaults = self.split_params(params)
-            device_role, created = DeviceRole.objects.get_or_create(
-                **matching_params, defaults=defaults
-            )
+            device_role, created = DeviceRole.objects.get_or_create(**matching_params, defaults=defaults)
 
             if created:
                 print("🎨 Created device role", device_role.name)

src/netbox_initializers/initializers/yaml/device_roles.yml

@@ -1,8 +1,18 @@
+# - name: Network Structure
+#   slug: network-structure
+#   description: Network infrastructure device providing connectivity between systems
+# - name: Gateway
+#   slug: gateway
+#   parent: network-structure
+#   description: Network edge device category (routing, firewalling, protocol coupling)
+#   color: White
 # - name: switch
 #   slug: switch
+#   parent: network-structure
 #   color: Grey
 # - name: router
 #   slug: router
+#   parent: gateway
 #   color: Cyan
 # - name: load-balancer
 #   slug: load-balancer