Merge branch 'fix/disable-vlan-sync-option-#245' into development

Author: bb-Ricardo

module/netbox/object_classes.py

@@ -1067,7 +1067,6 @@ def unset_attribute(self, attribute_name=None):
         # mark attribute to unset, this way it will be deleted in NetBox before any other updates are performed
         log.info(f"Setting attribute '{attribute_name}' for '{self.get_display_name()}' to None")
         self.unset_items.append(attribute_name)
-        self.data["assigned_object_id"] = None
 
     def get_nb_reference(self):
         """

module/sources/common/source_base.py

@@ -295,11 +295,6 @@ def add_update_interface(self, interface_object, device_object, interface_data,
         if len(tagged_vlans) > 0:
             del interface_data["tagged_vlans"]
 
-        # delete information about any vlans if sync is disable.
-        if disable_vlan_sync is True:
-            untagged_vlan = None
-            tagged_vlans = list()
-
         # get device tenant
         device_tenant = grab(device_object, "data.tenant")
 
@@ -312,11 +307,6 @@ def add_update_interface(self, interface_object, device_object, interface_data,
         else:
             interface_object.update(data=interface_data, source=self)
 
-        # delete information about any vlans if sync is disable.
-        if disable_vlan_sync is True:
-            interface_object.unset_attribute("untagged_vlan")
-            interface_object.unset_attribute("tagged_vlans")
-
         ip_address_objects = list()
         matching_ip_prefixes = list()
         # add all interface IPs
@@ -555,6 +545,11 @@ def add_update_interface(self, interface_object, device_object, interface_data,
         if untagged_vlan is not None or (untagged_vlan is None and len(tagged_vlans) == 0):
             if matching_untagged_vlan is None and untagged_vlan is not None:
                 matching_untagged_vlan = self.get_vlan_object_if_exists(untagged_vlan, site_name)
+
+                # don't sync newly discovered VLANs to NetBox
+                if disable_vlan_sync is True and not isinstance(matching_untagged_vlan, NetBoxObject):
+                    matching_untagged_vlan = None
+
             elif matching_untagged_vlan is not None:
                 log.debug2(f"Found matching prefix VLAN {matching_untagged_vlan.get_display_name()} for "
                            f"untagged interface VLAN.")
@@ -575,7 +570,12 @@ def add_update_interface(self, interface_object, device_object, interface_data,
             else:
                 matching_tagged_vlan = self.get_vlan_object_if_exists(tagged_vlan, site_name)
 
-            compiled_tagged_vlans.append(matching_tagged_vlan)
+                # don't sync newly discovered VLANs to NetBox
+                if disable_vlan_sync is True and not isinstance(matching_tagged_vlan, NetBoxObject):
+                    matching_tagged_vlan = None
+
+            if matching_tagged_vlan is not None:
+                compiled_tagged_vlans.append(matching_tagged_vlan)
 
         if len(compiled_tagged_vlans) > 0:
             vlan_interface_data["tagged_vlans"] = compiled_tagged_vlans