Adding method to return a dict instead of a list for easier usage
This commit is contained in:
parent
fe5d856f98
commit
af44cc256c
1 changed files with 13 additions and 1 deletions
14
db.py
14
db.py
|
@ -41,7 +41,7 @@ class BaseDeDonnees:
|
|||
except sqlite3.Error as e: # s'il y a eu une erreur SQLite
|
||||
print(e)
|
||||
|
||||
def affichageResultat(self, curseur: Cursor) -> list:
|
||||
def affichageResultat(self, curseur: tuple) -> list:
|
||||
"""Affiche le résultat d'une requête."""
|
||||
tableau = []
|
||||
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:
|
||||
tableau.append(ligne) # on les ajoute au tableau
|
||||
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))
|
||||
|
|
Reference in a new issue