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
|
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))
|
||||||
|
|
Reference in a new issue