optional assoc to cluster & circuit startup_script

Author: ryanmerolle

initializers/circuits.yml

@@ -1,6 +1,7 @@
 # - cid: Circuit_ID-1
 #   provider: Provider1
 #   type: Internet
+#   tenant: tenant1
 # - cid: Circuit_ID-2
 #   provider: Provider2
 #   type: MPLS

initializers/clusters.yml

@@ -1,6 +1,7 @@
 # - name: cluster1
 #   type: Hyper-V
 #   group: Group 1
+#   tenant: tenant1
 # - name: cluster2
 #   type: Hyper-V
 #   site: SING 1

startup_scripts/170_clusters.py

@@ -3,6 +3,7 @@
 from dcim.models import Site
 from startup_script_utils import *
 from virtualization.models import Cluster, ClusterType, ClusterGroup
+from tenancy.models import Tenant
 
 clusters = load_yaml('/opt/netbox/initializers/clusters.yml')
 
@@ -15,7 +16,8 @@
 
 optional_assocs = {
   'site': (Site, 'name'),
-  'group': (ClusterGroup, 'name')
+  'group': (ClusterGroup, 'name'),
+  'tenant': (Tenant, 'name')
 }
 
 for params in clusters:

startup_scripts/300_circuits.py

@@ -1,4 +1,5 @@
 from circuits.models import Circuit, Provider, CircuitType
+from tenancy.models import Tenant
 from startup_script_utils import *
 import sys
 
@@ -12,6 +13,10 @@
   'type': (CircuitType, 'name')
 }
 
+optional_assocs = {
+  'tenant': (Tenant, 'name')
+}
+
 for params in circuits:
   custom_field_data = pop_custom_fields(params)
 
@@ -21,6 +26,13 @@
 
     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)
+
   circuit, created = Circuit.objects.get_or_create(**params)
 
   if created: