munch-ease-backend/shopping/models.py

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)