dpb/backend/app/api/v1/module_bre/personnel/controller.py

import urllib.parse
from typing import Annotated

from fastapi import APIRouter, Body, Depends, File, Path, Query, 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_schema import AuthSchema, PageResultSchema, PaginationQueryParam, ImportResultSchema
from app.core.dependencies import AuthPermission, db_getter
from app.core.router_class import OperationLogRoute
from app.utils.common_util import bytes2file_response

from .schema import (
        PersonnelCreateSchema,
        PersonnelOutSchema,
        PersonnelQueryParam,
        PersonnelUpdateSchema,
)
from .service import PersonnelService

PersonnelRouter = APIRouter(route_class=OperationLogRoute, prefix="/personnel", tags=["人员花名册"])


@PersonnelRouter.get("/detail/{id}", summary="获取人员花名册详情", response_model=ResponseSchema[PersonnelOutSchema])
async def get_personnel__detail_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:detail"]))],
    id: Annotated[int, Path(description="人员花名册ID")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    result_dict = await service.detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取人员花名册详情成功")


@PersonnelRouter.get("/list", summary="分页查询人员花名册", response_model=ResponseSchema[PageResultSchema[PersonnelOutSchema]])
async def get_personnel__list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:query"]))],
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[PersonnelQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    result_dict = await service.page(
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by,
    )
    return SuccessResponse(data=result_dict, msg="查询人员花名册列表成功")


@PersonnelRouter.get("/options", summary="人员花名册下拉选项")
async def get_personnel__options_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    options = await service.list_options()
    return SuccessResponse(data=options, msg="获取人员花名册选项成功")


@PersonnelRouter.post("/create", summary="创建人员花名册", response_model=ResponseSchema[PersonnelOutSchema])
async def create_personnel__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:create"]))],
    data: Annotated[PersonnelCreateSchema, Body(description="创建参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    result_dict = await service.create(data=data)
    return SuccessResponse(data=result_dict, msg="创建人员花名册成功")


@PersonnelRouter.put("/update/{id}", summary="修改人员花名册", response_model=ResponseSchema[PersonnelOutSchema])
async def update_personnel__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:update"]))],
    id: Annotated[int, Path(description="人员花名册ID")],
    data: Annotated[PersonnelUpdateSchema, Body(description="修改参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    result_dict = await service.update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改人员花名册成功")


@PersonnelRouter.delete("/delete", summary="删除人员花名册", response_model=ResponseSchema[None])
async def delete_personnel__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:delete"]))],
    ids: Annotated[list[int], Body(description="ID列表")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    await service.delete(ids=ids)
    return SuccessResponse(msg="删除人员花名册成功")


@PersonnelRouter.post("/export", summary="导出人员花名册")
async def export_personnel__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:export"]))],
    search: Annotated[PersonnelQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
    service = PersonnelService(auth, db)
    result_dict_list = await service.get_list(search=search)
    export_result = PersonnelService.batch_export(obj_list=[item.model_dump() for item in result_dict_list])
    return StreamResponse(
        data=bytes2file_response(export_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={"Content-Disposition": f"attachment; filename={urllib.parse.quote('人员花名册管理.xlsx')}"},
    )


@PersonnelRouter.post("/import", summary="导入人员花名册", response_model=ResponseSchema[ImportResultSchema])
async def import_personnel__controller(
    file: Annotated[UploadFile, File(description="导入文件")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:personnel:import"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PersonnelService(auth, db)
    batch_import_result = await service.batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入人员花名册成功")


@PersonnelRouter.post("/download/template", summary="获取人员花名册导入模板", dependencies=[Depends(AuthPermission(["module_bre:personnel:download"]))])
async def download_personnel__template_controller() -> StreamingResponse:
    import_template_result = PersonnelService.import_template_download()
    return StreamResponse(
        data=bytes2file_response(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",
        },
    )