Переделал модель БД под новую, переделал Репозиторий, переделал сервисы, убрал монгодб, изменил необходимые пакеты, марзбан я добавил, но не настроил. Весь старый бот вроде работает(только в рефералке не уверен)
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from typing import List
|
||||
from fastapi import APIRouter, HTTPException, Depends
|
||||
from pydantic import BaseModel
|
||||
from app.services.db_manager import DatabaseManager
|
||||
from app.services import DatabaseManager
|
||||
from instance.configdb import get_database_manager
|
||||
from uuid import UUID
|
||||
import logging
|
||||
@@ -14,7 +14,7 @@ router = APIRouter()
|
||||
|
||||
|
||||
class BuySubscriptionRequest(BaseModel):
|
||||
telegram_id: str
|
||||
telegram_id: int
|
||||
plan_id: str
|
||||
|
||||
class SubscriptionResponse(BaseModel):
|
||||
@@ -56,17 +56,17 @@ async def buy_subscription(
|
||||
|
||||
|
||||
# Эндпоинт для получения последней подписки
|
||||
@router.get("/subscription/{user_id}/last", response_model=SubscriptionResponse)
|
||||
async def last_subscription(user_id: UUID, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
@router.get("/subscription/{telegram_id}/last", response_model=SubscriptionResponse)
|
||||
async def last_subscription(telegram_id: int, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
"""
|
||||
Возвращает последнюю подписку пользователя.
|
||||
"""
|
||||
logger.info(f"Получение последней подписки для пользователя: {user_id}")
|
||||
logger.info(f"Получение последней подписки для пользователя: {telegram_id}")
|
||||
try:
|
||||
subscriptions = await database_manager.get_last_subscriptions(user_id=user_id, limit=1)
|
||||
subscriptions = await database_manager.get_last_subscriptions(telegram_id=telegram_id, limit=1)
|
||||
|
||||
if not subscriptions:
|
||||
logger.warning(f"Подписки для пользователя {user_id} не найдены")
|
||||
logger.warning(f"Подписки для пользователя {telegram_id} не найдены")
|
||||
raise HTTPException(status_code=404, detail="No subscriptions found")
|
||||
|
||||
sub = subscriptions[0]
|
||||
@@ -80,7 +80,7 @@ async def last_subscription(user_id: UUID, database_manager: DatabaseManager = D
|
||||
"updated_at": sub.updated_at.isoformat(),
|
||||
}
|
||||
except SQLAlchemyError as e:
|
||||
logger.error(f"Ошибка базы данных при получении подписки для пользователя {user_id}: {e}")
|
||||
logger.error(f"Ошибка базы данных при получении подписки для пользователя {telegram_id}: {e}")
|
||||
raise HTTPException(status_code=500, detail="Database error")
|
||||
except HTTPException as e:
|
||||
# Пропускаем HTTPException, чтобы FastAPI обработал её автоматически
|
||||
@@ -89,18 +89,18 @@ async def last_subscription(user_id: UUID, database_manager: DatabaseManager = D
|
||||
logger.error(f"Неожиданная ошибка: {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)):
|
||||
@router.get("/subscriptions/{telegram_id}", response_model=List[SubscriptionResponse])
|
||||
async def get_subscriptions(telegram_id: int, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
"""
|
||||
Возвращает список подписок пользователя.
|
||||
"""
|
||||
logger.info(f"Получение подписок для пользователя: {user_id}")
|
||||
logger.info(f"Получение подписок для пользователя: {telegram_id}")
|
||||
try:
|
||||
# Получаем подписки без ограничений или с указанным лимитом
|
||||
subscriptions = await database_manager.last_subscriptions(user_id=str(user_id))
|
||||
subscriptions = await database_manager.get_last_subscriptions(telegram_id=telegram_id)
|
||||
|
||||
if not subscriptions:
|
||||
logger.warning(f"Подписки для пользователя {user_id} не найдены")
|
||||
logger.warning(f"Подписки для пользователя {telegram_id} не найдены")
|
||||
raise HTTPException(status_code=404, detail="No subscriptions found")
|
||||
|
||||
# Формируем список подписок для ответа
|
||||
@@ -116,17 +116,15 @@ async def get_subscriptions(user_id: UUID, database_manager: DatabaseManager = D
|
||||
for sub in subscriptions
|
||||
]
|
||||
except SQLAlchemyError as e:
|
||||
logger.error(f"Ошибка базы данных при получении подписок для пользователя {user_id}: {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))
|
||||
except HTTPException as e:
|
||||
# Пропускаем HTTPException, чтобы FastAPI обработал её автоматически
|
||||
raise e
|
||||
|
||||
|
||||
@router.get("/uri", response_model=dict)
|
||||
async def get_uri(telegram_id: str, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
async def get_uri(telegram_id: int, database_manager: DatabaseManager = Depends(get_database_manager)):
|
||||
"""
|
||||
Возвращает список подписок пользователя.
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user