mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-11-02 18:23:40 -05:00
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from uuid import UUID
|
|
|
|
from sqlalchemy import or_, select
|
|
|
|
from mealie.db.models.group.mealplan import GroupMealPlanRules
|
|
from mealie.schema.meal_plan.plan_rules import PlanRulesDay, PlanRulesOut, PlanRulesType
|
|
|
|
from .repository_generic import RepositoryGeneric
|
|
|
|
|
|
class RepositoryMealPlanRules(RepositoryGeneric[PlanRulesOut, GroupMealPlanRules]):
|
|
def by_group(self, group_id: UUID) -> "RepositoryMealPlanRules":
|
|
return super().by_group(group_id)
|
|
|
|
def get_rules(self, day: PlanRulesDay, entry_type: PlanRulesType) -> list[PlanRulesOut]:
|
|
stmt = select(GroupMealPlanRules).filter(
|
|
or_(
|
|
GroupMealPlanRules.day == day,
|
|
GroupMealPlanRules.day.is_(None),
|
|
GroupMealPlanRules.day == PlanRulesDay.unset.value,
|
|
),
|
|
or_(
|
|
GroupMealPlanRules.entry_type == entry_type,
|
|
GroupMealPlanRules.entry_type.is_(None),
|
|
GroupMealPlanRules.entry_type == PlanRulesType.unset.value,
|
|
),
|
|
)
|
|
|
|
rules = self.session.execute(stmt).scalars().all()
|
|
|
|
return [self.schema.model_validate(x) for x in rules]
|