Change logic to prevent sys.exit from stopping script processing

startup_scripts/210_vlans.py

@@ -7,39 +7,38 @@ import sys
 
 vlans = load_yaml('/opt/netbox/initializers/vlans.yml')
 
-if vlans is None:
-  sys.exit()
+if not vlans is None:
 
-optional_assocs = {
-  'site': (Site, 'name'),
-  'tenant': (Tenant, 'name'),
-  'tenant_group': (TenantGroup, 'name'),
-  'group': (VLANGroup, 'name'),
-  'role': (Role, 'name')
-}
+  optional_assocs = {
+    'site': (Site, 'name'),
+    'tenant': (Tenant, 'name'),
+    'tenant_group': (TenantGroup, 'name'),
+    'group': (VLANGroup, 'name'),
+    'role': (Role, 'name')
+  }
 
-for params in vlans:
-  custom_fields = params.pop('custom_fields', None)
+  for params in vlans:
+    custom_fields = params.pop('custom_fields', None)
 
-  for assoc, details in optional_assocs.items():
-    if assoc in params:
-      model, field = details
-      query = { field: params.pop(assoc) }
+    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)
+        params[assoc] = model.objects.get(**query)
 
-  vlan, created = VLAN.objects.get_or_create(**params)
+    vlan, created = VLAN.objects.get_or_create(**params)
 
-  if created:
-    if custom_fields is not None:
-      for cf_name, cf_value in custom_fields.items():
-        custom_field = CustomField.objects.get(name=cf_name)
-        custom_field_value = CustomFieldValue.objects.create(
-          field=custom_field,
-          obj=vlan,
-          value=cf_value
-        )
+    if created:
+      if custom_fields is not None:
+        for cf_name, cf_value in custom_fields.items():
+          custom_field = CustomField.objects.get(name=cf_name)
+          custom_field_value = CustomFieldValue.objects.create(
+            field=custom_field,
+            obj=vlan,
+            value=cf_value
+          )
 
-        vlan.custom_field_values.add(custom_field_value)
+          vlan.custom_field_values.add(custom_field_value)
 
-    print("🏠 Created VLAN", vlan.name)
+      print("🏠 Created VLAN", vlan.name)