106 lines
4.3 KiB
Python
106 lines
4.3 KiB
Python
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 (
|
|
id INTEGER PRIMARY KEY,
|
|
pseudo TEXT,
|
|
passe TEXT,
|
|
metier INTEGER,
|
|
nom TEXT,
|
|
prenom TEXT,
|
|
naissance TEXT,
|
|
adresse TEXT,
|
|
postal INTEGER
|
|
);
|
|
"""
|
|
self.requete(requete)
|
|
# Ajout d'un utilisateur par défaut si aucun utilisateur n'existe dans la base de donnée
|
|
if len(self.listUtilisateurs()) == 0:
|
|
self.ajoutUtilisateurs(
|
|
pseudo="admin",
|
|
passe="P@ssword",
|
|
metier=0,
|
|
nom="Admin",
|
|
prenom="Système",
|
|
naissance="2000/10/09",
|
|
adresse="12 Rue de Montmartre",
|
|
postal=46800
|
|
)
|
|
|
|
def ajoutUtilisateurs(self, pseudo: str, passe: str, metier: int, nom: str, prenom: str, naissance: str, adresse: str, postal: str) -> list:
|
|
"""Ajoute un utilisateur et retourne l'ID de ce dernier."""
|
|
requete = """
|
|
INSERT INTO utilisateurs (
|
|
pseudo, passe, metier, nom, prenom, naissance, adresse, postal
|
|
) VALUES (
|
|
?, ?, ?, ?, ?, ?, ?, ?
|
|
);
|
|
"""
|
|
self.requete(requete, [pseudo, passe, metier, nom, prenom, naissance, adresse, postal])
|
|
return self.affichageResultat(self.requete("SELECT last_insert_rowid();"))
|
|
|
|
def suppressionUtilisateurs(self, pseudo: str) -> None:
|
|
"""Supprime un utilisateur."""
|
|
requete = """
|
|
DELETE FROM utilisateurs
|
|
WHERE pseudo = ?
|
|
"""
|
|
self.requete(requete, pseudo)
|
|
|
|
def verificationIdentifiants(self, pseudo: str, motDePasse: str):
|
|
"""
|
|
Retourne l'ID de l'utilisateur si trouvé dans la base de donnée ainsi
|
|
que son métier (`tuple`), sinon renvoie `(0,)`.
|
|
"""
|
|
requete = """
|
|
SELECT id, metier FROM utilisateurs
|
|
WHERE pseudo = ? AND passe = ?
|
|
"""
|
|
reponseBaseDeDonnee = self.affichageResultat(self.requete(requete, [pseudo, motDePasse]))
|
|
if len(reponseBaseDeDonnee) == 0: # si les identifiants renseignés sont mauvais
|
|
return (0,)
|
|
return reponseBaseDeDonnee[0]
|
|
|
|
def listUtilisateurs(self) -> list:
|
|
"""Retourne la liste des nom d'utilisateurs."""
|
|
requete = """
|
|
SELECT pseudo FROM utilisateurs
|
|
"""
|
|
# i[0] parce que sinon ça renvoie des Tuple qui ressemble à ça : `(Utilisateur,)`
|
|
return [i[0] for i in self.affichageResultat(self.requete(requete))]
|
|
|
|
def recuperationUtilisateur(self, id: int = None, pseudo: str = None) -> dict:
|
|
"""Retourne les informations d'un utilisateur grâce à son ID ou son pseudo (ID est priorité)."""
|
|
recuperation = [
|
|
"pseudo",
|
|
"passe",
|
|
"metier",
|
|
"nom",
|
|
"prenom",
|
|
"naissance",
|
|
"adresse",
|
|
"postal"
|
|
]
|
|
if not id: # si la variable `id` n'est pas définie
|
|
if not pseudo: # si la variable `pseudo` n'est pas définie
|
|
raise ValueError # Aucun utilisateur renseigné
|
|
else: # si un pseudo est renseigné, c'est ce qu'on va utilisé
|
|
requete = f"""
|
|
SELECT {", ".join(recuperation)} FROM utilisateurs
|
|
WHERE pseudo = ?
|
|
"""
|
|
utilisateur = pseudo
|
|
else: # si un id est renseigné, c'est ce qu'on va utilisé
|
|
requete = f"""
|
|
SELECT {", ".join(recuperation)} FROM utilisateurs
|
|
WHERE id = ?
|
|
"""
|
|
utilisateur = id
|
|
return self.affichageResultatDictionnaire(recuperation, self.requete(requete, utilisateur))
|