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 (
PedigreeCreateSchema,
PedigreeOutSchema,
PedigreeQueryParam,
PedigreeUpdateSchema,
)
from .service import PedigreeService
PedigreeRouter = APIRouter(route_class=OperationLogRoute, prefix="/pedigree", tags=["系谱管理"])
@PedigreeRouter.get("/detail/{id}", summary="获取系谱管理详情", response_model=ResponseSchema[PedigreeOutSchema])
async def get_pedigree__detail_controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:detail"]))],
id: Annotated[int, Path(description="系谱管理ID")],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(auth, db)
result_dict = await service.detail(id=id)
return SuccessResponse(data=result_dict, msg="获取系谱详情成功")
@PedigreeRouter.get("/list", summary="分页查询系谱管理", response_model=ResponseSchema[PageResultSchema[PedigreeOutSchema]])
async def get_pedigree__list_controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:query"]))],
page: Annotated[PaginationQueryParam, Depends()],
search: Annotated[PedigreeQueryParam, Query()],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(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="查询系谱列表成功")
@PedigreeRouter.get("/options", summary="系谱管理下拉选项")
async def get_pedigree__options_controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:query"]))],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(auth, db)
options = await service.list_options()
return SuccessResponse(data=options, msg="获取系谱选项成功")
@PedigreeRouter.post("/create", summary="创建系谱管理", response_model=ResponseSchema[PedigreeOutSchema])
async def create_pedigree__controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:create"]))],
data: Annotated[PedigreeCreateSchema, Body(description="创建参数")],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(auth, db)
result_dict = await service.create(data=data)
return SuccessResponse(data=result_dict, msg="创建系谱成功")
@PedigreeRouter.put("/update/{id}", summary="修改系谱管理", response_model=ResponseSchema[PedigreeOutSchema])
async def update_pedigree__controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:update"]))],
id: Annotated[int, Path(description="系谱管理ID")],
data: Annotated[PedigreeUpdateSchema, Body(description="修改参数")],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(auth, db)
result_dict = await service.update(id=id, data=data)
return SuccessResponse(data=result_dict, msg="修改系谱成功")
@PedigreeRouter.delete("/delete", summary="删除系谱管理", response_model=ResponseSchema[None])
async def delete_pedigree__controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:delete"]))],
ids: Annotated[list[int], Body(description="ID列表")],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(auth, db)
await service.delete(ids=ids)
return SuccessResponse(msg="删除系谱成功")
@PedigreeRouter.post("/export", summary="导出系谱管理")
async def export_pedigree__controller(
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:export"]))],
search: Annotated[PedigreeQueryParam, Query()],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
service = PedigreeService(auth, db)
result_dict_list = await service.get_list(search=search)
export_result = PedigreeService.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')}"},
)
@PedigreeRouter.post("/import", summary="导入系谱管理", response_model=ResponseSchema[ImportResultSchema])
async def import_pedigree__controller(
file: Annotated[UploadFile, File(description="导入文件")],
auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:pedigree:import"]))],
db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
service = PedigreeService(auth, db)
batch_import_result = await service.batch_import(file=file, update_support=True)
return SuccessResponse(data=batch_import_result, msg="导入系谱成功")
@PedigreeRouter.post("/download/template", summary="获取系谱管理导入模板", dependencies=[Depends(AuthPermission(["module_bre:pedigree:download"]))])
async def download_pedigree__template_controller() -> StreamingResponse:
import_template_result = PedigreeService.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 PedigreeModel
class BreedingPedigreeCRUD(CRUDBase[PedigreeModel, Any, Any]):
"""系谱管理 CRUD —— 直接复用 CRUDBase(已自动注入数据权限过滤)。"""
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
super().__init__(PedigreeModel, auth, db)
pedigree_crud = BreedingPedigreeCRUD
@@ -0,0 +1,40 @@
"""系谱管理 数据模型"""
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 PedigreeModel(ModelMixin, UserMixin, MappedBase):
"""系谱管理 主数据表。"""
__tablename__ = "bre_pedigree"
combination_id: Mapped[int] = mapped_column(
Integer, ForeignKey("bre_cross_combination.id", ondelete="CASCADE"),
index=True, nullable=False, comment="来源杂交组合"
)
child_code: Mapped[str] = mapped_column(String(64), nullable=False, comment="子代种质编号")
dam_id: Mapped[int | None] = mapped_column(
Integer, ForeignKey("bre_germplasm.id", ondelete="SET NULL"),
index=True, nullable=True, comment="母本(种质)"
)
sire_id: Mapped[int | None] = mapped_column(
Integer, ForeignKey("bre_germplasm.id", ondelete="SET NULL"),
index=True, nullable=True, comment="父本(种质)"
)
generation: Mapped[str | None] = mapped_column(String(16), 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_pedigree_created_deleted", "created_time", "is_deleted"),
)
@@ -0,0 +1,53 @@
from app.core.base_schema import CommonSchema
"""系谱管理 —— Pydantic 校验/序列化模型。"""
from datetime import datetime
from pydantic import BaseModel, ConfigDict, Field
class PedigreeBaseSchema(BaseModel):
model_config = ConfigDict(from_attributes=True)
combination_id: int = Field(..., description="来源杂交组合")
child_code: str = Field(..., description="子代种质编号")
dam_id: int | None = Field(default=None, description="母本种质(规格 §3.18)")
sire_id: int | None = Field(default=None, description="父本种质(规格 §3.18)")
generation: str | None = Field(default=None, description="世代")
remark: str | None = Field(default=None, description="备注")
class PedigreeCreateSchema(PedigreeBaseSchema):
pass
class PedigreeUpdateSchema(PedigreeBaseSchema):
combination_id: int | None = Field(default=None, description="来源杂交组合")
child_code: str | None = Field(default=None, description="子代种质编号")
dam_id: int | None = Field(default=None, description="母本种质(规格 §3.18)")
sire_id: int | None = Field(default=None, description="父本种质(规格 §3.18)")
generation: str | None = Field(default=None, description="世代")
remark: str | None = Field(default=None, description="备注")
class PedigreeOutSchema(PedigreeBaseSchema):
id: int
uuid: str
combination_name: str | None = None # 由 Service 层联表填充
dam_name: str | None = None # 母本名称(联表填充)
sire_name: str | None = None # 父本名称(联表填充)
child_code: str | None = None
generation: 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 PedigreeQueryParam(BaseModel):
combination_id: int | None = Field(default=None, description="来源杂交组合", json_schema_extra={"q": "eq"})
child_code: str | None = Field(default=None, description="子代种质编号", json_schema_extra={"q": "like"})
dam_id: int | None = Field(default=None, description="母本种质", json_schema_extra={"q": "eq"})
sire_id: int | None = Field(default=None, description="父本种质", json_schema_extra={"q": "eq"})
generation: str | None = Field(default=None, description="世代", json_schema_extra={"q": "eq"})
@@ -0,0 +1,255 @@
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 BreedingPedigreeCRUD
from .schema import (
PedigreeCreateSchema,
PedigreeOutSchema,
PedigreeQueryParam,
PedigreeUpdateSchema,
)
from app.api.v1.module_bre.cross_combination.crud import BreedingCrossCombinationCRUD
from app.api.v1.module_bre.germplasm.model import BreedingGermplasmModel
from app.api.v1.module_bre.germplasm.crud import BreedingGermplasmCRUD
from app.core.base_crud import assert_no_children, assert_parents_exist
from app.api.v1.module_bre.cross_combination.model import CrossCombinationModel
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 PedigreeService:
"""系谱管理 模块服务层"""
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
self.auth = auth
self.db = db
async def _attach_fk_labels(self, items: list[PedigreeOutSchema]) -> None:
if not items:
return
crud = BreedingPedigreeCRUD(self.auth, self.db)
combination_id_ids = {getattr(it, "combination_id") for it in items if getattr(it, "combination_id")}
if combination_id_ids:
refs = await BreedingCrossCombinationCRUD(self.auth, self.db).get_list(search={"id": ("in", list(combination_id_ids))})
ref_map = {r.id: getattr(r, "combination_code") for r in refs}
for it in items:
it.combination_name = ref_map.get(getattr(it, "combination_id"))
germ_ids = {
g for it in items
for g in (getattr(it, "dam_id"), getattr(it, "sire_id"))
if g
}
if germ_ids:
germ_map = {
r.id: getattr(r, "cultivar_name")
for r in await BreedingGermplasmCRUD(self.auth, self.db).get_list(search={"id": ("in", list(germ_ids))})
}
for it in items:
it.dam_name = germ_map.get(getattr(it, "dam_id"))
it.sire_name = germ_map.get(getattr(it, "sire_id"))
async def detail(self, id: int) -> PedigreeOutSchema:
obj = await BreedingPedigreeCRUD(self.auth, self.db).get(id=id)
if not obj:
raise CustomException(msg="该系谱不存在")
out = PedigreeOutSchema.model_validate(obj)
await self._attach_fk_labels([out])
return out
async def get_list(
self,
search: PedigreeQueryParam | None = None,
order_by: list[dict[str, str]] | None = None,
) -> list[PedigreeOutSchema]:
obj_list = await BreedingPedigreeCRUD(self.auth, self.db).get_list(
search=search_to_dict(search), order_by=order_by
)
outs = [PedigreeOutSchema.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: PedigreeQueryParam | None = None,
order_by: list[dict[str, str]] | None = None,
) -> PageResultSchema[PedigreeOutSchema]:
offset = (page_no - 1) * page_size
result = await BreedingPedigreeCRUD(self.auth, self.db).page(
offset=offset,
limit=page_size,
order_by=order_by or [{"id": "asc"}],
search=search_to_dict(search, {}),
out_schema=PedigreeOutSchema,
)
await self._attach_fk_labels(result.items)
return result
async def _check_parents(self, data: PedigreeCreateSchema | PedigreeUpdateSchema) -> None:
checks: list[tuple[Any, Any, str]] = [
(CrossCombinationModel, data.combination_id, '杂交组合'),
]
if getattr(data, "dam_id", None):
checks.append((BreedingGermplasmModel, data.dam_id, '母本种质'))
if getattr(data, "sire_id", None):
checks.append((BreedingGermplasmModel, data.sire_id, '父本种质'))
await assert_parents_exist(self.db, checks)
async def create(self, data: PedigreeCreateSchema) -> PedigreeOutSchema:
await self._check_parents(data)
obj = await BreedingPedigreeCRUD(self.auth, self.db).create(data=data)
out = PedigreeOutSchema.model_validate(obj)
await self._attach_fk_labels([out])
return out
async def update(self, id: int, data: PedigreeUpdateSchema) -> PedigreeOutSchema:
obj = await BreedingPedigreeCRUD(self.auth, self.db).get(id=id)
if not obj:
raise CustomException(msg="更新失败,该系谱不存在")
await self._check_parents(data)
obj = await BreedingPedigreeCRUD(self.auth, self.db).update(id=id, data=data)
out = PedigreeOutSchema.model_validate(obj)
await self._attach_fk_labels([out])
return out
async def delete(self, ids: list[int]) -> None:
if not ids:
raise CustomException(msg="删除失败,删除对象不能为空")
objs = await BreedingPedigreeCRUD(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 BreedingPedigreeCRUD(self.auth, self.db).delete(ids=ids)
async def list_options(self) -> list[dict[str, Any]]:
"""供前端下拉选择使用:返回 [{value, label}]。"""
obj_list = await BreedingPedigreeCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
return [{"value": o.id, "label": o.child_code} for o in obj_list]
@staticmethod
def batch_export(obj_list: list[dict[str, Any]]) -> bytes:
mapping_dict = {
"combination_name": "来源杂交组合",
"child_code": "子代种质编号",
"generation": "世代",
"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 = {
"来源杂交组合": "combination_id",
"子代种质编号": "child_code",
"世代": "generation",
"备注": "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)}")
combination_id_refs = await BreedingCrossCombinationCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
combination_id_map = {getattr(r, "combination_code"): r.id for r in combination_id_refs}
mapped_rows = []
for row in rows:
mapped_rows.append({en: row.get(ch) for ch, en in header_dict.items()})
required_fields = ["combination_id", "child_code"]
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 = BreedingPedigreeCRUD(self.auth, self.db)
for i, row in enumerate(mapped_rows, start=1):
try:
combination_id_val = combination_id_map.get(str(row.get("combination_id")).strip()) if not _is_blank(row.get("combination_id")) else None
fields = {
"combination_id": combination_id_val,
"child_code": _none_if_blank(row.get("child_code")),
"generation": _none_if_blank(row.get("generation")),
"remark": _none_if_blank(row.get("remark")),
}
create_data = PedigreeCreateSchema(**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,
)