Change logic to prevent sys.exit from stopping script processing

startup_scripts/130_devices.py

@@ -8,52 +8,51 @@ import sys
 
 devices = load_yaml('/opt/netbox/initializers/devices.yml')
 
-if devices is None:
-  sys.exit()
+if not devices is None:
 
-required_assocs = {
-  'device_role': (DeviceRole, 'name'),
-  'device_type': (DeviceType, 'model'),
-  'site': (Site, 'name')
-}
+  required_assocs = {
+    'device_role': (DeviceRole, 'name'),
+    'device_type': (DeviceType, 'model'),
+    'site': (Site, 'name')
+  }
 
-optional_assocs = {
-  'tenant': (Tenant, 'name'),
-  'platform': (Platform, 'name'),
-  'rack': (Rack, 'name'),
-  'cluster': (Cluster, 'name'),
-  'primary_ip4': (IPAddress, 'address'),
-  'primary_ip6': (IPAddress, 'address')
-}
+  optional_assocs = {
+    'tenant': (Tenant, 'name'),
+    'platform': (Platform, 'name'),
+    'rack': (Rack, 'name'),
+    'cluster': (Cluster, 'name'),
+    'primary_ip4': (IPAddress, 'address'),
+    'primary_ip6': (IPAddress, 'address')
+  }
 
-for params in devices:
-  custom_fields = params.pop('custom_fields', None)
+  for params in devices:
+    custom_fields = params.pop('custom_fields', None)
 
-  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:
+    for assoc, details in required_assocs.items():
       model, field = details
       query = { field: params.pop(assoc) }
 
       params[assoc] = model.objects.get(**query)
 
-  device, created = Device.objects.get_or_create(**params)
+    for assoc, details in optional_assocs.items():
+      if assoc in params:
+        model, field = details
+        query = { field: params.pop(assoc) }
 
-  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=device,
-          value=cf_value
-        )
+        params[assoc] = model.objects.get(**query)
 
-        device.custom_field_values.add(custom_field_value)
+    device, created = Device.objects.get_or_create(**params)
 
-    print("🖥️  Created device", device.name)
+    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=device,
+            value=cf_value
+          )
+
+          device.custom_field_values.add(custom_field_value)
+
+      print("🖥️  Created device", device.name)