Problem: KeyError occurs from netifaces

avoid network interfaces that doesn't have address info for netifaces.AF_INET
crypto-org-chain · Jun 21, 2024 · 5f1014e · 5f1014e
1 parent 903b590
commit 5f1014e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/testground/benchmark/benchmark/network.py b/testground/benchmark/benchmark/network.py
@@ -21,6 +21,7 @@ def get_data_ip(params: RunParams) -> ipaddress.IPv4Address:
 def ip4_addresses() -> List[ipaddress.IPv4Address]:
     ip_list = []
     for interface in netifaces.interfaces():
-        for link in netifaces.ifaddresses(interface)[netifaces.AF_INET]:
-            ip_list.append(ipaddress.IPv4Address(link["addr"]))
+        if netifaces.AF_INET in netifaces.ifaddresses(interface):
+            for link in netifaces.ifaddresses(interface)[netifaces.AF_INET]:
+                ip_list.append(ipaddress.IPv4Address(link["addr"]))
     return ip_list