Vladless
65 lines
2.6 KiB
Python
65 lines
2.6 KiB
Python
import asyncio
|
|
|
|
from fastapi import Depends, HTTPException, Path
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from api.depends import get_session, verify_admin_token
|
|
from api.v1.routes.base_crud import generate_crud_router
|
|
from api.v1.schemas.users import UserBase, UserResponse, UserUpdate
|
|
from database import async_session_maker, delete_user_data, get_servers
|
|
from database.access.resolution import resolve_user_optional
|
|
from database.models import Key, User
|
|
from logger import logger
|
|
from services.operations import delete_key_from_cluster
|
|
|
|
|
|
router = generate_crud_router(
|
|
model=User,
|
|
schema_response=UserResponse,
|
|
schema_create=UserBase,
|
|
schema_update=UserUpdate,
|
|
identifier_field="tg_id",
|
|
enabled_methods=["get_all", "get_one", "get_by_email", "create", "update"],
|
|
)
|
|
|
|
|
|
@router.delete("/{tg_id}", response_model=dict)
|
|
async def delete_user(
|
|
tg_id: int = Path(..., description="Telegram ID пользователя"),
|
|
admin=Depends(verify_admin_token),
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
try:
|
|
u = await resolve_user_optional(session, tg_id)
|
|
if u is None:
|
|
raise HTTPException(status_code=404, detail="Пользователь не найден")
|
|
result = await session.execute(select(Key.email, Key.client_id).where(Key.user_id == u.id))
|
|
key_records = result.all()
|
|
|
|
async with async_session_maker() as s:
|
|
servers = await get_servers(session=s)
|
|
cluster_ids = list(servers.keys())
|
|
|
|
async def _delete_one(cluster_id: str, email: str, client_id: str):
|
|
async with async_session_maker() as s:
|
|
await delete_key_from_cluster(cluster_id, email, client_id, s)
|
|
|
|
try:
|
|
tasks = [
|
|
_delete_one(cluster_id, email, client_id)
|
|
for email, client_id in key_records
|
|
for cluster_id in cluster_ids
|
|
]
|
|
await asyncio.gather(*tasks, return_exceptions=True)
|
|
except Exception as e:
|
|
logger.error(f"[DELETE] Ошибка при удалении ключей с серверов для пользователя {tg_id}: {e}")
|
|
|
|
await delete_user_data(session, tg_id)
|
|
|
|
return {"detail": f"Пользователь {tg_id} и его ключи успешно удалены."}
|
|
|
|
except Exception as e:
|
|
logger.error(f"[DELETE] Ошибка при удалении пользователя {tg_id}: {e}")
|
|
raise HTTPException(status_code=500, detail="Ошибка при удалении пользователя")
|