forked from HSE_team/BetterCallPraskovia
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
"""
|
||
API роутеры для работы с пользователями
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from uuid import UUID
|
||
from fastapi import APIRouter, status, Depends
|
||
from fastapi.responses import JSONResponse
|
||
from typing import List, Annotated
|
||
from dishka.integrations.fastapi import FromDishka, inject
|
||
from src.presentation.schemas.user_schemas import UserCreate, UserUpdate, UserResponse
|
||
from src.application.use_cases.user_use_cases import UserUseCases
|
||
from src.domain.entities.user import User
|
||
|
||
router = APIRouter(prefix="/users", tags=["users"])
|
||
|
||
|
||
@router.post("", response_model=UserResponse, status_code=status.HTTP_201_CREATED)
|
||
@inject
|
||
async def create_user(
|
||
user_data: UserCreate,
|
||
use_cases: UserUseCases = FromDishka()
|
||
) -> UserResponse:
|
||
"""Создать пользователя"""
|
||
user = await use_cases.create_user(
|
||
telegram_id=user_data.telegram_id,
|
||
role=user_data.role
|
||
)
|
||
return UserResponse.from_entity(user)
|
||
|
||
|
||
@router.get("/me", response_model=UserResponse)
|
||
@inject
|
||
async def get_current_user_info(
|
||
current_user: User = FromDishka()
|
||
):
|
||
"""Получить информацию о текущем пользователе"""
|
||
return UserResponse.from_entity(current_user)
|
||
|
||
|
||
@router.get("/telegram/{telegram_id}", response_model=UserResponse)
|
||
@inject
|
||
async def get_user_by_telegram_id(
|
||
telegram_id: str,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Получить пользователя по Telegram ID"""
|
||
user = await use_cases.get_user_by_telegram_id(telegram_id)
|
||
if not user:
|
||
from fastapi import HTTPException
|
||
raise HTTPException(status_code=404, detail=f"Пользователь с telegram_id {telegram_id} не найден")
|
||
return UserResponse.from_entity(user)
|
||
|
||
|
||
@router.post("/telegram/{telegram_id}/increment-questions", response_model=UserResponse)
|
||
@inject
|
||
async def increment_questions(
|
||
telegram_id: str,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Увеличить счетчик использованных вопросов"""
|
||
user = await use_cases.increment_questions_used(telegram_id)
|
||
return UserResponse.from_entity(user)
|
||
|
||
|
||
@router.post("/telegram/{telegram_id}/activate-premium", response_model=UserResponse)
|
||
@inject
|
||
async def activate_premium(
|
||
telegram_id: str,
|
||
days: int = 30,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Активировать premium статус"""
|
||
user = await use_cases.activate_premium(telegram_id, days=days)
|
||
return UserResponse.from_entity(user)
|
||
|
||
|
||
@router.get("/{user_id}", response_model=UserResponse)
|
||
@inject
|
||
async def get_user(
|
||
user_id: UUID,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Получить пользователя по ID"""
|
||
user = await use_cases.get_user(user_id)
|
||
return UserResponse.from_entity(user)
|
||
|
||
|
||
@router.put("/{user_id}", response_model=UserResponse)
|
||
@inject
|
||
async def update_user(
|
||
user_id: UUID,
|
||
user_data: UserUpdate,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Обновить пользователя"""
|
||
user = await use_cases.update_user(
|
||
user_id=user_id,
|
||
telegram_id=user_data.telegram_id,
|
||
role=user_data.role
|
||
)
|
||
return UserResponse.from_entity(user)
|
||
|
||
|
||
@router.delete("/{user_id}", status_code=status.HTTP_204_NO_CONTENT)
|
||
@inject
|
||
async def delete_user(
|
||
user_id: UUID,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Удалить пользователя"""
|
||
await use_cases.delete_user(user_id)
|
||
return JSONResponse(status_code=status.HTTP_204_NO_CONTENT, content=None)
|
||
|
||
|
||
@router.get("", response_model=List[UserResponse])
|
||
@inject
|
||
async def list_users(
|
||
skip: int = 0,
|
||
limit: int = 100,
|
||
use_cases: UserUseCases = FromDishka()
|
||
):
|
||
"""Получить список пользователей"""
|
||
users = await use_cases.list_users(skip=skip, limit=limit)
|
||
return [UserResponse.from_entity(user) for user in users]
|
||
|