from random import randint, uniform from db import BaseDeDonnees class Stock(BaseDeDonnees): """Gère une table "stock" pour une base de donnée donné.""" def __init__(self): super().__init__(r"db.sqlite3") def creationTable(self) -> None: """Créer la table qui stocker les stocks.""" requete = """ CREATE TABLE IF NOT EXISTS stocks ( id INTEGER PRIMARY KEY, type TEXT, nom TEXT, quantite INTEGER, prix REAL, image_url TEXT ); """ self.requete(requete) # Ajout d'un stock par défaut si aucun stock n'existe dans la base de donnée if len(self.listeStocks()) == 0: # Créer un dictionnaire d'éléments pour mieux voir ce que l'on ajoute à la base de donnée defaut = { "fruits legumes": [ ("banane", "img/banane.gif"), ("orange", "img/orange.gif"), ("betterave", "img/betterave.gif"), ("carottes", "img/carottes.gif"), ("tomates", "img/tomates.gif"), ("citron", "img/citron.gif"), ("kiwi", "img/kiwi.gif"), ("clementine", "img/clementine.gif"), ("pomme", "img/pomme.gif"), ("avocat", "img/avocat.gif") ], "boulangerie": [ ("brownie", "img/brownie.gif"), ("baguette", "img/baguette.gif"), ("pain au chocolat", "img/pain_au_chocolat.gif"), ("croissant", "img/croissant.gif"), ("macaron", "img/macaron.gif"), ("millefeuille", "img/millefeuille.gif"), ("paris-brest", "img/paris-brest.gif"), ("opera", "img/opera.gif"), ("fraisier", "img/fraisier.gif"), ("eclair", "img/eclair.gif") ], "boucherie poissonnerie": [ ("saucisson", "img/saucisson.gif"), ("côte de boeuf", "img/cote_de_boeuf.gif"), ("langue de boeuf", "img/langue_de_boeuf.gif"), ("collier de boeuf", "img/collier_de_boeuf.gif"), ("entrecote", "img/entrecote.gif"), ("cabillaud", "img/cabillaud.gif"), ("saumon", "img/saumon.gif"), ("colin", "img/colin.gif"), ("bar", "img/bar.gif"), ("dorade", "img/dorade.gif") ], "entretien": [ ("nettoyant air comprimé", "img/nettoyant_air_comprime.gif"), ("nettoyage anti-bactérien", "img/nettoyage_anti-bacterien.gif"), ("nettoyant pour écran", "img/nettoyant_pour_ecran.gif"), ("nettoyant pour lunettes", "img/nettoyant_pour_lunettes.gif"), ("pioche", "img/pioche.gif"), ("pelle", "img/pelle.gif"), ("lampe torche", "img/lampe_torche.gif"), ("gants", "img/gants.gif"), ("éponge", "img/eponge.gif"), ("essuie-tout", "img/essuie-tout.gif") ] } # Ajoute le dictionnaire précédemment créer à la base de donnée avec un prix et une quantité aléatoire for type in defaut: for element in defaut[type]: self.ajoutStock(type, element[0], randint(0, 10), round(uniform(2., 30.), 2), element[1]) def ajoutStock(self, type: str, nom: str, quantite: int, prix: float, imageURL: str) -> list: """Ajoute un élément dans le stock et retourne l'ID de ce dernier.""" requete = """ INSERT INTO stocks ( type, nom, quantite, prix, image_url ) VALUES ( ?, ?, ?, ?, ? ); """ self.requete(requete, [type.lower(), nom.lower(), quantite, prix, imageURL]) return self.affichageResultat(self.requete("SELECT last_insert_rowid();")) def suppressionStocks(self, id: int) -> None: """Supprime un stock.""" requete = """ DELETE FROM stocks WHERE id = ? """ self.requete(requete, id) def listeStocks(self) -> list: """Retourne la liste des éléments en stock sous forme de dictionnaire.""" recuperation = [ "id", "type", "nom", "quantite", "prix", "image_url" ] requete = f""" SELECT {", ".join(recuperation)} FROM stocks """ return [dict(zip(recuperation, element)) for element in self.affichageResultat(self.requete(requete))] def stockExistant(self, stock: str) -> bool: """Vérifie si le stock donnée existe déjà dans la base de donnée.""" requete = """ SELECT EXISTS ( SELECT 1 FROM stocks WHERE nom = ? ) """ return True if self.affichageResultat(self.requete(requete, stock.lower()))[0][0] == 1 else False def listeTypes(self) -> list: """Renvoie la liste des types disponibles dans la base de donnée.""" requete = """ SELECT type FROM stocks """ res = [] for i in self.affichageResultat(self.requete(requete)): if i[0] not in res: res.append(i[0]) return res