64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
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
|