信息架构/菜单框架大重构-20260109

backend/app/crud/drug_shipment.py

@@ -0,0 +1,76 @@
+import uuid
+from typing import Sequence
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.drug_shipment import DrugShipment
+from app.schemas.drug_shipment import DrugShipmentCreate, DrugShipmentUpdate
+
+
+async def create_shipment(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    shipment_in: DrugShipmentCreate,
+    created_by: uuid.UUID | None,
+) -> DrugShipment:
+    shipment = DrugShipment(
+        study_id=study_id,
+        direction=shipment_in.direction,
+        site_name=shipment_in.site_name,
+        drug_desc=shipment_in.drug_desc,
+        ship_date=shipment_in.ship_date,
+        carrier=shipment_in.carrier,
+        tracking_no=shipment_in.tracking_no,
+        from_party=shipment_in.from_party,
+        to_party=shipment_in.to_party,
+        status=shipment_in.status,
+        remark=shipment_in.remark,
+        created_by=created_by,
+    )
+    db.add(shipment)
+    await db.commit()
+    await db.refresh(shipment)
+    return shipment
+
+
+async def get_shipment(db: AsyncSession, shipment_id: uuid.UUID) -> DrugShipment | None:
+    result = await db.execute(select(DrugShipment).where(DrugShipment.id == shipment_id))
+    return result.scalar_one_or_none()
+
+
+async def list_shipments(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    site_name: str | None = None,
+    tracking_no: str | None = None,
+    status: str | None = None,
+    skip: int = 0,
+    limit: int = 100,
+) -> Sequence[DrugShipment]:
+    stmt = select(DrugShipment).where(DrugShipment.study_id == study_id)
+    if site_name:
+        stmt = stmt.where(DrugShipment.site_name.ilike(f"%{site_name}%"))
+    if tracking_no:
+        stmt = stmt.where(DrugShipment.tracking_no.ilike(f"%{tracking_no}%"))
+    if status:
+        stmt = stmt.where(DrugShipment.status == status)
+    stmt = stmt.order_by(DrugShipment.created_at.desc()).offset(skip).limit(limit)
+    result = await db.execute(stmt)
+    return result.scalars().all()
+
+
+async def update_shipment(
+    db: AsyncSession, shipment: DrugShipment, shipment_in: DrugShipmentUpdate
+) -> DrugShipment:
+    update_data = shipment_in.model_dump(exclude_unset=True)
+    for key, value in update_data.items():
+        setattr(shipment, key, value)
+    await db.commit()
+    await db.refresh(shipment)
+    return shipment
+
+
+async def delete_shipment(db: AsyncSession, shipment: DrugShipment) -> None:
+    await db.delete(shipment)
+    await db.commit()