optimisation fonction prix()

main.py

@@ -35,62 +35,25 @@ class _ScraperData:
             return None
         return cast(dict[str, object], current_data.get("attributes"))
 
-    def prix(self) -> float:
+    def prix(self) -> float | None:
         """
-            Retourne le prix unitaire d'une bouteille (75cl).
+        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
+        Si aucun prix n'est disponible, retourne None.
+        """
 
-            Formule générale :
-                prix_unitaire = offerPrice / (nbunit * equivbtl)
-
-            """
-
-        content = self._getcontent()  
-
-        # si content n'existe pas -> erreur
+        content = self._getcontent()
         if content is None:
-            raise ValueError("Contenu introuvable")
+            return None
 
-        # 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
+        # Vérifie que items existe et n'est pas vide
         if not isinstance(items, list) or len(items) == 0:
-            raise ValueError("Aucun prix disponible (items vide)")
+            return None
 
-        # --------------------------
-        # 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
+        prix_calcule: float | None = None
 
-            # 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):
@@ -102,23 +65,21 @@ class _ScraperData:
             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:
+            if not isinstance(p, (int, float)) or not nbunit or not equivbtl:
+                continue
 
-                # Calcul du nombre total de bouteilles équivalentes
-                denom = float(nbunit) * float(equivbtl)
+            nb = float(nbunit)
+            eq = float(equivbtl)
 
-                # Évite division par zéro
-                if denom > 0:
+            if nb <= 0 or eq <= 0:
+                continue
 
-                    # Calcul du prix unitaire
-                    prix_unitaire = float(p) / denom
+            if nb == 1 and eq == 1:
+                return float(p)
 
-                    # Arrondi à 2 décimales
-                    return round(prix_unitaire, 2)
+            prix_calcule = round(float(p) / (nb * eq), 2)
 
-        # Si aucun prix trouvé
-        raise ValueError("Impossible de trouver le prix unitaire.")
+        return prix_calcule
 
     def appellation(self) -> str | None:
         """_summary_
@@ -171,6 +132,23 @@ class _ScraperData:
     def getdata(self) -> dict[str, object]:
         return self._data
 
+    def informations(self) -> str:
+        """
+        Retourne toutes les informations sous la forme :
+        "Appelation,Parker,J.Robinson,J.Suckling,Prix"
+        """
+
+        appellation = self.appellation()
+        parker = self.parker()
+        robinson = self.robinson()
+        suckling = self.suckling()
+        try:
+            prix = self.prix()
+        except ValueError:
+            prix = None
+            
+        return f"{appellation},{parker},{robinson},{suckling},{prix}"
+
 
 class Scraper:
     """

test_main.py

@@ -225,18 +225,20 @@ def test_critiques(scraper: Scraper):
     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 vide.prix() is None
+    assert poubelle.prix() is None
     assert contenu.prix() == 65.0
-
+    
+    
+def test_informations(scraper: Scraper):
+    contenu = scraper.getjsondata("nino-negri-5-stelle-sfursat-2022.html")
+    assert contenu.informations() == "Sforzato di Valtellina,91,17,93.5,65.0"
+    vide = scraper.getjsondata("")
+    poubelle = scraper.getjsondata("poubelle")
+    assert vide.informations() == "None,None,None,None,None"
+    assert poubelle.informations() == "None,None,None,None,None"