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

前后端 + 后端 FastAPI 全量源码、部署脚本与文档。
This commit is contained in:
34047007@qq.com
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",
},
)
@@ -0,0 +1,18 @@
from typing import Any
from sqlalchemy.ext.asyncio import AsyncSession
from app.core.base_crud import CRUDBase
from app.core.base_schema import AuthSchema
from .model import TrialModel
class BreedingTrialCRUD(CRUDBase[TrialModel, Any, Any]):
"""试验 CRUD —— 直接复用 CRUDBase(已自动注入数据权限过滤)。"""
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
super().__init__(TrialModel, auth, db)
trial_crud = BreedingTrialCRUD
@@ -0,0 +1,43 @@
"""试验 数据模型"""
from datetime import datetime
from sqlalchemy import Float, ForeignKey, Index, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column
from app.core.base_model import MappedBase, ModelMixin, UserMixin
class TrialModel(ModelMixin, UserMixin, MappedBase):
"""试验 主数据表。"""
__tablename__ = "bre_trial"
trial_name: Mapped[str] = mapped_column(String(128), nullable=False, comment="试验名称")
plan_id: Mapped[int | None] = mapped_column(
Integer, ForeignKey("bre_plan.id", ondelete="SET NULL"), nullable=True, index=True,
comment="所属育种计划", default=None,
)
target_id: Mapped[int | None] = mapped_column(
Integer, ForeignKey("bre_target.id", ondelete="SET NULL"), nullable=True, index=True,
comment="服务育种目标", default=None,
)
trial_type: Mapped[str | None] = mapped_column(String(32), nullable=True, comment="试验类型", default=None)
design_type: Mapped[str | None] = mapped_column(String(32), nullable=True, comment="设计类型", default=None)
start_year: Mapped[int | None] = mapped_column(Integer, nullable=True, comment="起始年份", default=None)
end_year: Mapped[int | None] = mapped_column(Integer, nullable=True, comment="结束年份", default=None)
objective: Mapped[str | None] = mapped_column(Text, nullable=True, comment="试验目的", default=None)
remark: Mapped[str | None] = mapped_column(Text, nullable=True, comment="备注", default=None)
# 无 status 列:覆盖基类默认 ix_<表>_status_deleted 索引,
# 仅保留 (created_time, is_deleted) 复合索引用于数据权限过滤。
__table_args__ = (
Index("ix_bre_trial_created_deleted", "created_time", "is_deleted"),
)
@@ -0,0 +1,62 @@
from app.core.base_schema import CommonSchema
"""试验 —— Pydantic 校验/序列化模型。"""
from datetime import datetime
from pydantic import BaseModel, ConfigDict, Field
class TrialBaseSchema(BaseModel):
model_config = ConfigDict(from_attributes=True)
trial_name: str = Field(..., description="试验名称")
plan_id: int | None = Field(default=None, description="所属育种计划 id")
target_id: int | None = Field(default=None, description="服务育种目标 id")
trial_type: str | None = Field(default=None, description="试验类型")
design_type: str | None = Field(default=None, description="设计类型")
start_year: int | None = Field(default=None, description="起始年份")
end_year: int | None = Field(default=None, description="结束年份")
objective: str | None = Field(default=None, description="试验目的")
remark: str | None = Field(default=None, description="备注")
class TrialCreateSchema(TrialBaseSchema):
pass
class TrialUpdateSchema(TrialBaseSchema):
trial_name: str | None = Field(default=None, description="试验名称")
trial_type: str | None = Field(default=None, description="试验类型")
design_type: str | None = Field(default=None, description="设计类型")
start_year: int | None = Field(default=None, description="起始年份")
end_year: int | None = Field(default=None, description="结束年份")
objective: str | None = Field(default=None, description="试验目的")
remark: str | None = Field(default=None, description="备注")
class TrialOutSchema(TrialBaseSchema):
id: int
uuid: str
trial_name: str | None = None
plan_id: int | None = None
plan_name: str | None = None
target_id: int | None = None
target_name: str | None = None
trial_type: str | None = None
design_type: str | None = None
start_year: int | None = None
end_year: int | None = None
objective: str | None = None
remark: str | None = None
created_time: datetime | None = None
updated_time: datetime | None = None
created_by: CommonSchema | None = None
updated_by: CommonSchema | None = None
class TrialQueryParam(BaseModel):
trial_name: str | None = Field(default=None, description="试验名称", json_schema_extra={"q": "like"})
trial_type: str | None = Field(default=None, description="试验类型", json_schema_extra={"q": "eq"})
design_type: str | None = Field(default=None, description="设计类型", json_schema_extra={"q": "eq"})
start_year: int | None = Field(default=None, description="起始年份", json_schema_extra={"q": "eq"})
end_year: int | None = Field(default=None, description="结束年份", json_schema_extra={"q": "eq"})
@@ -0,0 +1,266 @@
from typing import Any
from fastapi import UploadFile
from sqlalchemy.ext.asyncio import AsyncSession
from app.core.base_schema import AuthSchema, PageResultSchema, ImportResultSchema
from app.core.exceptions import CustomException
from app.core.logger import logger
from app.utils.common_util import search_to_dict
from app.utils.excel_util import ExcelUtil
from .crud import BreedingTrialCRUD
from .schema import (
TrialCreateSchema,
TrialOutSchema,
TrialQueryParam,
TrialUpdateSchema,
)
from app.core.base_crud import assert_no_children, assert_parents_exist
from app.api.v1.module_bre.trial_study.model import TrialStudyModel
from app.api.v1.module_bre.plan.model import PlanModel
from app.api.v1.module_bre.plan.crud import BreedingPlanCRUD
from app.api.v1.module_bre.target.model import TargetModel
from app.api.v1.module_bre.target.crud import BreedingTargetCRUD
def _is_blank(v: Any) -> bool:
return v is None or (isinstance(v, str) and v.strip() == "")
def _none_if_blank(v: Any) -> Any:
if _is_blank(v):
return None
return str(v).strip() if isinstance(v, str) else v
def _to_float(v: Any) -> float | None:
if _is_blank(v):
return None
try:
return float(v)
except (TypeError, ValueError):
return None
def _to_int(v: Any) -> int | None:
if _is_blank(v):
return None
try:
return int(float(v))
except (TypeError, ValueError):
return None
class TrialService:
"""试验 模块服务层"""
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
self.auth = auth
self.db = db
async def _attach_fk_labels(self, items: list[TrialOutSchema]) -> None:
if not items:
return
crud = BreedingTrialCRUD(self.auth, self.db)
plan_id_ids = {getattr(it, "plan_id") for it in items if getattr(it, "plan_id")}
if plan_id_ids:
refs = await BreedingPlanCRUD(self.auth, self.db).get_list(search={"id": ("in", list(plan_id_ids))})
ref_map = {r.id: getattr(r, "plan_name") for r in refs}
for it in items:
it.plan_name = ref_map.get(getattr(it, "plan_id"))
target_id_ids = {getattr(it, "target_id") for it in items if getattr(it, "target_id")}
if target_id_ids:
refs = await BreedingTargetCRUD(self.auth, self.db).get_list(search={"id": ("in", list(target_id_ids))})
ref_map = {r.id: getattr(r, "target_name") for r in refs}
for it in items:
it.target_name = ref_map.get(getattr(it, "target_id"))
async def detail(self, id: int) -> TrialOutSchema:
obj = await BreedingTrialCRUD(self.auth, self.db).get(id=id)
if not obj:
raise CustomException(msg="该试验不存在")
out = TrialOutSchema.model_validate(obj)
await self._attach_fk_labels([out])
return out
async def get_list(
self,
search: TrialQueryParam | None = None,
order_by: list[dict[str, str]] | None = None,
) -> list[TrialOutSchema]:
obj_list = await BreedingTrialCRUD(self.auth, self.db).get_list(
search=search_to_dict(search), order_by=order_by
)
outs = [TrialOutSchema.model_validate(obj) for obj in obj_list]
await self._attach_fk_labels(outs)
return outs
async def page(
self,
page_no: int,
page_size: int,
search: TrialQueryParam | None = None,
order_by: list[dict[str, str]] | None = None,
) -> PageResultSchema[TrialOutSchema]:
offset = (page_no - 1) * page_size
result = await BreedingTrialCRUD(self.auth, self.db).page(
offset=offset,
limit=page_size,
order_by=order_by or [{"id": "asc"}],
search=search_to_dict(search, {}),
out_schema=TrialOutSchema,
)
await self._attach_fk_labels(result.items)
return result
async def create(self, data: TrialCreateSchema) -> TrialOutSchema:
await assert_parents_exist(
self.db,
[
(PlanModel, data.plan_id, '育种计划'),
(TargetModel, data.target_id, '育种目标'),
],
)
obj = await BreedingTrialCRUD(self.auth, self.db).create(data=data)
out = TrialOutSchema.model_validate(obj)
return out
async def update(self, id: int, data: TrialUpdateSchema) -> TrialOutSchema:
obj = await BreedingTrialCRUD(self.auth, self.db).get(id=id)
if not obj:
raise CustomException(msg="更新失败,该试验不存在")
await assert_parents_exist(
self.db,
[
(PlanModel, data.plan_id, '育种计划'),
(TargetModel, data.target_id, '育种目标'),
],
)
obj = await BreedingTrialCRUD(self.auth, self.db).update(id=id, data=data)
out = TrialOutSchema.model_validate(obj)
return out
async def delete(self, ids: list[int]) -> None:
if not ids:
raise CustomException(msg="删除失败,删除对象不能为空")
objs = await BreedingTrialCRUD(self.auth, self.db).get_list(search={"id": ("in", ids)})
obj_map = {o.id: o for o in objs}
for id_ in ids:
if id_ not in obj_map:
raise CustomException(msg="删除失败,该试验不存在")
await assert_no_children(
self.db,
ids,
[
(TrialStudyModel, 'trial_id', '试验执行'),
],
)
await BreedingTrialCRUD(self.auth, self.db).delete(ids=ids)
async def list_options(self) -> list[dict[str, Any]]:
"""供前端下拉选择使用:返回 [{value, label}]。"""
obj_list = await BreedingTrialCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
return [{"value": o.id, "label": o.trial_name} for o in obj_list]
@staticmethod
def batch_export(obj_list: list[dict[str, Any]]) -> bytes:
mapping_dict = {
"trial_name": "试验名称",
"trial_type": "试验类型",
"design_type": "设计类型",
"start_year": "起始年份",
"end_year": "结束年份",
"objective": "试验目的",
"remark": "备注",
"created_time": "创建时间",
"created_by": "创建者",
}
data = [dict(item) for item in obj_list]
for item in data:
creator = item.get("created_by")
item["created_by"] = creator.get("name", "未知") if isinstance(creator, dict) else "未知"
return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)
async def batch_import(self, file: UploadFile, update_support: bool = False) -> ImportResultSchema:
header_dict = {
"试验名称": "trial_name",
"试验类型": "trial_type",
"设计类型": "design_type",
"起始年份": "start_year",
"结束年份": "end_year",
"试验目的": "objective",
"备注": "remark",
}
try:
contents = await file.read()
rows = ExcelUtil.read_excel_to_dicts(contents)
await file.close()
if not rows:
raise CustomException(msg="导入文件为空")
missing_headers = [h for h in header_dict if h not in rows[0]]
if missing_headers:
raise CustomException(msg=f"导入文件缺少必要的列: {', '.join(missing_headers)}")
mapped_rows = []
for row in rows:
mapped_rows.append({en: row.get(ch) for ch, en in header_dict.items()})
required_fields = ["trial_name"]
errors = []
for field in required_fields:
missing_indices = [i + 1 for i, r in enumerate(mapped_rows) if _is_blank(r.get(field))]
if missing_indices:
field_name = next(k for k, v in header_dict.items() if v == field)
rows_str = "".join(str(i) for i in missing_indices)
errors.append(f"{field_name}不能为空,第{rows_str}")
if errors:
raise CustomException(msg=f"导入失败,以下行缺少必要字段:\n{'; '.join(errors)}")
error_msgs: list[str] = []
success_count = 0
crud = BreedingTrialCRUD(self.auth, self.db)
for i, row in enumerate(mapped_rows, start=1):
try:
fields = {
"trial_name": _none_if_blank(row.get("trial_name")),
"trial_type": _none_if_blank(row.get("trial_type")),
"design_type": _none_if_blank(row.get("design_type")),
"start_year": _to_int(row.get("start_year")),
"end_year": _to_int(row.get("end_year")),
"objective": _none_if_blank(row.get("objective")),
"remark": _none_if_blank(row.get("remark")),
}
create_data = TrialCreateSchema(**fields)
await crud.create(data=create_data)
success_count += 1
except Exception as e:
error_msgs.append(f"{i}行: {e!s}")
continue
return ImportResultSchema(
valid_count=success_count,
invalid_count=len(error_msgs),
message_list=error_msgs,
)
except Exception as e:
logger.error(f"批量导入试验失败: {e!s}")
raise CustomException(msg=f"导入失败: {e!s}")
@staticmethod
def import_template_download() -> bytes:
header_list = [
"试验名称",
"试验类型",
"设计类型",
"起始年份",
"结束年份",
"试验目的",
"备注",
]
selector_header_list = []
option_list = [
]
return ExcelUtil.get_excel_template(
header_list=header_list,
selector_header_list=selector_header_list,
option_list=option_list,
)