35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
|
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
|