Public Access
feat: add IngredientCreate/Update/Read and IngredientGroceryMatchRead schemas
This commit is contained in:
@@ -0,0 +1,59 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from decimal import Decimal
|
||||||
|
from typing import List, Optional
|
||||||
|
from uuid import UUID
|
||||||
|
|
||||||
|
from pydantic import BaseModel, Field, field_validator
|
||||||
|
|
||||||
|
|
||||||
|
class IngredientBase(BaseModel):
|
||||||
|
name: str = Field(min_length=1, max_length=200)
|
||||||
|
aliases: List[str] = Field(default_factory=list)
|
||||||
|
aisle: Optional[str] = Field(default=None, max_length=100)
|
||||||
|
unit: Optional[str] = Field(default=None, max_length=50)
|
||||||
|
typical_price: Optional[Decimal] = None
|
||||||
|
|
||||||
|
@field_validator("aliases")
|
||||||
|
@classmethod
|
||||||
|
def _strip_and_drop_empty(cls, v: List[str]) -> List[str]:
|
||||||
|
return [s.strip() for s in v if s and s.strip()]
|
||||||
|
|
||||||
|
|
||||||
|
class IngredientCreate(IngredientBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class IngredientUpdate(BaseModel):
|
||||||
|
name: Optional[str] = Field(default=None, min_length=1, max_length=200)
|
||||||
|
aliases: Optional[List[str]] = None
|
||||||
|
aisle: Optional[str] = Field(default=None, max_length=100)
|
||||||
|
unit: Optional[str] = Field(default=None, max_length=50)
|
||||||
|
typical_price: Optional[Decimal] = None
|
||||||
|
|
||||||
|
@field_validator("aliases")
|
||||||
|
@classmethod
|
||||||
|
def _strip_and_drop_empty(cls, v: Optional[List[str]]) -> Optional[List[str]]:
|
||||||
|
if v is None:
|
||||||
|
return None
|
||||||
|
return [s.strip() for s in v if s and s.strip()]
|
||||||
|
|
||||||
|
|
||||||
|
class IngredientRead(IngredientBase):
|
||||||
|
id: UUID
|
||||||
|
|
||||||
|
model_config = {"from_attributes": True}
|
||||||
|
|
||||||
|
|
||||||
|
class IngredientGroceryMatchRead(BaseModel):
|
||||||
|
id: UUID
|
||||||
|
ingredient_id: UUID
|
||||||
|
grocery_item_id: UUID
|
||||||
|
confidence: Decimal
|
||||||
|
source: str
|
||||||
|
grocery_item_name: Optional[str] = None
|
||||||
|
current_price: Optional[Decimal] = None
|
||||||
|
regular_price: Optional[Decimal] = None
|
||||||
|
is_on_sale: Optional[bool] = None
|
||||||
|
|
||||||
|
model_config = {"from_attributes": True}
|
||||||
@@ -0,0 +1,34 @@
|
|||||||
|
from decimal import Decimal
|
||||||
|
from uuid import uuid4
|
||||||
|
|
||||||
|
from app.schemas.ingredient import IngredientCreate, IngredientRead, IngredientGroceryMatchRead
|
||||||
|
|
||||||
|
|
||||||
|
def test_ingredient_create_normalizes_aliases() -> None:
|
||||||
|
payload = IngredientCreate(
|
||||||
|
name="Chicken Thighs",
|
||||||
|
aliases=[" chicken thigh ", "BSL chicken thighs", ""],
|
||||||
|
aisle="meat_seafood",
|
||||||
|
unit="lb",
|
||||||
|
)
|
||||||
|
assert payload.name == "Chicken Thighs"
|
||||||
|
assert payload.aliases == ["chicken thigh", "BSL chicken thighs"]
|
||||||
|
|
||||||
|
|
||||||
|
def test_ingredient_match_read_round_trip() -> None:
|
||||||
|
iid = uuid4()
|
||||||
|
gid = uuid4()
|
||||||
|
raw = {
|
||||||
|
"id": uuid4(),
|
||||||
|
"ingredient_id": iid,
|
||||||
|
"grocery_item_id": gid,
|
||||||
|
"confidence": Decimal("0.875"),
|
||||||
|
"source": "auto",
|
||||||
|
"grocery_item_name": "Foster Farms Chicken Thighs Family Pack",
|
||||||
|
"current_price": Decimal("3.99"),
|
||||||
|
"regular_price": Decimal("5.49"),
|
||||||
|
"is_on_sale": True,
|
||||||
|
}
|
||||||
|
parsed = IngredientGroceryMatchRead.model_validate(raw)
|
||||||
|
assert parsed.confidence == Decimal("0.875")
|
||||||
|
assert parsed.is_on_sale is True
|
||||||
Reference in New Issue
Block a user