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

backend/app/crud/finance_special.py

@@ -0,0 +1,69 @@
+import uuid
+from typing import Sequence
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.finance_special import FinanceSpecial
+from app.schemas.finance_special import FinanceSpecialCreate, FinanceSpecialUpdate
+
+
+async def create_special(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    special_in: FinanceSpecialCreate,
+    created_by: uuid.UUID | None,
+) -> FinanceSpecial:
+    special = FinanceSpecial(
+        study_id=study_id,
+        site_name=special_in.site_name,
+        fee_type=special_in.fee_type,
+        amount=special_in.amount,
+        occur_date=special_in.occur_date,
+        staff_name=special_in.staff_name,
+        remark=special_in.remark,
+        created_by=created_by,
+    )
+    db.add(special)
+    await db.commit()
+    await db.refresh(special)
+    return special
+
+
+async def get_special(db: AsyncSession, special_id: uuid.UUID) -> FinanceSpecial | None:
+    result = await db.execute(select(FinanceSpecial).where(FinanceSpecial.id == special_id))
+    return result.scalar_one_or_none()
+
+
+async def list_specials(
+    db: AsyncSession,
+    study_id: uuid.UUID,
+    site_name: str | None = None,
+    fee_type: str | None = None,
+    skip: int = 0,
+    limit: int = 100,
+) -> Sequence[FinanceSpecial]:
+    stmt = select(FinanceSpecial).where(FinanceSpecial.study_id == study_id)
+    if site_name:
+        stmt = stmt.where(FinanceSpecial.site_name.ilike(f"%{site_name}%"))
+    if fee_type:
+        stmt = stmt.where(FinanceSpecial.fee_type == fee_type)
+    stmt = stmt.order_by(FinanceSpecial.created_at.desc()).offset(skip).limit(limit)
+    result = await db.execute(stmt)
+    return result.scalars().all()
+
+
+async def update_special(
+    db: AsyncSession, special: FinanceSpecial, special_in: FinanceSpecialUpdate
+) -> FinanceSpecial:
+    update_data = special_in.model_dump(exclude_unset=True)
+    for key, value in update_data.items():
+        setattr(special, key, value)
+    await db.commit()
+    await db.refresh(special)
+    return special
+
+
+async def delete_special(db: AsyncSession, special: FinanceSpecial) -> None:
+    await db.delete(special)
+    await db.commit()