last

backend/app/db/schemas.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
+import json
 from datetime import datetime
 from typing import Any, Optional
 
-from pydantic import BaseModel, HttpUrl
+from pydantic import BaseModel, ConfigDict, HttpUrl, field_validator
 
 
 class ProductBase(BaseModel):
@@ -30,13 +31,12 @@ class ProductUpdate(BaseModel):
 
 
 class ProductRead(ProductBase):
+    model_config = ConfigDict(from_attributes=True)
+
     id: int
     cree_le: datetime
     modifie_le: datetime
 
-    class Config:
-        orm_mode = True
-
 
 class ProductSnapshotBase(BaseModel):
     prix_actuel: Optional[float]
@@ -55,22 +55,48 @@ class ProductSnapshotBase(BaseModel):
     description: Optional[str] = None
     carateristique: Optional[dict[str, Any]] = None
     details: Optional[dict[str, Any]] = None
+    categorie_amazon: Optional[str] = None
     statut_scrap: Optional[str]
     message_erreur: Optional[str]
 
 
 class ProductSnapshotRead(ProductSnapshotBase):
+    model_config = ConfigDict(from_attributes=True)
+
     id: int
     produit_id: int
     scrape_le: datetime
 
-    class Config:
-        orm_mode = True
+    @field_validator("a_propos", mode="before")
+    @classmethod
+    def parse_a_propos(cls, v: Any) -> list[str] | None:
+        if v is None:
+            return None
+        if isinstance(v, str):
+            try:
+                return json.loads(v)
+            except json.JSONDecodeError:
+                return None
+        return v
+
+    @field_validator("carateristique", "details", mode="before")
+    @classmethod
+    def parse_json_dict(cls, v: Any) -> dict[str, Any] | None:
+        if v is None:
+            return None
+        if isinstance(v, str):
+            try:
+                return json.loads(v)
+            except json.JSONDecodeError:
+                return None
+        return v
 
 
 class ProductWithSnapshot(ProductBase):
     """Produit enrichi avec les données du dernier snapshot."""
 
+    model_config = ConfigDict(from_attributes=True)
+
     id: int
     cree_le: datetime
     modifie_le: datetime
@@ -92,12 +118,10 @@ class ProductWithSnapshot(ProductBase):
     description: Optional[str] = None
     carateristique: Optional[dict[str, Any]] = None
     details: Optional[dict[str, Any]] = None
+    categorie_amazon: Optional[str] = None
     dernier_scrape: Optional[datetime] = None
     statut_scrap: Optional[str] = None
 
-    class Config:
-        orm_mode = True
-
 
 class ProductCreateWithSnapshot(ProductBase):
     """Création d'un produit avec données de snapshot initiales (depuis preview)."""
@@ -119,3 +143,4 @@ class ProductCreateWithSnapshot(ProductBase):
     description: Optional[str] = None
     carateristique: Optional[dict[str, Any]] = None
     details: Optional[dict[str, Any]] = None
+    categorie_amazon: Optional[str] = None