munch-ease-backend/ingredients/models.py

from __future__ import annotations

from typing import Any, ClassVar, List, Optional

from pydantic import Field, field_validator

from common import ApiModel
from products import Product
from units import ALL_UNITS


class Ingredient(ApiModel):
    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 = Field(
        title="Unit",
        description="Measurement unit (enum values are advisory; runtime accepts any string)",
        json_schema_extra={"enum": [u.name for u in ALL_UNITS]},
    )
    quantity: float
    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

    # Quantity must be > 0
    @field_validator("quantity")
    @classmethod
    def _positive_quantity(cls, v: float) -> float:
        if v is None:
            return v
        if isinstance(v, (int, float)) and v <= 0:
            raise ValueError("Ingredient quantity must be greater than 0")
        return v

    # Backward compatibility: DB may contain NULL preparation; normalize to empty string
    @field_validator("preparation", mode="before")
    @classmethod
    def _normalize_preparation(cls, v: Any) -> Any:
        return "" if v is None else v