Переделал модель БД под новую, переделал Репозиторий, переделал сервисы, убрал монгодб, изменил необходимые пакеты, марзбан я добавил, но не настроил. Весь старый бот вроде работает(только в рефералке не уверен)

This commit is contained in:
root
2025-11-24 23:43:40 +03:00
parent f0f3b96005
commit e975bf4774
18 changed files with 637 additions and 1217 deletions

View File

@@ -1,7 +1,7 @@
import sys
from fastapi import APIRouter, Depends, HTTPException
from fastapi.exceptions import HTTPException
from app.services.db_manager import DatabaseManager
from app.services import DatabaseManager
from sqlalchemy.exc import SQLAlchemyError
from instance.configdb import get_database_manager
from pydantic import BaseModel
@@ -20,20 +20,19 @@ router = APIRouter()
# Модели запросов и ответов
class CreateUserRequest(BaseModel):
telegram_id: str
referrer_id: Optional[str] = None
telegram_id: int
invited_by: Optional[int] = None
class UserResponse(BaseModel):
id: UUID
telegram_id: str
telegram_id: int
username: Optional[str]
balance: float
referrer_id: Optional[str]
invited_by: Optional[int] = None
created_at: str
updated_at: str
class AddReferal(BaseModel):
new_user_id: str
invited_id: int
@router.post("/user/create", response_model=UserResponse, summary="Создать пользователя")
async def create_user(
@@ -44,16 +43,15 @@ async def create_user(
Создание пользователя через Telegram ID.
"""
try:
user = await db_manager.create_user(request.telegram_id,request.referrer_id)
if user == "ERROR":
user = await db_manager.create_user(request.telegram_id,request.invited_by)
if user == None:
raise HTTPException(status_code=500, detail="Failed to create user")
return UserResponse(
id=user.id,
telegram_id=user.telegram_id,
username=user.username,
balance=user.balance,
referrer_id=user.referrer_id if user.referrer_id is not None else None,
invited_by=user.invited_by if user.invited_by is not None else None,
created_at=user.created_at.isoformat(),
updated_at=user.updated_at.isoformat()
)
@@ -64,7 +62,7 @@ async def create_user(
@router.get("/user/{telegram_id}", response_model=UserResponse, summary="Получить информацию о пользователе")
async def get_user(
telegram_id: str,
telegram_id: int,
db_manager: DatabaseManager = Depends(get_database_manager)
):
"""
@@ -77,13 +75,12 @@ async def get_user(
logger.warning(f"Пользователь с telegram_id {telegram_id} не найден.")
raise HTTPException(status_code=404, detail="User not found")
print(f"Пользователь найден: ID={user.id}, Username={user.username}")
print(f"Пользователь найден: ID={user.telegram_id}, Username={user.username}")
user_response = UserResponse(
id=user.id,
telegram_id=user.telegram_id,
username=user.username,
balance=user.balance,
referrer_id=user.referrer_id if user.referrer_id is not None else None,
invited_by=user.invited_by if user.invited_by is not None else None,
created_at=user.created_at.isoformat(),
updated_at=user.updated_at.isoformat()
)
@@ -105,7 +102,7 @@ async def get_user(
@router.post("/user/{telegram_id}/balance/{amount}", summary="Обновить баланс")
async def update_balance(
telegram_id: str,
telegram_id: int,
amount: float,
db_manager: DatabaseManager = Depends(get_database_manager)
):
@@ -132,52 +129,55 @@ async def update_balance(
raise HTTPException(status_code=500, detail=str(e))
@router.get("/user/{user_id}/transactions", summary="Последние транзакции пользователя")
@router.get("/user/{telegram_id}/transactions", summary="Последние транзакции пользователя")
async def last_transactions(
user_id: UUID,
telegram_id: int,
db_manager: DatabaseManager = Depends(get_database_manager)
):
"""
Возвращает список последних транзакций пользователя.
"""
logger.info(f"Получен запрос на транзакции для пользователя: {user_id}")
logger.info(f"Получен запрос на транзакции для пользователя: {telegram_id}")
try:
logger.debug(f"Вызов метода get_transaction с user_id={user_id}")
transactions = await db_manager.get_transaction(user_id)
logger.debug(f"Вызов метода get_transaction с user_id={telegram_id}")
transactions = await db_manager.get_transaction(telegram_id)
if transactions == "ERROR":
logger.error(f"Ошибка при получении транзакций для пользователя: {user_id}")
logger.error(f"Ошибка при получении транзакций для пользователя: {telegram_id}")
raise HTTPException(status_code=500, detail="Failed to fetch transactions")
logger.debug(f"Транзакции для {user_id}: {transactions}")
if transactions == None:
response = []
logger.info(f"Формирование ответа для пользователя {telegram_id}: {response}")
return response
logger.debug(f"Транзакции для {telegram_id}: {transactions}")
response = [
{
"id": tx.id,
"amount": tx.amount,
"created_at": tx.created_at.isoformat(),
"transaction_type": tx.transaction_type,
"type": tx.type,
} for tx in transactions
]
logger.info(f"Формирование ответа для пользователя {user_id}: {response}")
logger.info(f"Формирование ответа для пользователя {telegram_id}: {response}")
return response
except HTTPException as http_ex:
logger.warning(f"HTTP ошибка для {user_id}: {http_ex.detail}")
logger.warning(f"HTTP ошибка для {telegram_id}: {http_ex.detail}")
raise http_ex
except SQLAlchemyError as db_ex:
logger.error(f"Ошибка базы данных для {user_id}: {db_ex}")
logger.error(f"Ошибка базы данных для {telegram_id}: {db_ex}")
raise HTTPException(status_code=500, detail="Database error")
except Exception as e:
logger.exception(f"Неожиданная ошибка для {user_id}: {e}")
logger.exception(f"Неожиданная ошибка для {telegram_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,
referrer_id: int,
request: AddReferal,
db_manager: DatabaseManager = Depends(get_database_manager)
):
@@ -186,12 +186,12 @@ async def add_referal(
"""
logger.info(f"Получен запрос на добавление реферала: telegram_id={referrer_id}")
try:
result = await db_manager.add_referal(referrer_id)
result = await db_manager.add_referal(referrer_id,request.invited_id)
if result == "ERROR":
logger.error(f"Ошибка добавления реферала для {referrer_id} c айди {request.new_user_id}")
logger.error(f"Ошибка добавления реферала для {referrer_id} c айди {request.invited_id}")
raise HTTPException(status_code=500, detail="Failed to update balance")
logger.info(f"Добавлен реферал для {referrer_id} c айди {request.new_user_id}")
logger.info(f"Добавлен реферал для {referrer_id} c айди {request.invited_id}")
return {"message": "Balance updated successfully"}
except HTTPException as http_ex:
logger.warning(f"HTTP ошибка: {http_ex.detail}")