diff --git a/startup_scripts/020_tenant_groups.py b/startup_scripts/020_tenant_groups.py index 8673c72..5a80d7e 100644 --- a/startup_scripts/020_tenant_groups.py +++ b/startup_scripts/020_tenant_groups.py @@ -1,25 +1,25 @@ import sys -from tenancy.models import TenantGroup from startup_script_utils import load_yaml +from tenancy.models import TenantGroup -tenant_groups = load_yaml('/opt/netbox/initializers/tenant_groups.yml') +tenant_groups = load_yaml("/opt/netbox/initializers/tenant_groups.yml") if tenant_groups is None: - sys.exit() + sys.exit() optional_assocs = { - 'parent': (TenantGroup, 'name'), + '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) + 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) + tenant_group, created = TenantGroup.objects.get_or_create(**params) - if created: - print("🔳 Created Tenant Group", tenant_group.name) + if created: + print("🔳 Created Tenant Group", tenant_group.name)