From 8d8b9a157e219ce3ed5944f8fd19ca668b4c6cb2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christian=20M=C3=A4der?= <mail@cimnine.ch>
Date: Fri, 17 Jan 2020 15:59:21 +0100
Subject: [PATCH] Update VLAN Initializer for Netbox 2.7

The VLAN database model has changed in Netbox 2.7.
Therefore the initializer code, that was used before, broke.
As a user, you will need to update your vlans.yml file as follows:

- Make sure the status is spelled lowercase.

See the diff of this commit for further information how this is meant.
---
 initializers/vlans.yml       | 12 ++++++++++--
 startup_scripts/210_vlans.py |  7 -------
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/initializers/vlans.yml b/initializers/vlans.yml
index 26532c7..a8cd521 100644
--- a/initializers/vlans.yml
+++ b/initializers/vlans.yml
@@ -1,11 +1,19 @@
+## Possible Choices:
+##   status:
+##   - active
+##   - reserved
+##   - deprecated
+##
+## Examples:
+
 # - name: vlan1
 #   site: AMS 1
-#   status: Active
+#   status: active
 #   vid: 5
 #   role: Main Management
 #   description: VLAN 5 for MGMT
 # - group: VLAN group 2
 #   name: vlan2
 #   site: AMS 1
-#   status: Active
+#   status: active
 #   vid: 1300
diff --git a/startup_scripts/210_vlans.py b/startup_scripts/210_vlans.py
index e3a0ef1..ab6bd2b 100644
--- a/startup_scripts/210_vlans.py
+++ b/startup_scripts/210_vlans.py
@@ -1,6 +1,5 @@
 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
@@ -35,12 +34,6 @@ 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]
-            break
-
       vlan, created = VLAN.objects.get_or_create(**params)
 
       if created: