Update cisco_client_sdwan.py

netascode · Jan 9, 2025 · 399a6ec · 399a6ec
1 parent 4e01b46
commit 399a6ec
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/nac_collector/cisco_client_sdwan.py b/nac_collector/cisco_client_sdwan.py
@@ -402,7 +402,7 @@ def extract_feature_parcel(
     ):
         parcel_type = parcel["parcelType"]
         if parcel_type.startswith(upstream_parcel_type):
-            parcel_type = parcel_type[len(upstream_parcel_type) :].lstrip("/")
+            parcel_type = parcel_type[len(upstream_parcel_type):].lstrip("/")
         parcel_id = parcel["parcelId"]
         new_endpoint = upstream_endpoint + "/" + parcel_type + "/" + parcel_id
         response = self.get_request(self.base_url + new_endpoint)
@@ -415,9 +415,12 @@ def extract_feature_parcel(
             children_endpoint_type = (
                 parcel_type + "/" + self.strip_backslash(children_endpoint["endpoint"])
             )
-            children_endpoint_type = self.strip_backslash(children_endpoint_type)
+            children_endpoint_type1 = self.strip_backslash(children_endpoint_type)
+            children_endpoint_type2 = self.strip_backslash(children_endpoint_type)
+            if children_endpoint_type.startswith(parcel_type):
+                children_endpoint_type2 = children_endpoint_type1[len(parcel_type):].lstrip("/")
             for subparcel in parcel.get("subparcels", []):
-                if subparcel["parcelType"] == children_endpoint_type:
+                if subparcel["parcelType"] in [children_endpoint_type1, children_endpoint_type2]:
                     children_entries.append(
                         self.extract_feature_parcel(
                             new_endpoint,