dpb/backend/app/api/v1/module_bre/propagation/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 (
    PropagationCreateSchema,
    PropagationOutSchema,
    PropagationQueryParam,
    PropagationUpdateSchema,
)
from .service import PropagationService

PropagationRouter = APIRouter(route_class=OperationLogRoute, prefix="/propagation", tags=["克隆扩繁批次"])


@PropagationRouter.get("/detail/{id}", summary="获取扩繁批次详情", response_model=ResponseSchema[PropagationOutSchema])
async def get_propagation__detail_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:detail"]))],
    id: Annotated[int, Path(description="扩繁批次ID")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(auth, db)
    result_dict = await service.detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取扩繁批次详情成功")


@PropagationRouter.get("/list", summary="分页查询扩繁批次", response_model=ResponseSchema[PageResultSchema[PropagationOutSchema]])
async def get_propagation__list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:query"]))],
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[PropagationQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(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="查询扩繁批次列表成功")


@PropagationRouter.get("/options", summary="扩繁批次下拉选项")
async def get_propagation__options_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(auth, db)
    options = await service.list_options()
    return SuccessResponse(data=options, msg="获取扩繁批次选项成功")


@PropagationRouter.post("/create", summary="创建扩繁批次", response_model=ResponseSchema[PropagationOutSchema])
async def create_propagation__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:create"]))],
    data: Annotated[PropagationCreateSchema, Body(description="创建参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(auth, db)
    result_dict = await service.create(data=data)
    return SuccessResponse(data=result_dict, msg="创建扩繁批次成功")


@PropagationRouter.put("/update/{id}", summary="修改扩繁批次", response_model=ResponseSchema[PropagationOutSchema])
async def update_propagation__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:update"]))],
    id: Annotated[int, Path(description="扩繁批次ID")],
    data: Annotated[PropagationUpdateSchema, Body(description="修改参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(auth, db)
    result_dict = await service.update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改扩繁批次成功")


@PropagationRouter.delete("/delete", summary="删除扩繁批次", response_model=ResponseSchema[None])
async def delete_propagation__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:delete"]))],
    ids: Annotated[list[int], Body(description="ID列表")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(auth, db)
    await service.delete(ids=ids)
    return SuccessResponse(msg="删除扩繁批次成功")


@PropagationRouter.post("/export", summary="导出扩繁批次")
async def export_propagation__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:export"]))],
    search: Annotated[PropagationQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
    service = PropagationService(auth, db)
    result_dict_list = await service.get_list(search=search)
    export_result = PropagationService.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')}"},
    )


@PropagationRouter.post("/import", summary="导入扩繁批次", response_model=ResponseSchema[ImportResultSchema])
async def import_propagation__controller(
    file: Annotated[UploadFile, File(description="导入文件")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:propagation:import"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = PropagationService(auth, db)
    batch_import_result = await service.batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入扩繁批次成功")


@PropagationRouter.post("/download/template", summary="获取扩繁批次导入模板", dependencies=[Depends(AuthPermission(["module_bre:propagation:download"]))])
async def download_propagation__template_controller() -> StreamingResponse:
    import_template_result = PropagationService.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",
        },
    )