Cheng Zhou
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
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()
|