Change init order and location initializer (#34)

* __init__.py: Init platforms immediately after manufacturer. locations.py: enable parent location initialization via OPTIONAL_ASSOCS.

* Add location parent example YAML definitions

* Fixed Tests

Co-authored-by: Tobias Genannt <[email protected]>

Author: palmersample

src/netbox_initializers/initializers/__init__.py

@@ -24,14 +24,14 @@
     "power_panels",
     "power_feeds",
     "manufacturers",
+    "platforms",
     "device_roles",
     "device_types",
     "cluster_types",
     "cluster_groups",
     "clusters",
     "devices",
     "interfaces",
-    "platforms",
     "route_targets",
     "vrfs",
     "rirs",

src/netbox_initializers/initializers/locations.py

@@ -2,8 +2,7 @@
 
 from . import BaseInitializer, register_initializer
 
-MATCH_PARAMS = ["name", "slug", "site"]
-REQUIRED_ASSOCS = {"site": (Site, "name")}
+OPTIONAL_ASSOCS = {"site": (Site, "name"), "parent": (Location, "name")}
 
 
 class LocationInitializer(BaseInitializer):
@@ -15,12 +14,13 @@ def load_data(self):
             return
         for params in locations:
 
-            for assoc, details in REQUIRED_ASSOCS.items():
-                model, field = details
-                query = {field: params.pop(assoc)}
-                params[assoc] = model.objects.get(**query)
+            for assoc, details in OPTIONAL_ASSOCS.items():
+                if assoc in params:
+                    model, field = details
+                    query = {field: params.pop(assoc)}
+                    params[assoc] = model.objects.get(**query)
 
-            matching_params, defaults = self.split_params(params, MATCH_PARAMS)
+            matching_params, defaults = self.split_params(params)
             location, created = Location.objects.get_or_create(**matching_params, defaults=defaults)
 
             if created:

src/netbox_initializers/initializers/yaml/locations.yml

@@ -1,3 +1,10 @@
 # - name: cage 101
 #   slug: cage-101
 #   site: SING 1
+# - name: Parent Location
+#   slug: parent-location
+#   site: SING 1
+# - name: Child location
+#   slug: child-location
+#   site: SING 1
+#   parent: Parent Location