From 3b6b9b4cf7fea72a19eaff792ea62b0452205fb0 Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@tutanota.com>
Date: Thu, 18 Nov 2021 20:40:34 +0100
Subject: [PATCH] Adding DB support

---
 db.py    | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
 users.py | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 105 insertions(+)
 create mode 100644 db.py
 create mode 100644 users.py

diff --git a/db.py b/db.py
new file mode 100644
index 0000000..94aa2fe
--- /dev/null
+++ b/db.py
@@ -0,0 +1,51 @@
+import sqlite3
+
+class BaseDeDonnees:
+    """Gère la base de donnée."""
+    def __init__(self, urlBaseDeDonnee: str):
+        self.connexion = self.creerConnexion(urlBaseDeDonnee)
+
+    def creerConnexion(self, path):
+        """Connexion à une base de donnée SQLite"""
+        if not self.fichierExiste(path):
+            open(path, "x")
+        connnexion = None
+        try:
+            connnexion = sqlite3.connect(path)
+        except sqlite3.Error as e:
+            print(e)
+        return connnexion
+
+    def fichierExiste(self, path):
+        """Vérifie qu'un fichier existe"""
+        try:
+            open(path, "r")
+        except FileNotFoundError:
+            return False
+        else:
+            return True
+
+    def requete(self, requete, valeurs = None):
+        """Envois une requête vers la base de données"""
+        try:
+            curseur = self.connexion.cursor()
+            if valeurs:
+                if type(valeurs) not in [list, tuple]:
+                    valeurs = [valeurs]
+                curseur.execute(requete, valeurs)
+            else:
+                curseur.execute(requete)
+            self.connexion.commit()
+            return (curseur, curseur.lastrowid)
+        except sqlite3.Error as e:
+            print(e)
+
+    def affichageResultat(self, curseur):
+        """Affiche le résultat d'une requête"""
+        tableau = []
+        if curseur == None:
+            return tableau
+        lignes = curseur[0].fetchall()
+        for ligne in lignes:
+            tableau.append(ligne)
+        return tableau
diff --git a/users.py b/users.py
new file mode 100644
index 0000000..f236a79
--- /dev/null
+++ b/users.py
@@ -0,0 +1,54 @@
+from db import BaseDeDonnees
+
+class Utilisateurs(BaseDeDonnees):
+    """Gère une table "utilisateurs" pour une base de donnée donné."""
+    def __init__(self):
+        super().__init__(r"utilisateurs.sqlite3")
+
+    def creationTable(self) -> None:
+        """Créer la table qui stocker les utilisateurs."""
+        requete = """
+                  CREATE TABLE IF NOT EXISTS utilisateurs (
+                      nom TEXT,
+                      passe TEXT,
+                      metier TEXT
+                  );
+                  """
+        self.requete(requete)
+
+    def ajoutUtilisateurs(self, nom: str, motDePasse: str, metier: str) -> list:
+        """Ajoute un utilisateur et retourne l'ID de ce dernier."""
+        requete = """
+                  INSERT INTO utilisateurs (
+                      nom, passe, metier
+                  ) VALUES (
+                      ?, ?, ?
+                  );
+                  """
+        self.requete(requete, [nom, motDePasse, metier])
+        return self.affichageResultat(self.requete("SELECT last_insert_rowid();"))
+
+    def suppressionUtilisateurs(self, nom: int) -> None:
+        """Supprime un utilisateur."""
+        requete = """
+                  DELETE FROM utilisateurs
+                  WHERE nom = ?
+                  """
+        self.requete(requete, nom)
+
+    def verificationIdentifiants(self, nom: str, motDePasse: str) -> bool:
+        """Renvoie vrai ou faux si les identifiants données sont bons."""
+        requete = """
+                  SELECT EXISTS (
+                      SELECT 1 FROM utilisateurs
+                      WHERE nom = ? AND passe = ?
+                  )
+                  """
+        return True if self.affichageResultat(self.requete(requete, [nom, motDePasse]))[0][0] == 1 else False
+
+    def listUtilisateurs(self) -> list:
+        """Retourne la liste des utilisateurs."""
+        requete = """
+                  SELECT nom FROM utilisateurs
+                  """
+        return [i[0] for i in self.affichageResultat(self.requete(requete))]