Modifications:

- Fix typo
- Remove useless function
- Return a list of dict instead of a list of tuples in `listeStocks` for better clarity
This commit is contained in:
Mylloon 2021-11-25 16:21:24 +01:00
parent a278b6bada
commit 78c91d1b7e

View file

@ -21,7 +21,7 @@ class Stock(BaseDeDonnees):
""" """
self.requete(requete) self.requete(requete)
# Ajout d'un stock par défaut si aucun stock n'existe dans la base de donnée # 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 = { defaut = {
"fruits legumes": [ "fruits legumes": [
"banane", "banane",
@ -96,15 +96,8 @@ class Stock(BaseDeDonnees):
""" """
self.requete(requete, id) self.requete(requete, id)
def listStocks(self) -> list: def listeStocks(self) -> list:
"""Retourne la liste des éléments en stock.""" """Retourne la liste des éléments en stock sous forme de dictionnaire."""
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é)."""
recuperation = [ recuperation = [
"id", "id",
"type", "type",
@ -113,22 +106,10 @@ class Stock(BaseDeDonnees):
"prix", "prix",
"image_url" "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""" requete = f"""
SELECT {", ".join(recuperation)} FROM stocks SELECT {", ".join(recuperation)} FROM stocks
WHERE pseudo = ?
""" """
stock = nom return [dict(zip(recuperation, element)) for element in self.affichageResultat(self.requete(requete))]
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))
def stockExistant(self, stock: str) -> bool: def stockExistant(self, stock: str) -> bool:
"""Vérifie si l"stock donnée existe déjà dans la base de donnée.""" """Vérifie si l"stock donnée existe déjà dans la base de donnée."""