first commit

backend/app/repositories/image_repo.py

@@ -0,0 +1,39 @@
+from uuid import UUID
+
+from sqlalchemy import select, func
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.evidence_image import EvidenceImage, SourceApp, PageType
+from app.repositories.base import BaseRepository
+
+
+class ImageRepository(BaseRepository[EvidenceImage]):
+    def __init__(self, session: AsyncSession):
+        super().__init__(EvidenceImage, session)
+
+    async def find_by_hash(self, file_hash: str) -> EvidenceImage | None:
+        result = await self.session.execute(
+            select(EvidenceImage).where(EvidenceImage.file_hash == file_hash)
+        )
+        return result.scalar_one_or_none()
+
+    async def list_by_case(
+        self,
+        case_id: UUID,
+        source_app: SourceApp | None = None,
+        page_type: PageType | None = None,
+    ) -> list[EvidenceImage]:
+        query = select(EvidenceImage).where(EvidenceImage.case_id == case_id)
+        if source_app:
+            query = query.where(EvidenceImage.source_app == source_app)
+        if page_type:
+            query = query.where(EvidenceImage.page_type == page_type)
+        query = query.order_by(EvidenceImage.uploaded_at.desc())
+        result = await self.session.execute(query)
+        return list(result.scalars().all())
+
+    async def count_by_case(self, case_id: UUID) -> int:
+        result = await self.session.execute(
+            select(func.count()).select_from(EvidenceImage).where(EvidenceImage.case_id == case_id)
+        )
+        return result.scalar() or 0