From 46ffae30304efe5f3978e2b5f4f262dd0ffed684 Mon Sep 17 00:00:00 2001
From: A Letessier <Alexis.letessier@gmail.com>
Date: Thu, 30 Dec 2021 15:22:47 +0100
Subject: [PATCH] parent optional assoc in tenant_groups

---
 startup_scripts/020_tenant_groups.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/startup_scripts/020_tenant_groups.py b/startup_scripts/020_tenant_groups.py
index 65cf155..5a80d7e 100644
--- a/startup_scripts/020_tenant_groups.py
+++ b/startup_scripts/020_tenant_groups.py
@@ -8,7 +8,17 @@ tenant_groups = load_yaml("/opt/netbox/initializers/tenant_groups.yml")
 if tenant_groups is None:
     sys.exit()
 
+optional_assocs = {
+    'parent': (TenantGroup, 'name'),
+}
+
 for params in tenant_groups:
+    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)
+
     tenant_group, created = TenantGroup.objects.get_or_create(**params)
 
     if created: