Рабочая версия

app/routes/user_routes.py

@@ -1,3 +1,4 @@
+import sys
 from fastapi import APIRouter, Depends, HTTPException
 from fastapi.exceptions import HTTPException
 from app.services.db_manager import DatabaseManager
@@ -9,8 +10,12 @@ from uuid import UUID
 import logging
 
 logger = logging.getLogger(__name__)
-router = APIRouter()
-
+if not logger.handlers:
+    handler = logging.StreamHandler(sys.stdout)
+    handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
+    logger.addHandler(handler)
+    logger.setLevel(logging.INFO)
+    logger.propagate = False
 router = APIRouter()
 
 # Модели запросов и ответов
@@ -27,6 +32,8 @@ class UserResponse(BaseModel):
     created_at: str
     updated_at: str
 
+class AddReferal(BaseModel):
+    new_user_id: str
 
 @router.post("/user/create", response_model=UserResponse, summary="Создать пользователя")
 async def create_user(
@@ -64,15 +71,14 @@ async def get_user(
     Получение информации о пользователе.
     """
     try:
-        logger.info(f"Получение пользователя с telegram_id: {telegram_id}")
+        print(f"Получение пользователя с telegram_id: {telegram_id}")
         user = await db_manager.get_user_by_telegram_id(telegram_id)
         if not user:
             logger.warning(f"Пользователь с telegram_id {telegram_id} не найден.")
             raise HTTPException(status_code=404, detail="User not found")
 
-        logger.info(f"Пользователь найден: ID={user.id}, Username={user.username}")
-
-        return UserResponse(
+        print(f"Пользователь найден: ID={user.id}, Username={user.username}")
+        user_response = UserResponse(
             id=user.id,
             telegram_id=user.telegram_id,
             username=user.username,
@@ -81,7 +87,6 @@ async def get_user(
             created_at=user.created_at.isoformat(),
             updated_at=user.updated_at.isoformat()
         )
-        logger.debug(f"Формирование ответа для пользователя: {user_response}")
 
         return user_response
 
@@ -167,3 +172,33 @@ async def last_transactions(
     except Exception as e:
         logger.exception(f"Неожиданная ошибка для {user_id}: {e}")
         raise HTTPException(status_code=500, detail=str(e))
+
+
+
+@router.post("/user/{referrer_id}/add_referral", summary="Обновить баланс")
+async def add_referal(
+    referrer_id: str, 
+    request: AddReferal,
+    db_manager: DatabaseManager = Depends(get_database_manager)
+):
+    """
+    Обновляет баланс пользователя.
+    """
+    logger.info(f"Получен запрос на добавление реферала: telegram_id={referrer_id}")
+    try:
+        result = await db_manager.add_referal(referrer_id)
+        if result == "ERROR":
+            logger.error(f"Ошибка добавления реферала для {referrer_id} c айди {request.new_user_id}")
+            raise HTTPException(status_code=500, detail="Failed to update balance")
+        
+        logger.info(f"Добавлен реферал для {referrer_id} c айди {request.new_user_id}")
+        return {"message": "Balance updated successfully"}
+    except HTTPException as http_ex:
+        logger.warning(f"HTTP ошибка: {http_ex.detail}")
+        raise http_ex
+    except SQLAlchemyError as db_ex:
+        logger.error(f"Ошибка базы данных при добавлении рефералу {referrer_id}: {db_ex}")
+        raise HTTPException(status_code=500, detail="Database error")
+    except Exception as e:
+        logger.exception(f"Неожиданная ошибка при добавлении рефералу {referrer_id}: {e}")
+        raise HTTPException(status_code=500, detail=str(e))