felicity-lims/felicity/apps/patient/repository.py

import sqlalchemy as sa

from domain.patient.ports.repository import (
    IPatientRepository,
    IIdentificationRepository,
    IPatientIdentificationRepository,
)
from domain.shared.ports.paginator.cursor import PageCursor
from infrastructure.database.patient.entities import (
    Patient,
    Identification,
    PatientIdentification,
)
from infrastructure.database.repository.base import BaseRepository


class PatientRepository(BaseRepository[Patient], IPatientRepository):
    def __init__(self) -> None:
        self.model = Patient
        super().__init__()

    async def paginate_with_cursors(
            self,
            page_size: int | None = None,
            after_cursor: str | None = None,
            before_cursor: str | None = None,
            text: str | None = None,
            sort_by: list[str] | None = None,
            **kwargs,
    ) -> PageCursor:
        filters = {}

        _or_ = dict()
        if text:
            arg_list = [
                "first_name__ilike",
                "last_name__ilike",
                "middle_name__ilike",
                "client_patient_id__ilike",
                "patient_id__ilike",
                "client___name__ilike",
                "patient_id__ilike",
                "email__ilike",
                "phone_mobile__ilike",
                "phone_home__ilike",
            ]
            for _arg in arg_list:
                _or_[_arg] = f"%{text}%"

            filters = {sa.or_: _or_}

        return await super().paginate(
            page_size, after_cursor, before_cursor, filters, sort_by
        )


class IdentificationRepository(
    BaseRepository[Identification], IIdentificationRepository
):
    def __init__(self) -> None:
        self.model = Identification
        super().__init__()


class PatientIdentificationRepository(
    BaseRepository[PatientIdentification], IPatientIdentificationRepository
):
    def __init__(self) -> None:
        self.model = PatientIdentification
        super().__init__()