59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
from datetime import datetime
|
|
from enum import Enum
|
|
from typing import ClassVar, List, Optional
|
|
|
|
from pydantic import Field
|
|
|
|
from common import BaseLinkedModel
|
|
from persons.models import Person
|
|
|
|
|
|
class ShoppingListItem(BaseLinkedModel):
|
|
KEYS: ClassVar[List[str]] = [
|
|
"id",
|
|
"ingredient_id",
|
|
"list_id",
|
|
"person_id",
|
|
"meal_id",
|
|
"recipe_id",
|
|
"created_date",
|
|
"household_id",
|
|
"quantity",
|
|
"unit",
|
|
"added_by_id",
|
|
"purchased_at",
|
|
]
|
|
id: int = -1
|
|
list_id: Optional[int] = None
|
|
|
|
person_id: int = -1
|
|
|
|
ingredient_id: Optional[int] = None
|
|
|
|
recipe_id: Optional[int] = None
|
|
|
|
meal_id: Optional[int] = None
|
|
|
|
created_date: datetime = Field(default_factory=lambda: datetime.now().astimezone())
|
|
household_id: int | None = None
|
|
quantity: float | None = None
|
|
unit: str | None = None
|
|
added_by_id: int | None = None
|
|
purchased_at: datetime | None = None
|
|
added_by_name: str | None = None
|
|
|
|
|
|
class StoreEnum(str, Enum):
|
|
woolworths = "woolworths"
|
|
coles = "coles"
|
|
home = ""
|
|
|
|
|
|
class ShoppingList(BaseLinkedModel):
|
|
KEYS: ClassVar[List[str]] = ["id", "created_date", "store_name"]
|
|
id: int = -1
|
|
created_date: datetime = Field(default_factory=lambda: datetime.now().astimezone())
|
|
store_name: StoreEnum = StoreEnum.home
|
|
purchased_by_id: int = -1
|
|
purchased_by: Optional[Person] = None
|
|
items: List[ShoppingListItem] = Field(default_factory=list)
|