Coverage for opt/mealie/lib/python3.12/site-packages/mealie/db/models/household/mealplan.py: 90%

47 statements  

« prev     ^ index     » next       coverage.py v7.10.6, created at 2025-11-25 15:32 +0000

1import datetime 1a

2from typing import TYPE_CHECKING, Optional 1a

3 

4from sqlalchemy import Column, Date, ForeignKey, String, Table, UniqueConstraint, orm 1a

5from sqlalchemy.ext.associationproxy import AssociationProxy, association_proxy 1a

6from sqlalchemy.orm import Mapped, mapped_column 1a

7 

8from mealie.db.models.recipe.tag import Tag, plan_rules_to_tags 1a

9 

10from .._model_base import BaseMixins, SqlAlchemyBase 1a

11from .._model_utils.auto_init import auto_init 1a

12from .._model_utils.guid import GUID 1a

13from ..recipe.category import Category, plan_rules_to_categories 1a

14 

15if TYPE_CHECKING: 15 ↛ 16line 15 didn't jump to line 16 because the condition on line 15 was never true1a

16 from ..group import Group 

17 from ..recipe import RecipeModel 

18 from ..users import User 

19 from .household import Household 

20 

21plan_rules_to_households = Table( 1a

22 "plan_rules_to_households", 

23 SqlAlchemyBase.metadata, 

24 Column("group_plan_rule_id", GUID, ForeignKey("group_meal_plan_rules.id"), index=True), 

25 Column("household_id", GUID, ForeignKey("households.id"), index=True), 

26 UniqueConstraint("group_plan_rule_id", "household_id", name="group_plan_rule_id_household_id_key"), 

27) 

28 

29 

30class GroupMealPlanRules(BaseMixins, SqlAlchemyBase): 1a

31 __tablename__ = "group_meal_plan_rules" 1a

32 

33 id: Mapped[GUID] = mapped_column(GUID, primary_key=True, default=GUID.generate) 1a

34 group_id: Mapped[GUID | None] = mapped_column(GUID, ForeignKey("groups.id"), nullable=False, index=True) 1a

35 household_id: Mapped[GUID | None] = mapped_column(GUID, ForeignKey("households.id"), index=True) 1a

36 

37 day: Mapped[str] = mapped_column( 1a

38 String, nullable=False, default="unset" 

39 ) # "MONDAY", "TUESDAY", "WEDNESDAY", etc... 

40 entry_type: Mapped[str] = mapped_column( 1a

41 String, nullable=False, default="" 

42 ) # "breakfast", "lunch", "dinner", "side" 

43 query_filter_string: Mapped[str] = mapped_column(String, nullable=False, default="") 1a

44 

45 # Old filters - deprecated in favor of query filter strings 

46 categories: Mapped[list[Category]] = orm.relationship(Category, secondary=plan_rules_to_categories) 1a

47 tags: Mapped[list[Tag]] = orm.relationship(Tag, secondary=plan_rules_to_tags) 1a

48 households: Mapped[list["Household"]] = orm.relationship("Household", secondary=plan_rules_to_households) 1a

49 

50 @auto_init() 1a

51 def __init__(self, **_) -> None: 1a

52 pass 1bcdejkfghi

53 

54 

55class GroupMealPlan(SqlAlchemyBase, BaseMixins): 1a

56 __tablename__ = "group_meal_plans" 1a

57 

58 date: Mapped[datetime.date] = mapped_column(Date, index=True, nullable=False) 1a

59 entry_type: Mapped[str] = mapped_column(String, index=True, nullable=False) 1a

60 title: Mapped[str] = mapped_column(String, index=True, nullable=False) 1a

61 text: Mapped[str] = mapped_column(String, nullable=False) 1a

62 

63 group_id: Mapped[GUID | None] = mapped_column(GUID, ForeignKey("groups.id"), index=True) 1a

64 group: Mapped[Optional["Group"]] = orm.relationship("Group", back_populates="mealplans") 1a

65 household_id: AssociationProxy[GUID] = association_proxy("user", "household_id") 1a

66 household: AssociationProxy["Household"] = association_proxy("user", "household") 1a

67 user_id: Mapped[GUID | None] = mapped_column(GUID, ForeignKey("users.id"), index=True) 1a

68 user: Mapped[Optional["User"]] = orm.relationship("User", back_populates="mealplans") 1a

69 

70 recipe_id: Mapped[GUID | None] = mapped_column(GUID, ForeignKey("recipes.id"), index=True) 1a

71 recipe: Mapped[Optional["RecipeModel"]] = orm.relationship( 1a

72 "RecipeModel", back_populates="meal_entries", uselist=False 

73 ) 

74 

75 @auto_init() 1a

76 def __init__(self, **_) -> None: 1a

77 pass 1bcdefghi