first commit

backend/app/repositories/assessment_repo.py

@@ -0,0 +1,34 @@
+from uuid import UUID
+
+from sqlalchemy import select, func
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.orm import selectinload
+
+from app.models.assessment import FraudAssessment, ConfidenceLevel
+from app.repositories.base import BaseRepository
+
+
+class AssessmentRepository(BaseRepository[FraudAssessment]):
+    def __init__(self, session: AsyncSession):
+        super().__init__(FraudAssessment, session)
+
+    async def list_by_case(
+        self,
+        case_id: UUID,
+        confidence_level: ConfidenceLevel | None = None,
+    ) -> tuple[list[FraudAssessment], int]:
+        query = (
+            select(FraudAssessment)
+            .options(selectinload(FraudAssessment.transaction))
+            .where(FraudAssessment.case_id == case_id)
+        )
+        count_q = select(func.count()).select_from(FraudAssessment).where(FraudAssessment.case_id == case_id)
+
+        if confidence_level:
+            query = query.where(FraudAssessment.confidence_level == confidence_level)
+            count_q = count_q.where(FraudAssessment.confidence_level == confidence_level)
+
+        total = (await self.session.execute(count_q)).scalar() or 0
+        query = query.order_by(FraudAssessment.created_at.asc())
+        result = await self.session.execute(query)
+        return list(result.scalars().all()), total