wip

Signed-off-by: Patrick José Pereira <patrickelectric@gmail.com>

Author: patrickelectric

core/services/wifi/api/v2/routers/interfaces.py

@@ -1,4 +1,5 @@
-from typing import List
+import sys
+from typing import List, cast
 
 from fastapi import APIRouter, HTTPException, status
 from fastapi_versioning import versioned_api_route
@@ -15,12 +16,12 @@
 
 
 def get_wifi_manager() -> AbstractWifiManager:
-    # pylint: disable=import-outside-toplevel
-    from main import wifi_manager
-
-    if wifi_manager is None:
+    main_module = sys.modules.get("main") or sys.modules.get("__main__")
+    if main_module is None or not hasattr(main_module, "wifi_manager"):
+        raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail="WiFi manager module not found")
+    if main_module.wifi_manager is None:
         raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail="WiFi manager not initialized")
-    return wifi_manager
+    return cast(AbstractWifiManager, main_module.wifi_manager)
 
 
 @interfaces_router_v2.get("/", response_model=WifiInterfaceList, summary="List all WiFi interfaces.")

core/services/wifi/api/v2/routers/wifi.py

@@ -1,4 +1,5 @@
-from typing import List
+import sys
+from typing import List, cast
 
 from fastapi import APIRouter, HTTPException, status
 from fastapi_versioning import versioned_api_route
@@ -21,12 +22,12 @@
 
 
 def get_wifi_manager() -> AbstractWifiManager:
-    # pylint: disable=import-outside-toplevel
-    from main import wifi_manager
-
-    if wifi_manager is None:
+    main_module = sys.modules.get("main") or sys.modules.get("__main__")
+    if main_module is None or not hasattr(main_module, "wifi_manager"):
+        raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail="WiFi manager module not found")
+    if main_module.wifi_manager is None:
         raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail="WiFi manager not initialized")
-    return wifi_manager
+    return cast(AbstractWifiManager, main_module.wifi_manager)
 
 
 @wifi_router_v2.get(