save tokens endpoint

Controls/API/endpoints/hub/HubManager.py

@@ -38,3 +38,9 @@ class HubManager:
 
     def wifi(self, ssid: str, password: str):
         WiFi.save_and_connect(ssid, password)
+
+    def set_tokens(tokens_pair: schemas.TokensPair):
+        with open(f'{path}/{resources}/access_token.txt', 'w') as f:
+            f.write(tokens_pair.access_token)
+        with open(f'{path}/{resources}/refresh_token.txt', 'w') as f:
+            f.write(tokens_pair.refresh_token)

Controls/API/endpoints/hub/router.py

@@ -2,7 +2,7 @@ from uuid import UUID
 from fastapi import APIRouter, Depends
 import Controls.API.exceptions
 from .HubManager import HubManager
-from .schemas import Hub, PatchHub
+from .schemas import Hub, PatchHub, TokensPair
 
 
 router = APIRouter(
@@ -25,3 +25,7 @@ async def hub_patch(hub: PatchHub, manager: HubManager = Depends()):
 @router.post('/wifi')
 async def hub_wifi(ssid: str, password: str, manager: HubManager = Depends()):
     manager.wifi(ssid, password)
+
+@router.post('set_tokens')
+async def set_tokens(tokens: TokensPair):
+    manager.save_tokens(tokens)

Controls/API/endpoints/hub/schemas.py

@@ -12,3 +12,7 @@ class Hub(BaseModel):
 
     class Config:
         orm_mode = True
+
+class TokensPair(BaseModel):
+    access_token: str
+    refresh_token: str