добавил подсчёт количества рефералов

This commit is contained in:
2025-12-05 13:58:45 +03:00
parent eb9e00b27c
commit f16cb3bc2e
3 changed files with 56 additions and 1 deletions

View File

@@ -170,4 +170,27 @@ async def add_referal(
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))
@router.get("/user/{telegram_id}/referrals", summary="Получить колво рефералов")
async def get_referral_count(
telegram_id: int,
db_manager: DatabaseManager = Depends(get_database_manager)
):
try:
result = await db_manager.get_referrals_count(telegram_id)
if result == "ERROR":
logger.error(f"Ошибка получения рефералов для пользователя: {telegram_id}")
raise HTTPException(status_code=500, detail="Failed to get referrals")
logger.info(f"Количество приглашённых {result}")
return {"invited_count": result}
except HTTPException as http_ex:
logger.warning(f"HTTP ошибка: {http_ex.detail}")
raise http_ex
except SQLAlchemyError as db_ex:
logger.error(f"Ошибка базы данных при получении рефералов пользователя {telegram_id}: {db_ex}")
raise HTTPException(status_code=500, detail="Database error")
except Exception as e:
logger.exception(f"Неожиданная ошибка при получении рефералов {telegram_id}: {e}")
raise HTTPException(status_code=500, detail=str(e))