From 8e4a4dc014c3503faac1b66be37ff6cd9fc9b9e2 Mon Sep 17 00:00:00 2001 From: Mylloon Date: Mon, 29 Nov 2021 21:01:00 +0100 Subject: [PATCH] allow forcing the creation of the global csv file --- stats.py | 21 +++++++++++++++------ 1 file changed, 15 insertions(+), 6 deletions(-) diff --git a/stats.py b/stats.py index bb37f57..53cbda8 100644 --- a/stats.py +++ b/stats.py @@ -6,13 +6,18 @@ from users import Utilisateurs class Stats(Utilisateurs): """Gère les statistiques et son export en format CSV.""" - def creationCSV(self): - """Créer le fichier `CSV` qui stockera les statistiques pour tous les utilisateurs.""" - if not self.fichierExiste("stats.csv"): + def creationCSV(self, force: bool = False): + """ + Créer le fichier `CSV` qui stockera les statistiques pour tous les utilisateurs. + + Possibilité de forcer la création (càd même si le fichier existe déjà) en renseignant + `force = True` + """ + if not self.fichierExiste("stats.csv") or force: entete = ["id", "pseudo"] dateAujourdHui = date.today() for _ in range(0, 8): - entete.append(dateAujourdHui) + entete.append(dateAujourdHui.strftime("%Y/%m/%d")) dateAujourdHui = dateAujourdHui - timedelta(days=1) with open("stats.csv", 'w') as f: @@ -33,6 +38,9 @@ class Stats(Utilisateurs): Procéder normalement consiste à ajouter le prix au prix totale stocké dans la base de donnée. """ + self.miseAJourDatesCSV() + aujourdHui = date.today().strftime("%Y/%m/%d") + print(prix) pass @@ -56,5 +64,6 @@ class Stats(Utilisateurs): def miseAJourDatesCSV(self): """Mets-à-jour les dates trop anciennes du fichier globales `CSV`.""" with open("stats.csv", 'r') as f: - fichier = csv.DictReader(f) - print(fichier) + fichier = list(csv.DictReader(f)) + if fichier == []: # si fichier vide + self.creationCSV(True) # on recréer le fichier dans le doute