use plain dict in IOSDriver.get_bgp_neighbors_detail()

napalm-automation · Jan 28, 2025 · c625e4a · c625e4a
1 parent 9a539a8
commit c625e4a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/napalm/ios/ios.py b/napalm/ios/ios.py
@@ -2076,7 +2076,7 @@ def get_bgp_neighbors(self):
         return bgp_neighbor_data
 
     def get_bgp_neighbors_detail(self, neighbor_address=""):
-        bgp_detail = defaultdict(lambda: defaultdict(lambda: []))
+        bgp_detail = {}
 
         raw_bgp_sum = self._send_command("show ip bgp all sum").strip()
 
@@ -2227,6 +2227,14 @@ def get_bgp_neighbors_detail(self, neighbor_address=""):
                         "export_policy": bgp_neigh_afi["policy_out"],
                     }
                 )
+
+            vrf_name = details["routing_table"]
+            if vrf_name not in bgp_detail.keys():
+                bgp_detail[vrf_name] = {}
+            remote_as = details["remote_as"]
+            if remote_as not in bgp_detail[vrf_name].keys():
+                bgp_detail[vrf_name][remote_as] = []
+
             bgp_detail[details["routing_table"]][details["remote_as"]].append(details)
         return bgp_detail