first commit

backend/app/repositories/transaction_repo.py

@@ -0,0 +1,40 @@
+from uuid import UUID
+
+from sqlalchemy import select, func
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.transaction import TransactionRecord
+from app.repositories.base import BaseRepository
+
+
+class TransactionRepository(BaseRepository[TransactionRecord]):
+    def __init__(self, session: AsyncSession):
+        super().__init__(TransactionRecord, session)
+
+    async def list_by_case(
+        self,
+        case_id: UUID,
+        filter_type: str | None = None,
+    ) -> tuple[list[TransactionRecord], int]:
+        query = select(TransactionRecord).where(TransactionRecord.case_id == case_id)
+        count_q = select(func.count()).select_from(TransactionRecord).where(TransactionRecord.case_id == case_id)
+
+        if filter_type == "unique":
+            query = query.where(TransactionRecord.is_duplicate.is_(False))
+            count_q = count_q.where(TransactionRecord.is_duplicate.is_(False))
+        elif filter_type == "duplicate":
+            query = query.where(TransactionRecord.is_duplicate.is_(True))
+            count_q = count_q.where(TransactionRecord.is_duplicate.is_(True))
+
+        total = (await self.session.execute(count_q)).scalar() or 0
+        query = query.order_by(TransactionRecord.trade_time.asc())
+        result = await self.session.execute(query)
+        return list(result.scalars().all()), total
+
+    async def get_all_by_case(self, case_id: UUID) -> list[TransactionRecord]:
+        result = await self.session.execute(
+            select(TransactionRecord)
+            .where(TransactionRecord.case_id == case_id)
+            .order_by(TransactionRecord.trade_time.asc())
+        )
+        return list(result.scalars().all())