160 lines
4.9 KiB
Python
160 lines
4.9 KiB
Python
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,),
|
|
)
|