dpb/backend/scripts/_backup_query_param/germplasm__controller.py.bak

import urllib.parse
from typing import Annotated

from fastapi import APIRouter, Body, Depends, File, Path, Query, UploadFile, status
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
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 (
    BreedingGermplasmCreateSchema,
    BreedingGermplasmOutSchema,
    BreedingGermplasmQueryParam,
    BreedingGermplasmUpdateSchema,
)
from .service import BreedingGermplasmService

GermplasmRouter = APIRouter(route_class=OperationLogRoute, prefix="/germplasm", tags=["亲本资源"])


@GermplasmRouter.get("/detail/{id}", summary="获取亲本资源详情", response_model=ResponseSchema[BreedingGermplasmOutSchema])
async def get_obj_detail_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:detail"]))],
    id: Annotated[int, Path(description="亲本资源ID")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(auth, db)
    result_dict = await service.detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取亲本资源详情成功")


@GermplasmRouter.get("/list", summary="分页查询亲本资源", response_model=ResponseSchema[PageResultSchema[BreedingGermplasmOutSchema]])
async def get_obj_list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:query"]))],
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[BreedingGermplasmQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(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="查询亲本资源列表成功")


@GermplasmRouter.get("/options", summary="亲本资源下拉选项")
async def get_obj_options_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(auth, db)
    options = await service.list_options()
    return SuccessResponse(data=options, msg="获取亲本资源选项成功")


@GermplasmRouter.post("/create", status_code=status.HTTP_201_CREATED, summary="创建亲本资源", response_model=ResponseSchema[BreedingGermplasmOutSchema])
async def create_obj_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:create"]))],
    data: Annotated[BreedingGermplasmCreateSchema, Body(description="创建参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(auth, db)
    result_dict = await service.create(data=data)
    return SuccessResponse(data=result_dict, msg="创建亲本资源成功")


@GermplasmRouter.put("/update/{id}", summary="修改亲本资源", response_model=ResponseSchema[BreedingGermplasmOutSchema])
async def update_obj_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:update"]))],
    id: Annotated[int, Path(description="亲本资源ID")],
    data: Annotated[BreedingGermplasmUpdateSchema, Body(description="修改参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(auth, db)
    result_dict = await service.update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改亲本资源成功")


@GermplasmRouter.delete("/delete", summary="删除亲本资源", response_model=ResponseSchema[None])
async def delete_obj_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:delete"]))],
    ids: Annotated[list[int], Body(description="ID列表")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(auth, db)
    await service.delete(ids=ids)
    return SuccessResponse(msg="删除亲本资源成功")


@GermplasmRouter.post("/export", summary="导出亲本资源")
async def export_obj_list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:export"]))],
    search: Annotated[BreedingGermplasmQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
    service = BreedingGermplasmService(auth, db)
    result_dict_list = await service.get_list(search=search)
    export_result = BreedingGermplasmService.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')}"},
    )


@GermplasmRouter.post("/import", summary="导入亲本资源", response_model=ResponseSchema[str])
async def import_obj_list_controller(
    file: Annotated[UploadFile, File(description="导入文件")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:germplasm:import"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = BreedingGermplasmService(auth, db)
    batch_import_result = await service.batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入亲本资源成功")


@GermplasmRouter.post("/download/template", summary="获取亲本资源导入模板", dependencies=[Depends(AuthPermission(["module_bre:germplasm:download"]))])
async def export_obj_template_controller() -> StreamingResponse:
    import_template_result = BreedingGermplasmService.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",
        },
    )