Update compat.py

error handling to default to ndb calls if ipdb returns errors
svinota · Oct 6, 2023 · 47d74f4 · 47d74f4
1 parent 32f5fa0
commit 47d74f4
Showing 1 changed file with 39 additions and 18 deletions.
diff --git a/pyroute2/ndb/compat.py b/pyroute2/ndb/compat.py
@@ -18,30 +18,51 @@ def ipdb_interfaces_view(ndb):
         interface = record._as_dict()
         interface['ipdb_scope'] = 'system'
         interface['ipdb_priority'] = 0
-        interface['ipaddr'] = tuple(
-            (
-                (x.address, x.prefixlen)
-                for x in (
-                    ndb.addresses.dump().select_records(index=record.index)
+        try:
+            interface['ipaddr'] = tuple(
+                (
+                    (x.address, x.prefixlen)
+                    for x in (
+                        ndb.addresses.dump().select_records(index=record.index)
+                    )
                 )
             )
-        )
-        interface['ports'] = tuple(
-            (
-                x.index
-                for x in (
-                    ndb.interfaces.dump().select_records(master=record.index)
+        except:
+            with ndb.addresses.summary() as report:
+                report.select_records(ifname=f"{record.ifname}")
+                interface['ipaddr'] = tuple(
+                    ((x.address, x.prefixlen) for x in report)
+                )
+        try:
+            interface['ports'] = tuple(
+                (
+                    x.index
+                    for x in (
+                        ndb.interfaces.dump().select_records(master=record.index)
+                    )
                 )
             )
-        )
-        interface['neighbours'] = tuple(
-            (
-                x.dst
-                for x in (
-                    ndb.neighbours.dump().select_records(ifindex=record.index)
+        except:
+            with ndb.interfaces.dump() as report:
+                report.select_records(ifname=f"{record.ifname}")
+                interface['ports'] = tuple(
+                    (x.index for x in report)
+                )
+        try:
+            interface['neighbours'] = tuple(
+                (
+                    x.dst
+                    for x in (
+                        ndb.neighbours.dump().select_records(ifindex=record.index)
+                    )
                 )
             )
-        )
+        except:
+            with ndb.neighbours.dump() as report:
+                report.select_records(ifindex=record.index)
+                interface['neighbours'] = tuple(
+                    (x.dst for x in report)
+                )
         ret[record.ifname] = interface
 
     return ret