From af44cc256c81837f89e50ad32db1fa750e0b7739 Mon Sep 17 00:00:00 2001 From: Mylloon Date: Sat, 20 Nov 2021 00:57:21 +0100 Subject: [PATCH] Adding method to return a dict instead of a list for easier usage --- db.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/db.py b/db.py index d7aa3e6..6fda53b 100644 --- a/db.py +++ b/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))