Step 5：里程碑与任务（伦理/启动会进度）

2025-12-16 17:18:26 +08:00
parent 6c4a2eb1dc
commit a9e15faedc
72 changed files with 535 additions and 1 deletions
@@ -0,0 +1,48 @@
+import uuid
+from typing import Sequence
+
+from sqlalchemy import select, update as sa_update
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.milestone import Milestone
+from app.schemas.milestone import MilestoneCreate, MilestoneUpdate
+
+
+async def create(db: AsyncSession, study_id: uuid.UUID, milestone_in: MilestoneCreate) -> Milestone:
+    milestone = Milestone(
+        study_id=study_id,
+        type=milestone_in.type,
+        name=milestone_in.name or milestone_in.type,
+        planned_date=milestone_in.planned_date,
+        actual_date=None,
+        status=milestone_in.status or "NOT_STARTED",
+        owner_id=milestone_in.owner_id,
+        notes=milestone_in.notes,
+    )
+    db.add(milestone)
+    await db.commit()
+    await db.refresh(milestone)
+    return milestone
+
+
+async def get(db: AsyncSession, milestone_id: uuid.UUID) -> Milestone | None:
+    result = await db.execute(select(Milestone).where(Milestone.id == milestone_id))
+    return result.scalar_one_or_none()
+
+
+async def list_milestones(db: AsyncSession, study_id: uuid.UUID) -> Sequence[Milestone]:
+    result = await db.execute(select(Milestone).where(Milestone.study_id == study_id).order_by(Milestone.planned_date))
+    return result.scalars().all()
+
+
+async def update(db: AsyncSession, milestone: Milestone, milestone_in: MilestoneUpdate) -> Milestone:
+    update_data = milestone_in.model_dump(exclude_unset=True)
+    if update_data:
+        await db.execute(
+            sa_update(Milestone)
+            .where(Milestone.id == milestone.id)
+            .values(**update_data)
+        )
+        await db.commit()
+    await db.refresh(milestone)
+    return milestone