Adding method to return a dict instead of a list for easier usage

This commit is contained in:
Mylloon 2021-11-20 00:57:21 +01:00
parent fe5d856f98
commit af44cc256c

14
db.py
View file

@ -41,7 +41,7 @@ class BaseDeDonnees:
except sqlite3.Error as e: # s'il y a eu une erreur SQLite except sqlite3.Error as e: # s'il y a eu une erreur SQLite
print(e) print(e)
def affichageResultat(self, curseur: Cursor) -> list: def affichageResultat(self, curseur: tuple) -> list:
"""Affiche le résultat d'une requête.""" """Affiche le résultat d'une requête."""
tableau = [] tableau = []
if curseur == None: # si le curseur est vide il n'y a rien a affiché (tableau vide) if curseur == None: # si le curseur est vide il n'y a rien a affiché (tableau vide)
@ -50,3 +50,15 @@ class BaseDeDonnees:
for ligne in lignes: for ligne in lignes:
tableau.append(ligne) # on les ajoute au tableau tableau.append(ligne) # on les ajoute au tableau
return tableau # on le renvoie return tableau # on le renvoie
def affichageResultatDictionnaire(self, cles, curseur: Cursor) -> dict:
"""
Même but que `affichageResultat()` mais avec
les clés qui correspondent aux valeurs.
"""
valeurs = self.affichageResultat(curseur)[0]
if type(cles) not in [list, tuple]:
cles = [cles]
if len(cles) != len(valeurs):
raise IndexError # il y a pas autant de clés que de valeurs
return dict(zip(cles, valeurs))