316 lines
14 KiB
Python
316 lines
14 KiB
Python
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.dict_util import DictLabelResolver
|
|
from app.utils.excel_util import ExcelUtil
|
|
|
|
from .crud import BreedingFieldOperationCRUD
|
|
from .schema import (
|
|
FieldOperationCreateSchema,
|
|
FieldOperationOutSchema,
|
|
FieldOperationQueryParam,
|
|
FieldOperationUpdateSchema,
|
|
)
|
|
from app.core.base_crud import assert_parents_exist
|
|
from app.api.v1.module_bre.tree.crud import BreedingTreeCRUD
|
|
from app.api.v1.module_bre.tree.model import TreeModel
|
|
from app.api.v1.module_bre.site.crud import BreedingPlotCRUD, BreedingSiteCRUD
|
|
from app.api.v1.module_bre.site.model import BreedingPlotModel
|
|
from app.api.v1.module_bre.personnel.crud import BreedingPersonnelCRUD
|
|
from app.api.v1.module_bre.personnel.model import PersonnelModel
|
|
from app.api.v1.module_bre.trait.crud import BreedingTraitCRUD
|
|
from app.api.v1.module_bre.trait_observation.crud import BreedingTraitObservationCRUD
|
|
from app.api.v1.module_bre.trait_observation.schema import (
|
|
TraitObservationCreateSchema,
|
|
TraitObservationUpdateSchema,
|
|
)
|
|
|
|
|
|
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_int(v: Any) -> int | None:
|
|
if _is_blank(v):
|
|
return None
|
|
try:
|
|
return int(float(v))
|
|
except (TypeError, ValueError):
|
|
return None
|
|
|
|
|
|
class FieldOperationService:
|
|
"""农事操作 模块服务层"""
|
|
|
|
# 采收产量构成 → 统计性状编码映射（bre_yield_phenology_traits.sql 种子）
|
|
_HARVEST_TRAIT_CODES = {
|
|
"yield_kg": "single_tree_yield",
|
|
"fruit_count": "fruit_number",
|
|
"avg_fruit_weight": "avg_fruit_weight",
|
|
"marketable_rate": "marketable_rate",
|
|
}
|
|
|
|
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
|
|
self.auth = auth
|
|
self.db = db
|
|
|
|
async def _sync_harvest_to_trait_observation(self, obj: FieldOperationOutSchema) -> None:
|
|
"""采收(op_type=harvest)结构化产量 → 同步写统计性状观测(bre_trait_observation)。
|
|
|
|
仅单株(tree_id)级采收同步（树级统计入口）；小区级采收只留 field_operation 记录。
|
|
以 (tree_id, trait_id, evaluate_year) 幂等 upsert：同年重复采收覆盖值。
|
|
"""
|
|
if not obj.op_type or obj.op_type != "harvest" or not obj.tree_id:
|
|
return
|
|
values = [
|
|
(code, getattr(obj, field))
|
|
for field, code in self._HARVEST_TRAIT_CODES.items()
|
|
if getattr(obj, field) is not None
|
|
]
|
|
if not values:
|
|
return
|
|
year = obj.op_date.year if obj.op_date else None
|
|
to_crud = BreedingTraitObservationCRUD(self.auth, self.db)
|
|
for code, value in values:
|
|
traits = await BreedingTraitCRUD(self.auth, self.db).get_list(search={"trait_code": ("eq", code)})
|
|
if not traits:
|
|
continue
|
|
trait = traits[0]
|
|
existing = await to_crud.get(tree_id=obj.tree_id, trait_id=trait.id, evaluate_year=year)
|
|
payload: dict[str, Any] = {"value_numeric": float(value), "stage": "evaluation"}
|
|
if year is not None:
|
|
payload["evaluate_year"] = year
|
|
if existing:
|
|
await to_crud.update(id=existing.id, data=TraitObservationUpdateSchema(**payload))
|
|
else:
|
|
create = {"tree_id": obj.tree_id, "trait_id": trait.id}
|
|
create.update(payload)
|
|
await to_crud.create(data=TraitObservationCreateSchema(**create))
|
|
|
|
async def _attach_fk_labels(self, items: list[FieldOperationOutSchema]) -> None:
|
|
if not items:
|
|
return
|
|
tree_ids = {getattr(it, "tree_id") for it in items if getattr(it, "tree_id")}
|
|
if tree_ids:
|
|
refs = await BreedingTreeCRUD(self.auth, self.db).get_list(search={"id": ("in", list(tree_ids))})
|
|
ref_map = {r.id: getattr(r, "tree_no") for r in refs}
|
|
for it in items:
|
|
it.tree_no = ref_map.get(getattr(it, "tree_id"))
|
|
plot_ids = {getattr(it, "plot_id") for it in items if getattr(it, "plot_id")}
|
|
if plot_ids:
|
|
plot_refs = await BreedingPlotCRUD(self.auth, self.db).get_list(search={"id": ("in", list(plot_ids))})
|
|
plot_map = {r.id: r for r in plot_refs}
|
|
site_ids = {getattr(r, "site_id") for r in plot_map.values()}
|
|
site_refs = await BreedingSiteCRUD(self.auth, self.db).get_list(search={"id": ("in", list(site_ids))})
|
|
site_map = {r.id: getattr(r, "site_name") for r in site_refs}
|
|
for it in items:
|
|
plot = plot_map.get(getattr(it, "plot_id"))
|
|
if plot:
|
|
it.plot_label = f"{site_map.get(plot.site_id) or ''}-{plot.plot_code}"
|
|
operator_ids = {getattr(it, "operator_id") for it in items if getattr(it, "operator_id")}
|
|
if operator_ids:
|
|
refs = await BreedingPersonnelCRUD(self.auth, self.db).get_list(search={"id": ("in", list(operator_ids))})
|
|
ref_map = {r.id: getattr(r, "name") for r in refs}
|
|
for it in items:
|
|
it.operator_name = ref_map.get(getattr(it, "operator_id"))
|
|
|
|
async def detail(self, id: int) -> FieldOperationOutSchema:
|
|
obj = await BreedingFieldOperationCRUD(self.auth, self.db).get(id=id)
|
|
if not obj:
|
|
raise CustomException(msg="该农事操作不存在")
|
|
out = FieldOperationOutSchema.model_validate(obj)
|
|
await self._attach_fk_labels([out])
|
|
return out
|
|
|
|
async def get_list(
|
|
self,
|
|
search: FieldOperationQueryParam | None = None,
|
|
order_by: list[dict[str, str]] | None = None,
|
|
) -> list[FieldOperationOutSchema]:
|
|
obj_list = await BreedingFieldOperationCRUD(self.auth, self.db).get_list(
|
|
search=search_to_dict(search), order_by=order_by
|
|
)
|
|
outs = [FieldOperationOutSchema.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: FieldOperationQueryParam | None = None,
|
|
order_by: list[dict[str, str]] | None = None,
|
|
) -> PageResultSchema[FieldOperationOutSchema]:
|
|
offset = (page_no - 1) * page_size
|
|
result = await BreedingFieldOperationCRUD(self.auth, self.db).page(
|
|
offset=offset,
|
|
limit=page_size,
|
|
order_by=order_by or [{"id": "asc"}],
|
|
search=search_to_dict(search, {}),
|
|
out_schema=FieldOperationOutSchema,
|
|
)
|
|
await self._attach_fk_labels(result.items)
|
|
return result
|
|
|
|
async def create(self, data: FieldOperationCreateSchema) -> FieldOperationOutSchema:
|
|
if _is_blank(data.op_type):
|
|
raise CustomException(msg="操作类型不能为空")
|
|
await assert_parents_exist(
|
|
self.db,
|
|
[
|
|
(TreeModel, data.tree_id, '单株'),
|
|
(BreedingPlotModel, data.plot_id, '小区'),
|
|
(PersonnelModel, data.operator_id, '育种人员'),
|
|
],
|
|
)
|
|
obj = await BreedingFieldOperationCRUD(self.auth, self.db).create(data=data)
|
|
out = FieldOperationOutSchema.model_validate(obj)
|
|
await self._sync_harvest_to_trait_observation(out)
|
|
await self._attach_fk_labels([out])
|
|
return out
|
|
|
|
async def update(self, id: int, data: FieldOperationUpdateSchema) -> FieldOperationOutSchema:
|
|
obj = await BreedingFieldOperationCRUD(self.auth, self.db).get(id=id)
|
|
if not obj:
|
|
raise CustomException(msg="更新失败，该农事操作不存在")
|
|
await assert_parents_exist(
|
|
self.db,
|
|
[
|
|
(TreeModel, data.tree_id, '单株'),
|
|
(BreedingPlotModel, data.plot_id, '小区'),
|
|
(PersonnelModel, data.operator_id, '育种人员'),
|
|
],
|
|
)
|
|
obj = await BreedingFieldOperationCRUD(self.auth, self.db).update(id=id, data=data)
|
|
out = FieldOperationOutSchema.model_validate(obj)
|
|
await self._sync_harvest_to_trait_observation(out)
|
|
await self._attach_fk_labels([out])
|
|
return out
|
|
|
|
async def delete(self, ids: list[int]) -> None:
|
|
if not ids:
|
|
raise CustomException(msg="删除失败，删除对象不能为空")
|
|
objs = await BreedingFieldOperationCRUD(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 BreedingFieldOperationCRUD(self.auth, self.db).delete(ids=ids)
|
|
|
|
async def list_options(self) -> list[dict[str, Any]]:
|
|
"""供前端下拉选择使用：返回 [{value, label}]。"""
|
|
obj_list = await BreedingFieldOperationCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
|
|
return [{"value": o.id, "label": o.op_type} for o in obj_list]
|
|
|
|
@staticmethod
|
|
def batch_export(obj_list: list[dict[str, Any]]) -> bytes:
|
|
mapping_dict = {
|
|
"tree_no": "单株",
|
|
"plot_label": "小区",
|
|
"op_type": "操作类型",
|
|
"op_date": "操作日期",
|
|
"op_detail": "操作详情",
|
|
"operator_name": "操作人",
|
|
"remark": "备注",
|
|
"status": "状态",
|
|
"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 = {
|
|
"单株": "tree_id",
|
|
"小区": "plot_id",
|
|
"操作类型": "op_type",
|
|
"操作日期": "op_date",
|
|
"操作详情": "op_detail",
|
|
"操作人": "operator_id",
|
|
"备注": "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)}")
|
|
tree_refs = await BreedingTreeCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
|
|
tree_map = {getattr(r, "tree_no"): r.id for r in tree_refs}
|
|
plot_refs = await BreedingPlotCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
|
|
site_ids = {getattr(r, "site_id") for r in plot_refs}
|
|
site_refs = await BreedingSiteCRUD(self.auth, self.db).get_list(search={"id": ("in", list(site_ids))}) if site_ids else []
|
|
site_map = {r.id: getattr(r, "site_name") for r in site_refs}
|
|
plot_map = {f"{site_map.get(r.site_id) or ''}-{r.plot_code}": r.id for r in plot_refs}
|
|
operator_refs = await BreedingPersonnelCRUD(self.auth, self.db).get_list(order_by=[{"id": "asc"}])
|
|
operator_map = {getattr(r, "name"): r.id for r in operator_refs}
|
|
error_msgs: list[str] = []
|
|
success_count = 0
|
|
crud = BreedingFieldOperationCRUD(self.auth, self.db)
|
|
resolver = DictLabelResolver(self.auth, self.db, ["field_operation"])
|
|
for i, row in enumerate(rows, start=1):
|
|
try:
|
|
fields = {
|
|
"tree_id": tree_map.get(str(row.get("tree_id")).strip())
|
|
if not _is_blank(row.get("tree_id")) else None,
|
|
"plot_id": plot_map.get(str(row.get("plot_id")).strip())
|
|
if not _is_blank(row.get("plot_id")) else None,
|
|
"op_type": await resolver.resolve("field_operation", row.get("op_type")),
|
|
"op_date": _none_if_blank(row.get("op_date")),
|
|
"op_detail": _none_if_blank(row.get("op_detail")),
|
|
"operator_id": operator_map.get(str(row.get("operator_id")).strip())
|
|
if not _is_blank(row.get("operator_id")) else None,
|
|
"remark": _none_if_blank(row.get("remark")),
|
|
}
|
|
create_data = FieldOperationCreateSchema(**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,
|
|
)
|