diff --git a/stock.py b/stock.py index 51164d9..687e2e1 100644 --- a/stock.py +++ b/stock.py @@ -21,7 +21,7 @@ class Stock(BaseDeDonnees): """ self.requete(requete) # Ajout d'un stock par défaut si aucun stock n'existe dans la base de donnée - if len(self.listStocks()) == 0: + if len(self.listeStocks()) == 0: defaut = { "fruits legumes": [ "banane", @@ -96,15 +96,8 @@ class Stock(BaseDeDonnees): """ self.requete(requete, id) - def listStocks(self) -> list: - """Retourne la liste des éléments en stock.""" - requete = """ - SELECT id, type, nom, quantite, prix, image_url FROM stocks - """ - return self.affichageResultat(self.requete(requete)) - - def recuperationStock(self, id: int = None, nom: str = None) -> dict: - """Retourne les informations d'un stock grâce à son ID ou son nom (ID en priorité).""" + def listeStocks(self) -> list: + """Retourne la liste des éléments en stock sous forme de dictionnaire.""" recuperation = [ "id", "type", @@ -113,22 +106,10 @@ class Stock(BaseDeDonnees): "prix", "image_url" ] - if not id: # si la variable `id` n'est pas définie - if not nom: # si la variable `nom` n'est pas définie - raise ValueError # Aucun élément renseigné - else: # si seul un nom est renseigné, c'est ce qu'on va utilisé - requete = f""" - SELECT {", ".join(recuperation)} FROM stocks - WHERE pseudo = ? - """ - stock = nom - else: # si un id est renseigné, c'est ce qu'on va utilisé - requete = f""" - SELECT {", ".join(recuperation)} FROM stocks - WHERE id = ? - """ - stock = id - return self.affichageResultatDictionnaire(recuperation, self.requete(requete, stock)) + requete = f""" + SELECT {", ".join(recuperation)} FROM stocks + """ + return [dict(zip(recuperation, element)) for element in self.affichageResultat(self.requete(requete))] def stockExistant(self, stock: str) -> bool: """Vérifie si l"stock donnée existe déjà dans la base de donnée."""