import sqlalchemy as sa

from app.adapters.db import Repository
from app.models import DeliveryQuotaDefault


class DeliveryQuotaDefaultRepository(Repository[DeliveryQuotaDefault]):
    def get_by_domain_id_and_provider(
        self, *, domain_id: str, provider: str
    ) -> DeliveryQuotaDefault | None:
        query = sa.select(DeliveryQuotaDefault).where(
            DeliveryQuotaDefault.domain_id == domain_id,
            DeliveryQuotaDefault.provider == provider,
        )
        result = self.db.session.execute(query)
        return result.scalar_one_or_none()