Fixed further requirements.

startup_scripts/210_vlans.py

@@ -1,5 +1,6 @@
 from dcim.models import Site
 from ipam.models import VLAN, VLANGroup, Role
+from ipam.constants import VLAN_STATUS_CHOICES
 from tenancy.models import Tenant, TenantGroup
 from extras.models import CustomField, CustomFieldValue
 from ruamel.yaml import YAML
@@ -34,6 +35,11 @@ with file.open('r') as stream:
 
           params[assoc] = model.objects.get(**query)
 
+      if 'status' in params:
+        for vlan_status in VLAN_STATUS_CHOICES:
+          if params['status'] in vlan_status:
+            params['status'] = vlan_status[0]
+
       vlan, created = VLAN.objects.get_or_create(**params)
 
       if created:

startup_scripts/220_prefixes.py

@@ -1,5 +1,6 @@
 from dcim.models import Site
 from ipam.models import Prefix, VLAN, Role, VRF
+from ipam.constants import PREFIX_STATUS_CHOICES
 from tenancy.models import Tenant, TenantGroup
 from extras.models import CustomField, CustomFieldValue
 from ruamel.yaml import YAML
@@ -37,6 +38,11 @@ with file.open('r') as stream:
 
           params[assoc] = model.objects.get(**query)
 
+      if 'status' in params:
+        for prefix_status in PREFIX_STATUS_CHOICES:
+          if params['status'] in prefix_status:
+            params['status'] = prefix_status[0]
+
       prefix, created = Prefix.objects.get_or_create(**params)
 
       if created:

startup_scripts/230_virtual_machines.py

@@ -1,5 +1,6 @@
 from dcim.models import Site, Platform, DeviceRole
 from virtualization.models import Cluster, VirtualMachine
+from virtualization.constants import VM_STATUS_CHOICES
 from tenancy.models import Tenant
 from extras.models import CustomField, CustomFieldValue
 from ruamel.yaml import YAML
@@ -42,6 +43,11 @@ with file.open('r') as stream:
 
           params[assoc] = model.objects.get(**query)
 
+      if 'status' in params:
+        for vm_status in VM_STATUS_CHOICES:
+          if params['status'] in vm_status:
+            params['status'] = vm_status[0]
+
       virtual_machine, created = VirtualMachine.objects.get_or_create(**params)
 
       if created:

startup_scripts/240_virtualization_interfaces.py

@@ -14,7 +14,7 @@ with file.open('r') as stream:
   yaml = YAML(typ='safe')
   interfaces = yaml.load(stream)
 
-  optional_assocs = {
+  required_assocs = {
     'virtual_machine': (VirtualMachine, 'name')
   }
 
@@ -22,12 +22,11 @@ with file.open('r') as stream:
     for params in interfaces:
       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 required_assocs.items():
+        model, field = details
+        query = { field: params.pop(assoc) }
 
-          params[assoc] = model.objects.get(**query)
+        params[assoc] = model.objects.get(**query)
 
       interface, created = Interface.objects.get_or_create(**params)
 

startup_scripts/250_dcim_interfaces.py

@@ -1,4 +1,5 @@
 from dcim.models import Interface, Device
+from dcim.constants import IFACE_TYPE_CHOICES
 from extras.models import CustomField, CustomFieldValue
 from ruamel.yaml import YAML
 
@@ -13,7 +14,7 @@ with file.open('r') as stream:
   yaml = YAML(typ='safe')
   interfaces = yaml.load(stream)
 
-  optional_assocs = {
+  required_assocs = {
     'device': (Device, 'name')
   }
 
@@ -21,12 +22,21 @@ with file.open('r') as stream:
     for params in interfaces:
       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 required_assocs.items():
+        model, field = details
+        query = { field: params.pop(assoc) }
 
-          params[assoc] = model.objects.get(**query)
+        params[assoc] = model.objects.get(**query)
+
+      if 'form_factor' in params:
+        for outer_list in IFACE_TYPE_CHOICES:
+          for ffactor_choices in outer_list[1]:
+            if params['form_factor'] in ffactor_choices:
+              params['form_factor'] = ffactor_choices[0]
+              break
+          else:
+            continue
+          break
 
       interface, created = Interface.objects.get_or_create(**params)
 
@@ -43,4 +53,3 @@ with file.open('r') as stream:
             interface.custom_field_values.add(custom_field_value)
 
         print("🧷 Created interface", interface.name, interface.device.name)
-

startup_scripts/260_ip_addresses.py

@@ -1,4 +1,5 @@
 from ipam.models import IPAddress, VRF
+from ipam.constants import IPADDRESS_STATUS_CHOICES
 from dcim.models import Device, Interface
 from virtualization.models import VirtualMachine
 from tenancy.models import Tenant
@@ -30,6 +31,10 @@ with file.open('r') as stream:
       custom_fields = params.pop('custom_fields', None)
       params['address'] = IPNetwork(params['address'])
 
+      if vm and device:
+        print("IP Address can only specify one of the following: virtual_machine or device.")
+        sys.exit()
+
       for assoc, details in optional_assocs.items():
         if assoc in params:
           model, field = details
@@ -44,6 +49,11 @@ with file.open('r') as stream:
               query = { field: params.pop(assoc) }
           params[assoc] = model.objects.get(**query)
 
+      if 'status' in params:
+        for ip_status in IPADDRESS_STATUS_CHOICES:
+          if params['status'] in ip_status:
+            params['status'] = ip_status[0]
+
       ip_address, created = IPAddress.objects.get_or_create(**params)
 
       if created: