Merge pull request #61 from yanndegat/addinterfaceuntaggedvlan

interfaces: add untagged_vlan optional assoc

Author: tobiasge

src/netbox_initializers/initializers/__init__.py

@@ -33,14 +33,14 @@
     "cluster_types",
     "cluster_groups",
     "clusters",
+    "prefix_vlan_roles",
+    "vlan_groups",
+    "vlans",
     "devices",
     "interfaces",
     "route_targets",
     "vrfs",
     "aggregates",
-    "prefix_vlan_roles",
-    "vlan_groups",
-    "vlans",
     "virtual_machines",
     "virtualization_interfaces",
     "prefixes",

src/netbox_initializers/initializers/interfaces.py

@@ -1,9 +1,13 @@
 from dcim.models import Device, Interface
+from ipam.models import VLAN
 
 from . import BaseInitializer, register_initializer
 
 MATCH_PARAMS = ["device", "name"]
 REQUIRED_ASSOCS = {"device": (Device, "name")}
+OPTIONAL_ASSOCS = {
+    "untagged_vlan": (VLAN, "name"),
+}
 RELATED_ASSOCS = {
     "bridge": (Interface, "name"),
     "lag": (Interface, "name"),
@@ -29,6 +33,13 @@ def load_data(self):
 
                 params[assoc] = model.objects.get(**query)
 
+            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)
+
             matching_params, defaults = self.split_params(params, MATCH_PARAMS)
             interface, created = Interface.objects.get_or_create(
                 **matching_params, defaults=defaults

src/netbox_initializers/initializers/yaml/interfaces.yml

@@ -29,6 +29,7 @@
 #   enabled: true
 #   type: 1000base-t
 #   name: eth0
+#   untagged_vlan: vlan2
 # - device: server02
 #   enabled: true
 #   type: virtual