Add schema. Fix migration name

openfoodfacts · Nov 21, 2023 · 255c273 · 255c273
1 parent 0895125
commit 255c273
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/...1856_aead2a83cfa8_create_product_table.py → ...1856_df15f7cacf48_create_product_table.py b/...1856_aead2a83cfa8_create_product_table.py → ...1856_df15f7cacf48_create_product_table.py
diff --git a/app/schemas.py b/app/schemas.py
@@ -2,10 +2,17 @@
 from typing import Optional
 
 from fastapi_filter.contrib.sqlalchemy import Filter
-from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_validator
+from pydantic import (
+    AnyHttpUrl,
+    BaseModel,
+    ConfigDict,
+    Field,
+    field_serializer,
+    field_validator,
+)
 from sqlalchemy_utils import Currency
 
-from app.enums import LocationOSMType
+from app.enums import LocationOSMType, ProductOFFSource
 from app.models import Price
 
 
@@ -16,6 +23,22 @@ class UserBase(BaseModel):
     token: str
 
 
+class ProductCreate(BaseModel):
+    model_config = ConfigDict(from_attributes=True, arbitrary_types_allowed=True)
+
+    code: str
+
+
+class ProductBase(BaseModel):
+    id: int
+    off_source: ProductOFFSource | None
+    off_name: str | None
+    off_quantity: int | None
+    off_image_url: AnyHttpUrl | None
+    created: datetime
+    updated: datetime | None
+
+
 class LocationCreate(BaseModel):
     model_config = ConfigDict(from_attributes=True, arbitrary_types_allowed=True)