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,135 @@
+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 (
+        TrialCreateSchema,
+        TrialOutSchema,
+        TrialQueryParam,
+        TrialUpdateSchema,
+)
+from .service import TrialService
+
+TrialRouter = APIRouter(route_class=OperationLogRoute, prefix="/trial", tags=["试验"])
+
+
+@TrialRouter.get("/detail/{id}", summary="获取试验详情", response_model=ResponseSchema[TrialOutSchema])
+async def get_trial__detail_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:detail"]))],
+    id: Annotated[int, Path(description="试验ID")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(auth, db)
+    result_dict = await service.detail(id=id)
+    return SuccessResponse(data=result_dict, msg="获取试验详情成功")
+
+
+@TrialRouter.get("/list", summary="分页查询试验", response_model=ResponseSchema[PageResultSchema[TrialOutSchema]])
+async def get_trial__list_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:query"]))],
+    page: Annotated[PaginationQueryParam, Depends()],
+    search: Annotated[TrialQueryParam, Query()],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(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="查询试验列表成功")
+
+
+@TrialRouter.get("/options", summary="试验下拉选项")
+async def get_trial__options_controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:query"]))],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(auth, db)
+    options = await service.list_options()
+    return SuccessResponse(data=options, msg="获取试验选项成功")
+
+
+@TrialRouter.post("/create", summary="创建试验", response_model=ResponseSchema[TrialOutSchema])
+async def create_trial__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:create"]))],
+    data: Annotated[TrialCreateSchema, Body(description="创建参数")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(auth, db)
+    result_dict = await service.create(data=data)
+    return SuccessResponse(data=result_dict, msg="创建试验成功")
+
+
+@TrialRouter.put("/update/{id}", summary="修改试验", response_model=ResponseSchema[TrialOutSchema])
+async def update_trial__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:update"]))],
+    id: Annotated[int, Path(description="试验ID")],
+    data: Annotated[TrialUpdateSchema, Body(description="修改参数")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(auth, db)
+    result_dict = await service.update(id=id, data=data)
+    return SuccessResponse(data=result_dict, msg="修改试验成功")
+
+
+@TrialRouter.delete("/delete", summary="删除试验", response_model=ResponseSchema[None])
+async def delete_trial__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:delete"]))],
+    ids: Annotated[list[int], Body(description="ID列表")],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(auth, db)
+    await service.delete(ids=ids)
+    return SuccessResponse(msg="删除试验成功")
+
+
+@TrialRouter.post("/export", summary="导出试验")
+async def export_trial__controller(
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:export"]))],
+    search: Annotated[TrialQueryParam, Query()],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> StreamingResponse:
+    service = TrialService(auth, db)
+    result_dict_list = await service.get_list(search=search)
+    export_result = TrialService.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')}"},
+    )
+
+
+@TrialRouter.post("/import", summary="导入试验", response_model=ResponseSchema[ImportResultSchema])
+async def import_trial__controller(
+    file: Annotated[UploadFile, File(description="导入文件")],
+    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:trial:import"]))],
+    db: Annotated[AsyncSession, Depends(db_getter)],
+) -> JSONResponse:
+    service = TrialService(auth, db)
+    batch_import_result = await service.batch_import(file=file, update_support=True)
+    return SuccessResponse(data=batch_import_result, msg="导入试验成功")
+
+
+@TrialRouter.post("/download/template", summary="获取试验导入模板", dependencies=[Depends(AuthPermission(["module_bre:trial:download"]))])
+async def download_trial__template_controller() -> StreamingResponse:
+    import_template_result = TrialService.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",
+        },
+    )
+