FastapiAdmin/backend/app/api/v1/module_system/user/controller.py

import urllib.parse
from typing import Annotated

from fastapi import APIRouter, Body, Depends, Path, UploadFile
from fastapi.responses import JSONResponse, StreamingResponse
from sqlalchemy.ext.asyncio import AsyncSession

from app.common.response import ResponseSchema, StreamResponse, SuccessResponse
from app.core.base_params import PaginationQueryParam
from app.core.base_schema import AuthSchema, BatchSetAvailable, PageResultSchema
from app.core.dependencies import AuthPermission, db_getter, get_current_user
from app.core.logger import logger
from app.core.router_class import OperationLogRoute
from app.utils.common_util import bytes2file_response

from .schema import (
    CurrentUserUpdateSchema,
    ResetPasswordSchema,
    UserChangePasswordSchema,
    UserCreateSchema,
    UserForgetPasswordSchema,
    UserOutSchema,
    UserQueryParam,
    UserRegisterSchema,
    UserUpdateSchema,
)
from .service import UserService

UserRouter = APIRouter(route_class=OperationLogRoute, prefix="/user", tags=["系统管理", "用户管理"])

@UserRouter.get(
    "/current/info",
    summary="查询当前用户信息",
    response_model=ResponseSchema[UserOutSchema],
)
async def get_current_user_info_controller(
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    user_dict = await UserService(auth).current_info()
    return SuccessResponse(data=user_dict, msg="获取当前用户信息成功")

@UserRouter.put(
    "/current/info/update",
    summary="更新当前用户基本信息",
    response_model=ResponseSchema[UserOutSchema],
)
async def update_current_user_info_controller(
    data: CurrentUserUpdateSchema,
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    result_dict = await UserService(auth).update_current_info(data=data)
    return SuccessResponse(data=result_dict, msg="更新当前用户基本信息成功")

@UserRouter.put(
    "/password/change",
    summary="修改当前用户密码",
    response_model=ResponseSchema[UserOutSchema],
)
async def change_current_user_password_controller(
    data: UserChangePasswordSchema,
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    result_dict = await UserService(auth).change_password(data=data)
    return SuccessResponse(data=result_dict, msg="修改密码成功, 请重新登录")

@UserRouter.put(
    "/password/reset/{id}",
    summary="重置用户密码",
    response_model=ResponseSchema[UserOutSchema],
)
async def reset_password_controller(
    id: Annotated[int, Path(description="用户ID")],
    data: ResetPasswordSchema,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:update"]))],
) -> JSONResponse:
    data.id = id
    result_dict = await UserService(auth).reset_password(data=data)
    return SuccessResponse(data=result_dict, msg="重置密码成功")

@UserRouter.post(
    "/register",
    summary="注册用户",
    response_model=ResponseSchema[UserOutSchema],
)
async def register_user_controller(
    data: UserRegisterSchema,
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    auth = AuthSchema(db=db, check_data_scope=False)
    user_register_result = await UserService(auth).register(data=data)
    logger.info(f"{data.username} 注册用户成功: {user_register_result}")
    return SuccessResponse(data=user_register_result, msg="注册用户成功")

@UserRouter.post(
    "/password/forget",
    summary="忘记密码",
    response_model=ResponseSchema[UserOutSchema],
)
async def forget_password_controller(
    data: UserForgetPasswordSchema,
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    auth = AuthSchema(db=db, check_data_scope=False)
    user_forget_password_result = await UserService(auth).forget_password(data=data)
    logger.info(f"{data.username} 重置密码成功: {user_forget_password_result}")
    return SuccessResponse(data=user_forget_password_result, msg="重置密码成功")

@UserRouter.get(
    "/list",
    summary="查询用户",
    response_model=ResponseSchema[PageResultSchema[UserOutSchema]],
)
async def get_user_list_controller(
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[UserQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:query"]))],
) -> JSONResponse:
    result_dict = await UserService(auth).page(
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by,
    )
    return SuccessResponse(data=result_dict, msg="查询用户成功")

@UserRouter.get(
    "/detail/{id}",
    summary="查询用户详情",
    response_model=ResponseSchema[UserOutSchema],
)
async def get_user_detail_controller(
    id: Annotated[int, Path(description="用户ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:detail"]))],
) -> JSONResponse:
    result_dict = await UserService(auth).detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取用户详情成功")

@UserRouter.post(
    "/create",
    summary="创建用户",
    response_model=ResponseSchema[UserOutSchema],
)
async def create_user_controller(
    data: UserCreateSchema,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:create"]))],
) -> JSONResponse:
    result_dict = await UserService(auth).create(data=data)
    return SuccessResponse(data=result_dict, msg="创建用户成功")

@UserRouter.put(
    "/update/{id}",
    summary="修改用户",
    response_model=ResponseSchema[UserOutSchema],
)
async def update_user_controller(
    data: UserUpdateSchema,
    id: Annotated[int, Path(description="用户ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:update"]))],
) -> JSONResponse:
    result_dict = await UserService(auth).update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改用户成功")

@UserRouter.delete(
    "/delete",
    summary="删除用户",
    response_model=ResponseSchema[None],
)
async def delete_user_controller(
    ids: Annotated[list[int], Body(description="ID列表")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:delete"]))],
) -> JSONResponse:
    await UserService(auth).delete(ids=ids)
    return SuccessResponse(msg="删除用户成功")

@UserRouter.patch(
    "/status/batch",
    summary="批量修改用户状态",
    response_model=ResponseSchema[None],
)
async def batch_set_available_user_controller(
    data: BatchSetAvailable,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:patch"]))],
) -> JSONResponse:
    await UserService(auth).set_available(data=data)
    return SuccessResponse(msg="批量修改用户状态成功")

@UserRouter.get(
    "/import/template",
    summary="获取用户导入模板",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_system:user:download"]))],
)
async def export_user_import_template_controller() -> StreamingResponse:
    user_import_template_result = UserService.get_import_template()

    return StreamResponse(
        data=bytes2file_response(user_import_template_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={
            "Content-Disposition": f"attachment; filename={urllib.parse.quote('用户导入模板.xlsx')}",
            "Access-Control-Expose-Headers": "Content-Disposition",
        },
    )

@UserRouter.get(
    "/export",
    summary="导出用户",
    response_model=ResponseSchema[None],
)
async def export_user_list_controller(
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[UserQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:export"]))],
) -> StreamingResponse:
    user_list = await UserService(auth).get_list(search=search, order_by=page.order_by)
    user_export_result = UserService.export_list(user_list=user_list)

    return StreamResponse(
        data=bytes2file_response(user_export_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={"Content-Disposition": "attachment; filename=user.xlsx"},
    )

@UserRouter.post(
    "/import/data",
    summary="导入用户",
    response_model=ResponseSchema[None],
)
async def import_user_list_controller(
    file: UploadFile,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_system:user:import"]))],
) -> JSONResponse:
    batch_import_result = await UserService(auth).batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入用户成功")