Дополнение логики подключения к WiFi, если профиль оказывается битый

SRV/services/arp_service.py

@@ -24,6 +24,8 @@ from netaddr import IPNetwork
 from utils.i18n import get_translations, _
 from babel.support import Translations
 
+logger = logging.getLogger(__name__)
+
 try:
     from scapy.layers.l2 import ARP, Ether
     from scapy.sendrecv import srp
@@ -53,35 +55,57 @@ class NetworkScanner:
 
         Инициализирует интерфейс, информацию об IP-адресах и локальный MAC-адрес.
         """
-        try:
-            if interface:
-                self.interface = interface
-            else:
-                default_gateway = netifaces.gateways().get("default", {}).get(
-                    netifaces.AF_INET)
-                if default_gateway:
-                    self.interface = default_gateway[1]
-                else:
-                    raise KeyError("Default gateway not found")
+        self.translations = translations
+        self.interface = None
+        self.ip_info = []
+        self.networks = []
+        self.local_mac = None
 
-            self.ip_info = netifaces.ifaddresses(self.interface).get(
-                netifaces.AF_INET, [])
-            self.networks = [
-                IPNetwork(f"{ip['addr']}/{ip['netmask']}") for ip in self.ip_info
-            ]
-            self.local_mac = netifaces.ifaddresses(
-                self.interface)[netifaces.AF_LINK][0]["addr"]
+        try:
+            candidate_interface = interface
+            if not candidate_interface:
+                gateways = netifaces.gateways()
+                default_gateway = gateways.get("default", {}).get(netifaces.AF_INET)
+                if default_gateway:
+                    candidate_interface = default_gateway[1]
+                else:
+                    for iface in netifaces.interfaces():
+                        if iface == "lo":
+                            continue
+                        addrs = netifaces.ifaddresses(iface)
+                        if netifaces.AF_INET in addrs and addrs[netifaces.AF_INET]:
+                            candidate_interface = iface
+                            break
+
+            if candidate_interface:
+                self.interface = candidate_interface
+                addresses = netifaces.ifaddresses(self.interface)
+                self.ip_info = addresses.get(netifaces.AF_INET, [])
+                self.networks = [
+                    IPNetwork(
+                        f"{ip['addr']}/{ip['netmask']}"
+                    ) for ip in self.ip_info if ip.get("addr") and ip.get("netmask")
+                ]
+                link_info = addresses.get(netifaces.AF_LINK, [])
+                if link_info:
+                    self.local_mac = link_info[0].get("addr")
+            else:
+                logger.warning(
+                    "NetworkScanner: IPv4 интерфейс не найден, сканирование будет недоступно"
+                )
         except Exception as exc:
             if os.getenv("SPHINX_BUILD"):
                 self.interface = interface or "eth0"
                 self.ip_info = []
                 self.networks = []
                 self.local_mac = "00:00:00:00:00:00"
-                logger = logging.getLogger(__name__)
-                logger.warning("NetworkScanner falling back to mocked interface during documentation build: %s", exc)
+                logger.warning(
+                    "NetworkScanner falling back to mocked interface during documentation build: %s",
+                    exc,
+                )
             else:
+                logger.error("NetworkScanner инициализация завершилась ошибкой: %s", exc)
                 raise
-        self.translations = translations
 
     def _create_arp_packet(self, target_ip):
         """Создаёт ARP-пакет для отправки на указанный IP-адрес.
@@ -112,6 +136,16 @@ class NetworkScanner:
         Note:
             Учитывает локальные IP-адреса интерфейса и удаляет дубликаты.
         """
+        if not self.interface or not self.networks or ARP is None or Ether is None:
+            fallback_clients = []
+            if self.ip_info:
+                for ip_info in self.ip_info:
+                    fallback_clients.append({
+                        "ip": ip_info.get("addr"),
+                        "mac": self.local_mac,
+                    })
+            return [client for client in fallback_clients if client["ip"]]
+
         all_clients = []
 
         for network in self.networks:
@@ -132,7 +166,12 @@ class NetworkScanner:
             all_clients.extend(clients)
 
         for ip_info in self.ip_info:
-            local_client = {"ip": ip_info["addr"], "mac": self.local_mac}
+            if not ip_info.get("addr"):
+                continue
+            local_client = {
+                "ip": ip_info["addr"],
+                "mac": self.local_mac,
+            }
             if local_client not in all_clients:
                 all_clients.append(local_client)