import sqlalchemy as sa

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


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