Рефакторинга чуть чуть вроде
This commit is contained in:
@@ -38,17 +38,21 @@ async def buy_subscription(
|
||||
result = await database_manager.buy_sub(request_data.telegram_id, request_data.plan_id)
|
||||
|
||||
if result == "ERROR":
|
||||
raise HTTPException(status_code=500, detail="Failed to buy subscription")
|
||||
raise HTTPException(status_code=500, detail="ERROR")
|
||||
elif result == "INSUFFICIENT_FUNDS":
|
||||
raise HTTPException(status_code=400, detail="Insufficient funds")
|
||||
raise HTTPException(status_code=400, detail="INSUFFICIENT_FUNDS")
|
||||
elif result == "TARIFF_NOT_FOUND":
|
||||
raise HTTPException(status_code=400, detail="Tariff not found")
|
||||
raise HTTPException(status_code=400, detail="TARIFF_NOT_FOUND")
|
||||
elif result == "ACTIVE_SUBSCRIPTION_EXISTS":
|
||||
raise HTTPException(status_code=400, detail="User already had subscription",)
|
||||
|
||||
return {"message": "Subscription purchased successfully"}
|
||||
raise HTTPException(status_code=400, detail="ACTIVE_SUBSCRIPTION_EXISTS")
|
||||
result = await database_manager.generate_uri(request_data.telegram_id)
|
||||
return {"message": result}
|
||||
except HTTPException as http_exc:
|
||||
# Пропускаем HTTPException, чтобы FastAPI обработал его корректно
|
||||
raise http_exc
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
# Обрабатываем остальные исключения
|
||||
raise HTTPException(status_code=500, detail=f"Unexpected error: {str(e)}")
|
||||
|
||||
|
||||
# Эндпоинт для получения последней подписки
|
||||
@@ -59,11 +63,12 @@ async def last_subscription(user_id: UUID, database_manager: DatabaseManager = D
|
||||
"""
|
||||
logger.info(f"Получение последней подписки для пользователя: {user_id}")
|
||||
try:
|
||||
subscriptions = await database_manager.last_subscriptions(user_id=str(user_id), limit=1)
|
||||
subscriptions = await database_manager.get_last_subscriptions(user_id=user_id, limit=1)
|
||||
|
||||
if not subscriptions:
|
||||
logger.warning(f"Подписки для пользователя {user_id} не найдены")
|
||||
raise HTTPException(status_code=404, detail="No subscriptions found")
|
||||
|
||||
sub = subscriptions[0]
|
||||
|
||||
return {
|
||||
@@ -77,9 +82,12 @@ async def last_subscription(user_id: UUID, database_manager: DatabaseManager = D
|
||||
except SQLAlchemyError as e:
|
||||
logger.error(f"Ошибка базы данных при получении подписки для пользователя {user_id}: {e}")
|
||||
raise HTTPException(status_code=500, detail="Database error")
|
||||
except HTTPException as e:
|
||||
# Пропускаем HTTPException, чтобы FastAPI обработал её автоматически
|
||||
raise e
|
||||
except Exception as e:
|
||||
logger.error(f"Неожиданная ошибка: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
raise HTTPException(status_code=500, detail="Internal Server Error")
|
||||
|
||||
@router.get("/subscriptions/{user_id}", response_model=List[SubscriptionResponse])
|
||||
async def get_subscriptions(user_id: UUID, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
@@ -114,3 +122,25 @@ async def get_subscriptions(user_id: UUID, database_manager: DatabaseManager = D
|
||||
logger.error(f"Неожиданная ошибка: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
@router.get("/uri", response_model=dict)
|
||||
async def get_uri(telegram_id: int, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
"""
|
||||
Возвращает список подписок пользователя.
|
||||
"""
|
||||
logger.info(f"Получение подписок для пользователя: {telegram_id}")
|
||||
try:
|
||||
# Получаем подписки без ограничений или с указанным лимитом
|
||||
uri = await database_manager.generate_uri(telegram_id)
|
||||
if uri == "SUB_ERROR":
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
if not uri:
|
||||
logger.warning(f"Не удалось сгенерировать URI для пользователя с telegram_id {telegram_id}")
|
||||
raise HTTPException(status_code=404, detail="URI not found")
|
||||
|
||||
return {"detail": uri }
|
||||
except SQLAlchemyError as e:
|
||||
logger.error(f"Ошибка базы данных при получении подписок для пользователя {telegram_id}: {e}")
|
||||
raise HTTPException(status_code=500, detail="Database error")
|
||||
except Exception as e:
|
||||
logger.error(f"Неожиданная ошибка: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
Reference in New Issue
Block a user