Подключение к точкам доступа Wi-Fi

SRV/api/v1/subnet.py

@@ -2,9 +2,10 @@ from fastapi import APIRouter, HTTPException
 from typing import List, Dict
 from models.subnet import SubnetScanResult
 from models.wifi import WifiScanResult
-from models.request import NoteRequest
+from models.request import NoteRequest, WifiConnectRequest, WifiDisconnectRequest
 from services.subnet_service import SubnetService
 from services.wifi_service import WifiService
+from services.connect_service import WifiConnectionService
 
 router = APIRouter()
 
@@ -128,4 +129,31 @@ def delete_wifi_note(ssid: str):
     except HTTPException as e:
         raise e
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}")
+
+@router.post("/wifi/connect")
+def connect_to_wifi(request: WifiConnectRequest):
+    """Подключается к Wi-Fi сети по SSID и паролю."""
+    wifi_service = WifiConnectionService()
+    success, message = wifi_service.connect(request.ssid, request.password)
+    if not success:
+        raise HTTPException(status_code=400, detail=message)
+    return {"message": f"Успешно подключено к {request.ssid}"}
+
+@router.post("/wifi/disconnect")
+def disconnect_from_wifi(request: WifiDisconnectRequest):
+    """Отключается от Wi-Fi сети по указанному SSID."""
+    wifi_service = WifiConnectionService()
+    success, message = wifi_service.disconnect(request.ssid)
+    if not success:
+        raise HTTPException(status_code=400, detail=message)
+    return {"message": message or f"Успешно отключено от {request.ssid}"}
+
+@router.get("/wifi/status")
+def get_wifi_status():
+    """Проверяет статус текущего подключения к Wi-Fi."""
+    wifi_service = WifiConnectionService()
+    success, message = wifi_service.get_status()
+    if not success:
+        raise HTTPException(status_code=400, detail=message)
+    return {"status": "connected", "ssid": message}