nornir-collection 0.0.23__tar.gz → 0.0.25__tar.gz

{nornir_collection-0.0.23 → nornir_collection-0.0.25}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: nornir-collection
-Version: 0.0.23
+Version: 0.0.25
 Summary: Nornir-Collection contains network automation functions and complete IaC workflows with         Nornir and other python libraries. It contains Nornir tasks and general functions in Nornir style.
 Author: Willi Kubny
 Author-email: willi.kubny@gmail.ch
@@ -47,6 +47,7 @@ Dynamic: author-email
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary

{nornir_collection-0.0.23 → nornir_collection-0.0.25}/nornir_collection/cisco/configuration_management/netconf/config_tasks.py

@@ -360,6 +360,71 @@ def nc_edit_cleanup_svi(task: Task, verbose: bool = False) -> Result:
     return Result(host=task.host, custom_result=custom_result, failed=task_failed)
 
 
+def nc_edit_cleanup_vlan(task: Task, verbose: bool = False) -> Result:
+    """
+    TBD
+    """
+    # The custom_result list will be filled with the results
+    custom_result = []
+
+    #### Get all current configured VLANs with RESTCONF #####################################################
+
+    task_text = "RESTCONF GET VLANs"
+
+    # Set the RESTCONF port, yang query and url to get all configured SVIs
+    yang_query = "Cisco-IOS-XE-native:native/vlan/Cisco-IOS-XE-vlan:vlan-list"
+    url = f"https://{task.host.hostname}:443/restconf/data/{yang_query}"
+
+    # RESTCONF HTTP Get for all SVIs
+    response = rc_cisco_get(url=url, auth=(task.host.username, task.host.password), verify=False)
+
+    # Set the RESTCONF result and return the Nornir result if the RESTCONF response status_code is not 200
+    custom_result = set_restconf_result(
+        task=Task,
+        task_text=task_text,
+        yang_query=yang_query,
+        response=response,
+        custom_result=custom_result,
+        verbose=verbose,
+    )
+
+    # Create a list with all current configured VLANs except the default VLAN 1
+    current_vlans = [x for x in response["json"]["Cisco-IOS-XE-vlan:vlan-list"] if x["id"] != 1]
+
+    #### Render NETCONF payload to remove VLANs with Jinja2 #################################################
+
+    task_text = "Render Jinja2 NETCONF payload"
+    # Create a list with all configured VLANs which are not part of the inventory
+    remove_vlans = [x for x in current_vlans if x["id"] not in [x["vid"] for x in task.host["cfg_vlans"]]]
+
+    # If there are no VLANs to remove, return the Nornir result
+    if not remove_vlans:
+        custom_result.append(
+            f"{task_result(text=task_text, changed=False, level_name='INFO', failed=False)}\n"
+            + f"'{task_text}' -> NornirResponse <Success: True>\n"
+            + "-> No VLANs to remove"
+        )
+        return Result(host=task.host, custom_result=custom_result, failed=False)
+
+    # Set the info message for the Nornir result
+    info_msg = f"VLANs to remove: {len(remove_vlans)}"
+
+    # Render the Jinja2 payload template and configure the NETCONF candidate datastore
+    custom_result, task_failed = netconf_configure_jinja2_rendered_payload_template(
+        task=task,
+        j2_task_text=task_text,
+        j2_tpl_path="iosxe_netconf/tpl_sys/cleanup/cleanup_vlan.j2",
+        custom_result=custom_result,
+        task_failed=False,
+        info_msg=info_msg,
+        verbose=verbose,
+        j2_kwargs={"remove_vlans": remove_vlans},
+    )
+
+    # Return the Nornir NETCONF result
+    return Result(host=task.host, custom_result=custom_result, failed=task_failed)
+
+
 def nc_edit_tpl_config(task: Task, tpl_startswith: str, verbose: bool = False) -> Result:
     """
     TBD
@@ -477,7 +542,6 @@ def nc_cfg_cleanup(nr: Nornir, cfg_status: bool = True, verbose: bool = False) -
     """
     TBD
     """
-
     # Return False if cfg_status argument is False
     if not cfg_status:
         return False
@@ -485,6 +549,7 @@ def nc_cfg_cleanup(nr: Nornir, cfg_status: bool = True, verbose: bool = False) -
     tasks = {
         nc_edit_cleanup_portchannel: "NETCONF portchannel cleanup",
         nc_edit_cleanup_svi: "NETCONF vlan interface cleanup",
+        nc_edit_cleanup_vlan: "NETCONF vlan cleanup",
     }
 
     # Run each task from the Nornir tasks list

{nornir_collection-0.0.23 → nornir_collection-0.0.25}/nornir_collection/netbox/inventory.py

@@ -91,8 +91,8 @@ def _load_interface_data(task: Task, all_interfaces: List[Dict], enrich_vlan_dat
     # Dict keys for comprehension to include only some key-value pairs
     # fmt: off
     include_keys = [
-        "name", "int_template", "description", "type", "lag", "count_ipaddresses",
-        "count_fhrp_groups", "int_peer_device", "int_config_context", "enabled",
+        "name", "int_template", "description", "type", "lag", "mode", "untagged_vlan", "tagged_vlans",
+        "count_ipaddresses", "count_fhrp_groups", "int_peer_device", "int_config_context", "enabled",
     ]
     # fmt: on
     # Make some normalization to the device interfaces data with the interface_filter list and others

{nornir_collection-0.0.23 → nornir_collection-0.0.25}/nornir_collection.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: nornir-collection
-Version: 0.0.23
+Version: 0.0.25
 Summary: Nornir-Collection contains network automation functions and complete IaC workflows with         Nornir and other python libraries. It contains Nornir tasks and general functions in Nornir style.
 Author: Willi Kubny
 Author-email: willi.kubny@gmail.ch
@@ -47,6 +47,7 @@ Dynamic: author-email
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary

{nornir_collection-0.0.23 → nornir_collection-0.0.25}/setup.py

@@ -11,7 +11,7 @@ with open("requirements.txt", encoding="utf-8") as f:
 
 setuptools.setup(
     name="nornir-collection",
-    version="0.0.23",
+    version="0.0.25",
     author="Willi Kubny",
     author_email="willi.kubny@gmail.ch",
     description="Nornir-Collection contains network automation functions and complete IaC workflows with \