добавил подсчёт количества рефералов
This commit is contained in:
@@ -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))
|
||||
Reference in New Issue
Block a user