费用管理内容优化-初步

backend/app/crud/fee_attachment.py

@@ -0,0 +1,70 @@
+import uuid
+from typing import Sequence
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.fee_attachment import FeeAttachment
+
+
+async def create_attachment(
+    db: AsyncSession,
+    *,
+    entity_type: str,
+    entity_id: uuid.UUID,
+    file_type: str,
+    filename: str,
+    mime_type: str | None,
+    size: int,
+    storage_key: str | None,
+    url: str | None,
+    uploaded_by: uuid.UUID,
+) -> FeeAttachment:
+    attachment = FeeAttachment(
+        entity_type=entity_type,
+        entity_id=entity_id,
+        file_type=file_type,
+        filename=filename,
+        mime_type=mime_type,
+        size=size,
+        storage_key=storage_key,
+        url=url,
+        uploaded_by=uploaded_by,
+    )
+    db.add(attachment)
+    await db.commit()
+    await db.refresh(attachment)
+    return attachment
+
+
+async def list_attachments(
+    db: AsyncSession,
+    *,
+    entity_type: str,
+    entity_id: uuid.UUID,
+) -> Sequence[FeeAttachment]:
+    result = await db.execute(
+        select(FeeAttachment)
+        .where(
+            FeeAttachment.entity_type == entity_type,
+            FeeAttachment.entity_id == entity_id,
+            FeeAttachment.is_deleted.is_(False),
+        )
+        .order_by(FeeAttachment.uploaded_at.desc())
+    )
+    return result.scalars().all()
+
+
+async def get_attachment(db: AsyncSession, attachment_id: uuid.UUID) -> FeeAttachment | None:
+    result = await db.execute(
+        select(FeeAttachment).where(FeeAttachment.id == attachment_id, FeeAttachment.is_deleted.is_(False))
+    )
+    return result.scalar_one_or_none()
+
+
+async def soft_delete_attachment(db: AsyncSession, attachment: FeeAttachment) -> FeeAttachment:
+    attachment.is_deleted = True
+    db.add(attachment)
+    await db.commit()
+    await db.refresh(attachment)
+    return attachment