Встроил марзбан в бекенд, исправил бывшие проблемы с получением активной подписки

app/routes/subscription_routes.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 from fastapi import APIRouter, HTTPException, Depends
 from pydantic import BaseModel
 from app.services import DatabaseManager
@@ -15,15 +15,17 @@ router = APIRouter()
 
 class BuySubscriptionRequest(BaseModel):
     telegram_id: int
-    plan_id: str
+    plan_name: str
 
 class SubscriptionResponse(BaseModel):
-    id: str
-    plan: str
-    vpn_server_id: str
-    expiry_date: str
+    id: str 
+    user_id: int
+    plan_name: str
+    vpn_server_id: Optional[str]
+    status: str
+    start_date: str
+    end_date: str  
     created_at: str
-    updated_at: str
 
 # Эндпоинт для покупки подписки
 @router.post("/subscription/buy", response_model=dict)
@@ -35,23 +37,34 @@ async def buy_subscription(
     Покупка подписки.
     """
     try:
-        result = await database_manager.buy_sub(request_data.telegram_id, request_data.plan_id)
+        logger.info(f"Получен запрос на покупку подписки: {request_data.dict()}")
+        
+        result = await database_manager.buy_sub(request_data.telegram_id, request_data.plan_name)
 
-        if result == "ERROR":
-            raise HTTPException(status_code=500, detail="ERROR")
+        logger.info(f"Результат buy_sub: {result}")
+
+        if result == "ERROR" or result is None:
+            raise HTTPException(status_code=500, detail="Internal server error")
         elif result == "INSUFFICIENT_FUNDS":
             raise HTTPException(status_code=400, detail="INSUFFICIENT_FUNDS")
         elif result == "TARIFF_NOT_FOUND":
             raise HTTPException(status_code=400, detail="TARIFF_NOT_FOUND")
         elif result == "ACTIVE_SUBSCRIPTION_EXISTS":
             raise HTTPException(status_code=400, detail="ACTIVE_SUBSCRIPTION_EXISTS")
-        result = await database_manager.generate_uri(request_data.telegram_id)
-        return {"message": result}
+        
+        # Если успешно, генерируем URI
+        if isinstance(result, dict) and result.get('status') == 'OK':
+            uri_result = await database_manager.generate_uri(request_data.telegram_id)
+            logger.info(f"Результат генерации URI: {uri_result}")
+            return {"status": "success", "subscription_id": result.get('subscription_id'), "uri": uri_result[0]}
+        else:
+            return {"status": "success", "message": "Subscription created"}
+            
     except HTTPException as http_exc:
-        # Пропускаем HTTPException, чтобы FastAPI обработал его корректно
+        logger.error(f"HTTPException в buy_subscription: {http_exc.detail}")
         raise http_exc
     except Exception as e:
-        # Обрабатываем остальные исключения
+        logger.error(f"Неожиданная ошибка в buy_subscription: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Unexpected error: {str(e)}")
 
 
@@ -63,21 +76,24 @@ async def last_subscription(telegram_id: int, database_manager: DatabaseManager
     """
     logger.info(f"Получение последней подписки для пользователя: {telegram_id}")
     try:
-        subscriptions = await database_manager.get_last_subscriptions(telegram_id=telegram_id, limit=1)
+        subscription = await database_manager.get_last_subscriptions(telegram_id=telegram_id)
 
-        if not subscriptions:
+        plan = await database_manager.get_plan_by_id(subscription.plan_id)
+
+        if not subscription or not plan:
             logger.warning(f"Подписки для пользователя {telegram_id} не найдены")
             raise HTTPException(status_code=404, detail="No subscriptions found")
         
-        sub = subscriptions[0]
 
         return {
-            "id": sub.id,
-            "plan": sub.plan,
-            "vpn_server_id": sub.vpn_server_id,
-            "expiry_date": sub.expiry_date.isoformat(),
-            "created_at": sub.created_at.isoformat(),
-            "updated_at": sub.updated_at.isoformat(),
+            "id": str(subscription.id),
+            "user_id": subscription.user_id,
+            "plan_name": plan.name,
+            "vpn_server_id": subscription.vpn_server_id,
+            "status": subscription.status.value,
+            "start_date": subscription.start_date.isoformat(),
+            "end_date": subscription.end_date.isoformat(),
+            "created_at": subscription.created_at.isoformat(),
         }
     except SQLAlchemyError as e:
         logger.error(f"Ошибка базы данных при получении подписки для пользователя {telegram_id}: {e}")
@@ -135,10 +151,10 @@ async def get_uri(telegram_id: int, database_manager: DatabaseManager = Depends(
         if uri == "SUB_ERROR":
             raise HTTPException(status_code=404, detail="SUB_ERROR")
         if not uri:
-            logger.warning(f"Не удалось сгенерировать URI для пользователя с telegram_id {telegram_id}")
+            logger.warning(f"Не удалось сгенерировать URI для пользователя с telegram_id {telegram_id}, данные -> {uri}")
             raise HTTPException(status_code=404, detail="URI not found")
 
-        return {"detail": uri}
+        return {"detail": uri[0]}
 
     except HTTPException as e:
         # Пропускаем HTTPException, чтобы FastAPI обработал её автоматически