Add device seeds

initializers/devices.yml

@@ -0,0 +1,27 @@
+# - name: server01
+#   device_role: server
+#   device_type: other
+#   site: AMS 1
+#   rack: rack-01
+#   face: Front
+#   position: 1
+#   custom_fields:
+#     text_field: Description
+# - name: server02
+#   device_role: server
+#   device_type: other
+#   site: AMS 2
+#   rack: rack-02
+#   face: Front
+#   position: 2
+#   custom_fields:
+#     text_field: Description
+# - name: server03
+#   device_role: server
+#   device_type: other
+#   site: SING 1
+#   rack: rack-03
+#   face: Front
+#   position: 3
+#   custom_fields:
+#     text_field: Description

startup_scripts/47_devices.py

@@ -0,0 +1,46 @@
+from dcim.models import Site, Rack, DeviceRole, DeviceType, Device, Platform
+from dcim.constants import RACK_FACE_CHOICES
+from ipam.models import IPAddress
+from tenancy.models import Tenant
+from extras.models import CustomField, CustomFieldValue
+from ruamel.yaml import YAML
+
+with open('/opt/netbox/initializers/devices.yml', 'r') as stream:
+  yaml = YAML(typ='safe')
+  devices = yaml.load(stream)
+
+  optional_assocs = {
+    'platform': Platform,
+    'tenant': Tenant
+    'rack': Rack
+    'primary_ip4': IPAddress
+    'primary_ip6': IPAddress
+  }
+
+  if devices is not None:
+    for device_params in devices:
+      custom_fields = device_params.pop('custom_fields', None)
+
+      device_params['device_role'] = DeviceRole.objects.get(name=device_params.pop('device_role'))
+      device_params['device_type'] = DeviceType.objects.get(model=device_params.pop('device_type'))
+      device_params['site'] = Site.objects.get(name=device_params.pop('site'))
+
+      for param_name, model in optional_assoc.items():
+        if param_name in device_params:
+          device_params[param_name] = model.objects.get(name=device_params.pop(param_name))
+
+      for rack_face in RACK_FACE_CHOICES:
+        if device_params['face'] in rack_face:
+          device_params['face'] = rack_face[0]
+
+      device, created = Device.objects.get_or_create(**device_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=device, value=cf_value)
+
+            device.custom_field_values.add(custom_field_value)
+
+        print("Created device", device.name)