from sqlalchemy import select
from sqlalchemy.orm import Session

from app.models.entities import GrowthRewardRule


class GrowthRulesRepository:
    def __init__(self, db: Session) -> None:
        self.db = db

    def get_rule(self, rule_type: str) -> GrowthRewardRule | None:
        return self.db.scalar(
            select(GrowthRewardRule).where(GrowthRewardRule.rule_type == rule_type)
        )