16 lines
406 B
Python
16 lines
406 B
Python
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)
|
|
)
|
|
|