Modification de la fonction prix()

main.py

@@ -6,9 +6,8 @@ from json import loads
 
 
 class _ScraperData:
-    def __init__(self, data: dict[str, object], scraper: Scraper | None = None) -> None:
+    def __init__(self, data: dict[str, object]) -> None:
         self._data: dict[str, object] = data
-        self._scraper: Scraper | None = scraper
 
     def _getcontent(self) -> dict[str, object] | None:
         """_summary_
@@ -36,6 +35,91 @@ class _ScraperData:
             return None
         return cast(dict[str, object], current_data.get("attributes"))
 
+    def prix(self) -> float:
+        """
+            Retourne le prix unitaire d'une bouteille (75cl).
+
+            Le JSON contient plusieurs formats de vente dans content["items"] :
+            - bouteille seule : nbunit = 1 et equivbtl = 1 -> prix direct
+            - caisse de plusieurs bouteilles : nbunit > 1 -> on divise le prix total
+            - formats spéciaux (magnum etc.) : equivbtl > 1 -> même calcul
+
+            Formule générale :
+                prix_unitaire = offerPrice / (nbunit * equivbtl)
+
+            """
+
+        content = self._getcontent()  
+
+        # si content n'existe pas -> erreur
+        if content is None:
+            raise ValueError("Contenu introuvable")
+
+        # On récupère la liste des formats disponibles (bouteille, carton...)
+        items = content.get("items")
+
+        # Vérification que items est bien une liste non vide
+        if not isinstance(items, list) or len(items) == 0:
+            raise ValueError("Aucun prix disponible (items vide)")
+
+        # --------------------------
+        # CAS 1 : bouteille unitaire
+        # --------------------------
+        # On cherche un format où nbunit=1 et equivbtl=1 ->bouteille standard 75cl 
+        for item in items:
+            
+            if not isinstance(item, dict):
+                continue
+
+            # On récupère les attributs du format
+            attrs = item.get("attributes", {})
+
+            # On récupère nbunit et equivbtl
+            nbunit = attrs.get("nbunit", {}).get("value")
+            equivbtl = attrs.get("equivbtl", {}).get("value")
+
+            # Si c'est une bouteille unitaire
+            if nbunit == "1" and equivbtl == "1":
+
+                p = item.get("offerPrice")
+
+                # Vérification que c'est bien un nombre
+                if isinstance(p, (int, float)):
+                    return float(p)
+
+        # --------------------------
+        # CAS 2 : caisse ou autre format
+        # --------------------------
+        # On calcule le prix unitaire à partir du prix total
+        for item in items:
+
+            if not isinstance(item, dict):
+                continue
+
+            p = item.get("offerPrice")
+            attrs = item.get("attributes", {})
+
+            nbunit = attrs.get("nbunit", {}).get("value")
+            equivbtl = attrs.get("equivbtl", {}).get("value")
+
+            # Vérification que toutes les valeurs existent
+            if isinstance(p, (int, float)) and nbunit and equivbtl:
+
+                # Calcul du nombre total de bouteilles équivalentes
+                denom = float(nbunit) * float(equivbtl)
+
+                # Évite division par zéro
+                if denom > 0:
+
+                    # Calcul du prix unitaire
+                    prix_unitaire = float(p) / denom
+
+                    # Arrondi à 2 décimales
+                    return round(prix_unitaire, 2)
+
+        # Si aucun prix trouvé
+        raise ValueError("Impossible de trouver le prix unitaire.")
+
     def appellation(self) -> str | None:
         """_summary_
 
@@ -222,3 +306,4 @@ class Scraper:
             raise ValueError(f"Clé manquante dans le JSON : {key}")
 
         return _ScraperData(cast(dict[str, object], current_data))
+

requirements.txt

@@ -1,4 +1,3 @@
 requests>=2.32.5
 requests-mock>=1.12.1
 beautifulsoup4>=4.14.3
-    

test_main.py

@@ -94,17 +94,6 @@ def mock_site():
                                                 "type": "CHECKBOX",
                                                 "isSpirit": False,
                                             },
-                                        },
-                                        "stock": 12,
-                                        "availability": "2026-02-05",
-                                        "isCustomizable": False,
-                                        "gtin_cond": "",
-                                        "gtin_unit": "",
-                                        "stockOrigin": "EUR",
-                                        "isPrevSale": False,
-                                    }
-                                ],
-                                "attributes": {
                                             "equivbtl": {
                                                 "valueId": "1",
                                                 "name": "equivbtl",
@@ -117,6 +106,17 @@ def mock_site():
                                                 "value": "6",
                                                 "isSpirit": False,
                                             },
+                                        },
+                                        "stock": 12,
+                                        "availability": "2026-02-05",
+                                        "isCustomizable": False,
+                                        "gtin_cond": "",
+                                        "gtin_unit": "",
+                                        "stockOrigin": "EUR",
+                                        "isPrevSale": False,
+                                    }
+                                ],
+                                "attributes": {
                                     "appellation": {
                                         "valueId": "433",
                                         "name": "Appellation",
@@ -207,6 +207,7 @@ def test_fonctionprivee(scraper: Scraper):
     assert contenu._getattributes() is not None
 
 
+
 def test_critiques(scraper: Scraper):
     vide = scraper.getjsondata("")
     poubelle = scraper.getjsondata("poubelle")
@@ -223,3 +224,19 @@ def test_critiques(scraper: Scraper):
     assert contenu.robinson() == "17"
     assert contenu.suckling() == "93.5"
     assert contenu._getcritiques("test_ts") is None
+
+def test_prix(scraper: Scraper):
+    vide = scraper.getjsondata("")
+    poubelle = scraper.getjsondata("poubelle")
+    contenu = scraper.getjsondata("nino-negri-5-stelle-sfursat-2022.html")
+
+    # Cas vide : items == [] -> on ne peut pas calculer -> ValueError
+    with pytest.raises(ValueError):
+        _ = vide.prix()
+
+    # Cas poubelle : JSON incomplet -> _getcontent() None -> ValueError
+    with pytest.raises(ValueError):
+        _ = poubelle.prix()
+
+    assert contenu.prix() == 65.0
+