munch-ease-backend/ingredients/db.py

from typing import Any, AsyncIterator, ClassVar, List, Optional

from pydantic import BaseModel, field_validator

from products import Product


class Ingredient(BaseModel):
    KEYS: ClassVar[List[str]] = [
        "id",
        "name",
        "line",
        "preparation",
        "unit",
        "quantity",
        "product_id",
        "recipe_id",
        "meal_id",
    ]
    id: int = -1
    name: str
    line: str
    unit: str
    quantity: float | str
    preparation: str
    product_id: Optional[int] = None
    recipe_id: Optional[int] = None
    meal_id: Optional[int] = None
    product: Optional[Product] = None

    # Ensure quantity is stored as a float even if provided as a string in tests
    @field_validator("quantity", mode="before")
    @classmethod
    def _coerce_quantity(cls, v: Any) -> Any:
        if isinstance(v, str):
            try:
                return float(v)
            except ValueError:
                return v
        return v


async def create(conn):
    await conn.execute(
        """
        CREATE TABLE IF NOT EXISTS Ingredient (
            id INTEGER PRIMARY KEY,
            name TEXT,
            line TEXT,
            preparation TEXT,
            unit TEXT,
            quantity REAL,
            product_id INTEGER,
            recipe_id INTEGER,
            meal_id INTEGER,
            FOREIGN KEY (product_id) REFERENCES Product(id),
            FOREIGN KEY (recipe_id) REFERENCES Recipe(id),
            FOREIGN KEY (meal_id) REFERENCES Meal(id)
        );"""
    )


async def insert_ingredient(conn, ingredient: Ingredient):
    if ingredient.product:
        ingredient.product_id = ingredient.product.id

    if ingredient.product_id is None or ingredient.product_id < 0:
        ingredient.product_id = None

    async with conn.execute(
        """
        INSERT INTO Ingredient (name, line, preparation, unit, quantity, product_id, recipe_id, meal_id)
        VALUES (?, ?, ?, ?, ?, ?, ?, ?)
    """,
        (
            ingredient.name,
            ingredient.line,
            ingredient.preparation,
            ingredient.unit,
            ingredient.quantity,
            ingredient.product_id,
            ingredient.recipe_id,
            ingredient.meal_id,
        ),
    ) as cursor:
        ingredient.id = cursor.lastrowid


async def find_ingredient_by_id(conn, ingredient_id: int) -> Optional[Ingredient]:
    ingredient_cols = [f"ingredient.{key}" for key in Ingredient.KEYS]
    product_cols = [f"product.{key}" for key in Product.KEYS]

    async with conn.execute(
        f"""
    SELECT {','.join(ingredient_cols + product_cols)} FROM Ingredient
        LEFT JOIN Product ON Ingredient.product_id = Product.id
        WHERE Ingredient.id = ?
    """,
        (ingredient_id,),
    ) as cursor:
        async for row in cursor:
            product_map = {k: v for k, v in zip(Product.KEYS, row[len(Ingredient.KEYS) :])}
            product = Product(**product_map) if product_map["id"] else None
            return Ingredient(
                **{k: v for k, v in zip(Ingredient.KEYS, row[: len(Ingredient.KEYS)])},
                product=product,
            )
    return None


async def find_ingredients_by_recipe_id(conn, recipe_id: int) -> AsyncIterator[Ingredient]:
    ingredient_cols = [f"ingredient.{key}" for key in Ingredient.KEYS]
    product_cols = [f"product.{key}" for key in Product.KEYS]

    async with conn.execute(
        f"""
    SELECT {','.join(ingredient_cols + product_cols)} FROM Ingredient
        LEFT JOIN Product ON Ingredient.product_id = Product.id
        WHERE recipe_id = ?
    """,
        (recipe_id,),
    ) as cursor:
        async for row in cursor:
            product_map = {k: v for k, v in zip(Product.KEYS, row[len(Ingredient.KEYS) :])}
            product = Product(**product_map) if product_map["id"] else None
            yield Ingredient(
                **{k: v for k, v in zip(Ingredient.KEYS, row[: len(Ingredient.KEYS)])},
                product=product,
            )


async def find_ingredients_by_meal_id(conn, meal_id: int) -> AsyncIterator[Ingredient]:
    ingredient_cols = [f"ingredient.{key}" for key in Ingredient.KEYS]
    product_cols = [f"product.{key}" for key in Product.KEYS]

    async with conn.execute(
        f"""
    SELECT {','.join(ingredient_cols + product_cols)} FROM Ingredient
        LEFT JOIN Product ON Ingredient.product_id = Product.id
        WHERE meal_id = ?
    """,
        (meal_id,),
    ) as cursor:
        async for row in cursor:
            product_map = {k: v for k, v in zip(Product.KEYS, row[len(Ingredient.KEYS) :])}
            product = Product(**product_map) if product_map["id"] else None
            yield Ingredient(
                **{k: v for k, v in zip(Ingredient.KEYS, row[: len(Ingredient.KEYS)])},
                product=product,
            )


async def delete_ingredients_by_meal_id(conn, meal_id: int):
    await conn.execute(
        """
        DELETE FROM Ingredient
        WHERE meal_id = ?
    """,
        (meal_id,),
    )