add refresh

app/core/database.py

@@ -50,18 +50,37 @@ def init_db():
     conn.commit()
     conn.close()
 
-def add_session(login, access_token, refresh_token):
+def add_session(login, access_token, refresh_token, update_existing=False):
     """Добавляет новую сессию или обновляет существующую."""
     conn = get_connection()
     cursor = conn.cursor()
-    # REPLACE INTO - удобный способ для вставки или обновления
+    
+    if update_existing:
+        # Обновляем существующую сессию по access_token
+        cursor.execute('''
+            UPDATE sessions 
+            SET access_token = ?, refresh_token = ?, created_at = ?
+            WHERE access_token = ?
+        ''', (access_token, refresh_token, datetime.now(), access_token))
+    else:
+        # Вставляем новую или заменяем существующую по логину
         cursor.execute('''
             INSERT OR REPLACE INTO sessions (login, access_token, refresh_token, created_at)
             VALUES (?, ?, ?, ?)
         ''', (login, access_token, refresh_token, datetime.now()))
+    
     conn.commit()
     conn.close()
 
+def logout(access_token: str):
+    """Удаляет сессию по токену доступа."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute('DELETE FROM sessions WHERE access_token = ?', (access_token,))
+    conn.commit()
+    conn.close()
+
+
 def get_session(login: str):
     """Получает сессию по логину."""
     conn = get_connection()
@@ -80,13 +99,6 @@ def get_all_sessions():
     conn.close()
     return sessions
 
-def delete_session(login: str):
-    """Удаляет сессию по логину."""
-    conn = get_connection()
-    cursor = conn.cursor()
-    cursor.execute('DELETE FROM sessions WHERE login = ?', (login,))
-    conn.commit()
-    conn.close()
 
 def get_last_login():
     """Получает логин последнего вошедшего пользователя."""

app/core/services/auth_service.py

@@ -1,7 +1,7 @@
 import httpx
 import asyncio
 from app.core import config
-from app.core.database import add_session
+from app.core.database import add_session, logout, get_session
 from app.core.localizer import localizer
 
 async def login(login, password):
@@ -91,50 +91,90 @@ async def register(login, password, invite=None):
         return False, f"{localizer.translate('Произошла ошибка')}: {e}"
 
 
-async def get_user_role(access_token: str):
+async def refresh_token(access_token: str, refresh_token: str):
     """
-    Получает роль и права пользователя по токену доступа.
+    Обновляет токен доступа, используя токен обновления.
 
-    :param access_token: Токен доступа пользователя
+    :param access_token: Истекший токен доступа
-    :return: Кортеж (успех: bool, данные: UserRoleData | str)
+    :param refresh_token: Токен обновления
+    :return: Кортеж (успех: bool, данные: dict | str)
     """
-    url = f"{config.BASE_URL}/v1/user/role"
+    url = f"{config.BASE_URL}/v1/auth/token/refresh"
-    headers = {"Authorization": f"Bearer {access_token}"}
+    payload = {"access_token": access_token, "refresh_token": refresh_token}
 
-    timeout = httpx.Timeout(connect=5.0, read=10.0, write=5.0, pool=5.0)
     try:
-        async with httpx.AsyncClient(http2=True, timeout=timeout) as client:
+        async with httpx.AsyncClient(http2=True) as client:
-            response = await client.get(url, headers=headers)
+            response = await client.post(url, json=payload)
-            print("headers", headers,"response", response, "status", response.json().get("status"))
 
             if response.status_code == 200:
                 data = response.json()
                 if data.get("status") == "fine":
-                    # Здесь можно добавить валидацию через Pydantic, если необходимо
+                    token_data = data["data"]
-                    # from app.core.models.user_models import UserRoleData
+                    # Обновляем сессию с новыми токенами
-                    # user_role_data = UserRoleData(**data['data'])
+                    add_session(
-                    return True, data['data']
+                        login=None,  # Логин не требуется для обновления
+                        access_token=token_data["access_token"],
+                        refresh_token=token_data["refresh_token"],
+                        update_existing=True
+                    )
+                    return True, token_data
                 else:
-                    return False, data.get("detail", localizer.translate("Неизвестная ошибка ответа"))
+                    return False, data.get("detail", "Unknown error")
 
             elif response.status_code == 401:
-                return False, localizer.translate("Токен недействителен или истек")
+                return False, "Refresh token is invalid or expired"
-            elif response.status_code == 404:
+            
-                return False, localizer.translate("Пользователь не найден")
             else:
-                return False, f"{localizer.translate('Ошибка сервера')}: {response.status_code}"
+                return False, f"Server error: {response.status_code}"
 
     except httpx.RequestError as e:
-        return False, f"{localizer.translate('Ошибка сети')}: {e}"
+        return False, f"Network error: {e}"
     except Exception as e:
-        return False, f"{localizer.translate('Произошла ошибка')}: {e}"
+        return False, f"An error occurred: {e}"
 
 
-# Пример использования (для тестирования)
+async def get_user_role(access_token: str, login: str):
-async def main():
+    """
-    # Замените на реальные данные для теста
+    Получает роль и права пользователя по токену доступа.
-    success, message = await login("testuser", "testpassword") 
+    В случае истечения срока действия токена, пытается его обновить.
-    print(f"Результат входа: {success}, Сообщение: {message}")
+    """
+    url = f"{config.BASE_URL}/v1/user/role"
+    headers = {"Authorization": f"Bearer {access_token}"}
 
-if __name__ == "__main__":
+    try:
-    asyncio.run(main())
+        async with httpx.AsyncClient(http2=True) as client:
+            response = await client.get(url, headers=headers)
+
+            if response.status_code == 200:
+                data = response.json()
+                return (True, data['data']) if data.get("status") == "fine" else (False, data.get("detail", "Unknown error"))
+
+            elif response.status_code == 401:
+                # Токен истек, пытаемся обновить
+                session = get_session(login)
+                if not session or not session['refresh_token']:
+                    return False, "No refresh token found"
+
+                refresh_success, refresh_data = await refresh_token(access_token, session['refresh_token'])
+                
+                if refresh_success:
+                    # Повторяем запрос с новым токеном
+                    new_access_token = refresh_data['access_token']
+                    headers["Authorization"] = f"Bearer {new_access_token}"
+                    response = await client.get(url, headers=headers)
+                    
+                    if response.status_code == 200:
+                        data = response.json()
+                        return (True, data['data']) if data.get("status") == "fine" else (False, "Failed to get role after refresh")
+                
+                # Если обновление не удалось, выходим из системы
+                logout(access_token)
+                return False, "Session expired, please log in again"
+
+            else:
+                return False, f"Server error: {response.status_code}"
+
+    except httpx.RequestError as e:
+        return False, f"Network error: {e}"
+    except Exception as e:
+        return False, f"An error occurred: {e}"

app/ui/views/yobble_home_view.py

@@ -305,7 +305,7 @@ class YobbleHomeView(QWidget):
             print("[Permissions] Preload failed: No access token.")
             return
 
-        success, data = await get_user_role(access_token)
+        success, data = await get_user_role(access_token, self.username)
         if success:
             user_permissions = set(data.get("user_permissions", []))
 
@@ -358,7 +358,7 @@ class YobbleHomeView(QWidget):
             self.show_error_message(localizer.translate("Сессия не найдена. Пожалуйста, войдите снова."))
             return
 
-        success, data = await get_user_role(access_token)
+        success, data = await get_user_role(access_token, self.username)
         if success and permission_code in data.get("user_permissions", []):
             # self.permission_cache.add(permission_code)
             user_permissions = set(data.get("user_permissions", []))