init: 初始化 dpb 桃育种系统代码库

前后端 + 后端 FastAPI 全量源码、部署脚本与文档。
2026-08-06 00:17:49 +08:00
commit b95053c52c
1469 changed files with 322298 additions and 0 deletions
@@ -0,0 +1,158 @@
+import urllib.parse
+from typing import Annotated
+from datetime import date as _date
+
+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 (
+        PollenCreateSchema,
+        PollenOutSchema,
+        PollenQueryParam,
+        PollenUpdateSchema,
+)
+from .service import PollenService
+
+PollenRouter = APIRouter(route_class=OperationLogRoute, prefix="/pollen", tags=["花粉档案"])
+
+
+@PollenRouter.get("/detail/{id}", summary="获取花粉档案详情", response_model=ResponseSchema[PollenOutSchema])
+async def get_pollen__detail_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:detail"]))],
+    id: Annotated[int, Path(description="花粉档案ID")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    result_dict = await service.detail(id=id)
+    return SuccessResponse(data=result_dict, msg="获取花粉详情成功")
+
+
+@PollenRouter.get("/list", summary="分页查询花粉档案", response_model=ResponseSchema[PageResultSchema[PollenOutSchema]])
+async def get_pollen__list_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:query"]))],
+    page: Annotated[PaginationQueryParam, Depends()],
+    search: Annotated[PollenQueryParam, Query()],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(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="查询花粉列表成功")
+
+
+@PollenRouter.get("/options", summary="花粉档案下拉选项")
+async def get_pollen__options_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:query"]))],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    options = await service.list_options()
+    return SuccessResponse(data=options, msg="获取花粉选项成功")
+
+
+@PollenRouter.get("/available_lots", summary="授粉窗口：查询当前有效期内可用花粉批次")
+async def get_pollen__available_lots_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:query"]))],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+    date_on: Annotated[_date | None, Query(description="查询日期（缺省今天）")] = None,
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    lots = await service.available_lots(date_on=date_on)
+    return SuccessResponse(data=lots, msg="获取可用花粉批次成功")
+
+
+@PollenRouter.get("/window-plan", summary="授粉窗口规划：花期×花粉库存覆盖分析")
+async def get_pollen__window_plan_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:query"]))],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+    window_start: Annotated[_date, Query(description="窗口起始")],
+    window_end: Annotated[_date, Query(description="窗口结束")],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    plan = await service.window_plan(window_start=window_start, window_end=window_end)
+    return SuccessResponse(data=plan, msg="生成授粉窗口规划成功")
+
+
+@PollenRouter.post("/create", summary="创建花粉档案", response_model=ResponseSchema[PollenOutSchema])
+async def create_pollen__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:create"]))],
+    data: Annotated[PollenCreateSchema, Body(description="创建参数")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    result_dict = await service.create(data=data)
+    return SuccessResponse(data=result_dict, msg="创建花粉成功")
+
+
+@PollenRouter.put("/update/{id}", summary="修改花粉档案", response_model=ResponseSchema[PollenOutSchema])
+async def update_pollen__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:update"]))],
+    id: Annotated[int, Path(description="花粉档案ID")],
+    data: Annotated[PollenUpdateSchema, Body(description="修改参数")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    result_dict = await service.update(id=id, data=data)
+    return SuccessResponse(data=result_dict, msg="修改花粉成功")
+
+
+@PollenRouter.delete("/delete", summary="删除花粉档案", response_model=ResponseSchema[None])
+async def delete_pollen__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:delete"]))],
+    ids: Annotated[list[int], Body(description="ID列表")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    await service.delete(ids=ids)
+    return SuccessResponse(msg="删除花粉成功")
+
+
+@PollenRouter.post("/export", summary="导出花粉档案")
+async def export_pollen__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:export"]))],
+    search: Annotated[PollenQueryParam, Query()],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> StreamingResponse:
+    service = PollenService(auth, db)
+    result_dict_list = await service.get_list(search=search)
+    export_result = PollenService.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')}"},
+    )
+
+
+@PollenRouter.post("/import", summary="导入花粉档案", response_model=ResponseSchema[ImportResultSchema])
+async def import_pollen__controller(
+    file: Annotated[UploadFile, File(description="导入文件")],
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pollen:import"]))],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = PollenService(auth, db)
+    batch_import_result = await service.batch_import(file=file, update_support=True)
+    return SuccessResponse(data=batch_import_result, msg="导入花粉成功")
+
+
+@PollenRouter.post("/download/template", summary="获取花粉档案导入模板", dependencies=[Depends(AuthPermission(["module_bre:pollen:download"]))])
+async def download_pollen__template_controller() -> StreamingResponse:
+    import_template_result = PollenService.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",
+        },
+    )