feat: 初始化工程

service/patch_service.py

@@ -0,0 +1,79 @@
+from sqlalchemy import select, delete
+from sqlalchemy.orm import Session, selectinload
+
+from models.patch import Patches, Crops, Records
+from schemas.patch import PatchCreate, PatchUpdate, PatchResponse, CropResponse
+
+
+def create_patch(db: Session, data: PatchCreate) -> bool:
+    patch = Patches(**data.model_dump())
+    db.add(patch)
+    db.commit()
+    db.refresh(patch)
+
+    return True
+
+
+def get_patches(db: Session) -> list[PatchResponse]:
+    patches = db.execute(select(Patches)).scalars().all()
+    result = []
+    for p in patches:
+        crop = db.execute(
+            select(Crops).where(Crops.patch_id == p.id)
+        ).scalar_one_or_none()
+        data = PatchResponse.model_validate(p)
+        data.crop = CropResponse.from_orm(crop) if crop else None
+        result.append(data)
+
+    return result
+
+
+def get_patch(db: Session, patch_id: int) -> PatchResponse | None:
+    patch = db.execute(
+        select(Patches).where(Patches.id == patch_id)
+    ).scalar_one_or_none()
+    if not patch:
+        return None
+    crop = db.execute(
+        select(Crops).where(Crops.patch_id == patch_id)
+    ).scalar_one_or_none()
+    data = PatchResponse.model_validate(patch)
+    data.crop = CropResponse.from_orm(crop) if crop else None
+
+    return data
+
+
+def update_patch(db: Session, patch_id: int, data: PatchUpdate) -> bool:
+    result = db.execute(select(Patches).where(Patches.id == patch_id))
+    patch = result.scalar_one_or_none()
+    if not patch:
+        return False
+
+    for k, v in data.model_dump(exclude_unset=True).items():
+        setattr(patch, k, v)
+    db.commit()
+    db.refresh(patch)
+
+    return True
+
+
+def delete_patch(db: Session, patch_id: int) -> bool:
+    result = db.execute(select(Patches).where(Patches.id == patch_id))
+    patch = result.scalar_one_or_none()
+    if not patch:
+        return False
+
+    # 先查作物
+    crop_result = db.execute(select(Crops).where(Crops.patch_id == patch_id))
+    crop = crop_result.scalar_one_or_none()
+
+    # 有作物就先删记录，再删作物
+    if crop:
+        db.execute(delete(Records).where(Records.crop_id == crop.id))
+        db.delete(crop)
+
+    # 最后删菜地
+    db.delete(patch)
+    db.commit()
+
+    return True