Coverage for opt/mealie/lib/python3.12/site-packages/mealie/repos/repository_meals.py: 61%

1from datetime import UTC, datetime 1 ctx1a

2 

3from sqlalchemy import select 1 ctx1a

4 

5from mealie.db.models.household import GroupMealPlan 1 ctx1a

6from mealie.schema.meal_plan.new_meal import ReadPlanEntry 1 ctx1a

7 

8from .repository_generic import HouseholdRepositoryGeneric 1 ctx1a

9 

10 

11class RepositoryMeals(HouseholdRepositoryGeneric[ReadPlanEntry, GroupMealPlan]): 1 ctx1a

12 def get_today(self, tz=UTC) -> list[ReadPlanEntry]: 1 ctx1a

13 if not self.household_id: 13 ↛ 14line 13 didn't jump to line 14 because the condition on line 13 was never true1 ctx1b

14 raise Exception("household_id not set") 

15 

16 today = datetime.now(tz=tz).date() 1 ctx1b

17 stmt = select(GroupMealPlan).filter( 1 ctx1b

18 GroupMealPlan.date == today, GroupMealPlan.household_id == self.household_id 

19 ) 

20 plans = self.session.execute(stmt).scalars().all() 1 ctx1b

21 return [self.schema.model_validate(x) for x in plans] 1 ctx1b

22 

23 def get_meals_by_date_range(self, start_date: datetime, end_date: datetime) -> list[ReadPlanEntry]: 1 ctx1a

24 if not self.household_id: 

25 raise Exception("household_id not set") 

26 

27 stmt = select(GroupMealPlan).filter( 

28 GroupMealPlan.date >= start_date.date(), 

29 GroupMealPlan.date <= end_date.date(), 

30 GroupMealPlan.household_id == self.household_id, 

31 ) 

32 plans = self.session.execute(stmt).scalars().all() 

33 return [self.schema.model_validate(x) for x in plans]