import sqlalchemy as sa

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


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