Mise a jour des pronos et des fichiers + visualisation

Author: Oreshout

app/__pycache__/admin_command.cpython-313.pyc

@@ -8,7 +8,7 @@
 import pronos as pr
 import json
 import f1api
-from datetime import timedelta,datetime
+from datetime import timedelta
 
 
 @bot.event
@@ -63,19 +63,54 @@ async def helping_tools(interaction: discord.Interaction):
 @app_commands.describe(premier="Le premier", deuxieme="Le deuxième", troisieme="Le troisième", best_lap="Meilleur Tour")
 async def submit(interaction: discord.Interaction, premier: str, deuxieme: str, troisieme: str, best_lap: str):
     await interaction.response.defer(ephemeral=True)
-    if (command_enabled):
+    if command_enabled:
+        try:
+            success = pr.pronos(
+                interaction.user.id,
+                str(interaction.user),
+                premier, deuxieme, troisieme, best_lap
+            )
+        except Exception as e:
+            logger.exception("Erreur pendant l'enregistrement du prono qualif")
+            await embed.Error(interaction, f"❌ Erreur pendant l'enregistrement du prono : `{e}`")
+            return
 
-        if(pr.pronos(interaction.user.id, str(interaction.user),
-                  premier, deuxieme, troisieme, best_lap)):
-             await interaction.followup.send("Ton prono a bien été pris en compte", ephemeral=True)
+        if success:
+            await interaction.followup.send("✅ Ton prono qualif a bien été pris en compte !", ephemeral=True)
+            logger.info(f'{interaction.user} à fais son pronos course')
         else:
-            await embed.Error(interaction, "Tu ne peux modifier ton pronostic qu'une fois")
-            
-       
+            await embed.Error(interaction, "❌ Tu ne peux modifier ton pronostic qu'une seule fois.")
+    else:
+        await embed.Error(interaction, "Il y a une heure pour tout faire, et celle ci n'est pas pour les pronos. Par conséquent ton prono n'a pas pu être enregistré. Si tu veux être notifié des prochaines sessions, utilise /role")
+
+# _______________________________________________________________________________________________________________________________
 
+@tree.command(name="pronos_qualif", description="Enregistre tes pronos ou modifie les si tu l'a déja fait par le passé(max 1 fois)")
+@app_commands.describe(premier="Le premier", deuxieme="Le deuxième", troisieme="Le troisième")
+async def submit_qualif(interaction: discord.Interaction, premier: str, deuxieme: str, troisieme: str):
+    await interaction.response.defer(ephemeral=True)
+
+    if command_enabled:
+        try:
+            success = pr.pronos_qualif(
+                interaction.user.id,
+                str(interaction.user),
+                premier, deuxieme, troisieme
+            )
+        except Exception as e:
+            logger.exception("Erreur pendant l'enregistrement du prono qualif")
+            await embed.Error(interaction, f"❌ Erreur pendant l'enregistrement du prono : `{e}`")
+            return
+
+        if success:
+            await interaction.followup.send("✅ Ton prono qualif a bien été pris en compte !", ephemeral=True)
+            logger.info(f'{interaction.user} à fais son pronos qualif')
+        else:
+            await embed.Error(interaction, "❌ Tu ne peux modifier ton pronostic qu'une seule fois.")
     else:
         await embed.Error(interaction, "Il y a une heure pour tout faire, et celle ci n'est pas pour les pronos. Par conséquent ton prono n'a pas pu être enregistré. Si tu veux être notifié des prochaines sessions, utilise /role")
 
+
 # _______________________________________________________________________________________________________________________________
 
 @tree.command(name="visualisation", description="Te montre tes pronos")

app/__pycache__/classement.cpython-313.pyc

@@ -1,12 +1,36 @@
 # Gestion des pronostiques, écriture et lecture
 import json
 import os
+import re
 import discord
+import fastf1 as f1
+from datetime import datetime, timezone
+import pandas as pd
 from config import logger
+import tools as tool
 
+def sanitize_filename(name):
+    return re.sub(r'[^\w\-]', '_', name)
+
+def country_fonction():
+    year = datetime.now(timezone.utc).year
+    calendar = f1.get_event_schedule(year)
+    now = datetime.now(timezone.utc)
+
+    for row in calendar.itertuples():
+        if pd.isna(row.Session5DateUtc):
+            continue
+        elif now <= row.Session5DateUtc.replace(tzinfo=timezone.utc):
+            return sanitize_filename(row.Country) 
+
+    return "Unknown"  # fallback par défaut si rien trouvé
+        
 
 def pronos(id: int, pseudo: str, premier: str, second: str, troisieme: str, bt: str):
-    file_path = 'data/Pronos.json'
+    
+    country = country_fonction()
+    file_path = f'data/pronos_{country}.json'
+
     if not os.path.exists(file_path):
         pronos_database = {
             str(id): {
@@ -31,7 +55,7 @@ def pronos(id: int, pseudo: str, premier: str, second: str, troisieme: str, bt:
                     "Modif":False
                 }
             else:
-                if(pronos_database[str(id)]["Modif"] == True):
+                if pronos_database[str(id)]["Modif"]:
                     return False
                 else:
 
@@ -49,61 +73,148 @@ def pronos(id: int, pseudo: str, premier: str, second: str, troisieme: str, bt:
 
 async def visualisation(interaction: discord.Interaction):
 
-    file_path = "data/Pronos.json"
-    with open(file_path, 'r', encoding='utf-8') as f:
+    try:
+        country = country_fonction()
+        file_path_course = f'data/pronos_{country}.json'
+        file_path_qualif = f'data/pronos_{country}_qualifs.json'
+        
+        tool.ensure_file_exists(file_path_course)
+        tool.ensure_file_exists(file_path_qualif)
+
+        with open(file_path_course, 'r', encoding='utf-8') as f:
             pronos_database = json.load(f)
-    
-    user = interaction.user.id 
-    
-    if str(user) in pronos_database:
-        premier=pronos_database[str(user)]["1"]
-        deuxieme=pronos_database[str(user)]["2"]
-        troisieme=pronos_database[str(user)]["3"]
-        best_lap=pronos_database[str(user)]["Best Lap"]
-        modif=pronos_database[str(user)]["Modif"]
-    
+
+        with open(file_path_qualif, 'r', encoding='utf-8') as f:
+            pronos_database_qualif = json.load(f)
+
+    except Exception as e:
+        await interaction.followup.send(
+            content=f"❌ Une erreur est survenue lors du chargement des fichiers : {e}",
+            ephemeral=True
+        )
+        return
+
+    user_id = str(interaction.user.id)
+
+    # Protection si le JSON est vide ou corrompu
+    if not isinstance(pronos_database, dict):
+        pronos_database = {}
+
+    if not isinstance(pronos_database_qualif, dict):
+        pronos_database_qualif = {}
+
+    # Cas : seulement prono course
+    if user_id in pronos_database and user_id not in pronos_database_qualif:
+        prono = pronos_database[user_id]
         embed = discord.Embed(
-            title = f"🐐 Merci pour vos pronos {interaction.user} !",
-            description="Voici tes pronostiques : ",
+            title=f"🐐 Merci pour vos pronos {interaction.user} !",
             color=discord.Color.red()
         )
-        
-        embed.add_field(name="Ton Premier 🥇 :", value=f"{premier}", inline=False)
-        embed.add_field(name="Ton Deuxième 🥈 :", value=f"{deuxieme}", inline=False)
-        embed.add_field(name="Ton Troisième 🥉 :", value=f"{troisieme}", inline=False)
-        embed.add_field(name="Ton Meilleur Tour ⏱️ :", value=f"{best_lap}", inline=False) 
-        if(modif):
-            value="❌ Tu as déja modifié ton pronostic par le passé"
-        else:
-            value="✅ Tu peux modifier ton pronostic en relançant /pronos_course" 
-        embed.add_field(name="Droit de modification :",value=value,inline=False) 
-       
+        embed.description = "Voici tes pronostics de course :"
+        embed.add_field(name="Ton Premier 🥇 :", value=prono.get("1", "Non renseigné"), inline=False)
+        embed.add_field(name="Ton Deuxième 🥈 :", value=prono.get("2", "Non renseigné"), inline=False)
+        embed.add_field(name="Ton Troisième 🥉 :", value=prono.get("3", "Non renseigné"), inline=False)
+        embed.add_field(name="Ton Meilleur Tour ⏱️ :", value=prono.get("Best Lap", "Non renseigné"), inline=False)
 
-        embed.set_footer(text="Bot créé par F1F Team", icon_url="https://cdn.discordapp.com/attachments/1339299411360088226/1367477935392428083/Votre_texte_de_paragraphe_12.png")
-        embed.set_thumbnail(url=interaction.user.display_avatar.url)
-        embed.set_image(url="https://cdn.discordapp.com/attachments/1339299411360088226/1368544420504272987/Design_sans_titre_2.png")
-        
+        modif_text = "❌ Tu as déjà modifié ton pronostic par le passé" if prono.get("Modif", True) else "✅ Tu peux modifier ton pronostic en relançant /pronos_course"
+        embed.add_field(name="Droit de modification :", value=modif_text, inline=False)
 
-        logger.info(f"{interaction.user.name} a regarder ses pronos course")
-        
-        await interaction.followup.send(embed=embed, ephemeral=True)  
-        
-        
-        
+    # Cas : seulement prono qualif
+    elif user_id in pronos_database_qualif and user_id not in pronos_database:
+        prono = pronos_database_qualif[user_id]
+        embed = discord.Embed(
+            title=f"🐐 Merci pour vos pronos {interaction.user} !",
+            color=discord.Color.red()
+        )
+        embed.description = "Voici tes pronostics de qualif :"
+        embed.add_field(name="Ton Premier 🥇 :", value=prono.get("1", "Non renseigné"), inline=False)
+        embed.add_field(name="Ton Deuxième 🥈 :", value=prono.get("2", "Non renseigné"), inline=False)
+        embed.add_field(name="Ton Troisième 🥉 :", value=prono.get("3", "Non renseigné"), inline=False)
+
+        modif_text = "❌ Tu as déjà modifié ton pronostic par le passé" if prono.get("Modif", True) else "✅ Tu peux modifier ton pronostic en relançant /pronos_qualif"
+        embed.add_field(name="Droit de modification :", value=modif_text, inline=False)
+
+    # Cas : les deux pronos existent
+    elif user_id in pronos_database and user_id in pronos_database_qualif:
+        pr_course = pronos_database[user_id]
+        pr_qualif = pronos_database_qualif[user_id]
+        embed = discord.Embed(
+            title=f"🐐 Merci pour vos pronos {interaction.user} !",
+            color=discord.Color.red()
+        )
+
+        embed.description = "Voici tes pronostics complets :"
+
+        embed.add_field(name="**Qualif - Premier 🥇**", value=pr_qualif.get("1", "Non renseigné"), inline=False)
+        embed.add_field(name="**Qualif - Deuxième 🥈**", value=pr_qualif.get("2", "Non renseigné"), inline=False)
+        embed.add_field(name="**Qualif - Troisième 🥉**", value=pr_qualif.get("3", "Non renseigné"), inline=False)
+
+        embed.add_field(name="**Course - Premier 🥇**", value=pr_course.get("1", "Non renseigné"), inline=False)
+        embed.add_field(name="**Course - Deuxième 🥈**", value=pr_course.get("2", "Non renseigné"), inline=False)
+        embed.add_field(name="**Course - Troisième 🥉**", value=pr_course.get("3", "Non renseigné"), inline=False)
+        embed.add_field(name="**Meilleur Tour ⏱️**", value=pr_course.get("Best Lap", "Non renseigné"), inline=False)
+
+        modif_q = "❌ Qualif déjà modifié" if pr_qualif.get("Modif", True) else "✅ Tu peux modifier ton prono Qualif (/pronos_qualif)"
+        modif_c = "❌ Course déjà modifié" if pr_course.get("Modif", True) else "✅ Tu peux modifier ton prono Course (/pronos_course)"
+        embed.add_field(name="Droits de modification :", value=f"{modif_q}\n{modif_c}", inline=False)
+
+    # Cas : aucun prono
     else:
         embed = discord.Embed(
-            title = f"Désolé {interaction.user} !",
-            description="On dirait que tu n'as pas encore fait de pronostique",
+            title=f"Désolé {interaction.user} !",
+            description="On dirait que tu n'as pas encore fait de pronostic.",
             color=discord.Color.red()
         )
-        
-        embed.set_footer(text="Bot créé par F1F Team", icon_url="https://cdn.discordapp.com/attachments/1339299411360088226/1367477935392428083/Votre_texte_de_paragraphe_12.png")
-        embed.set_thumbnail(url=interaction.user.display_avatar.url)
-        embed.set_image(url="https://cdn.discordapp.com/attachments/1339299411360088226/1368544420504272987/Design_sans_titre_2.png")
-        logger.info(f"{interaction.user.name} a tenter de regarder ses pronos course alors qu'il n'en avais pas fais")
-        await interaction.followup.send(embed=embed, ephemeral=True)
-        
-        
+
+    # Envoi final
+    await interaction.followup.send(embed=embed, ephemeral=True)
+    logger.info(f"{interaction.user} à visualisé ses pronos.")
+
+
+def pronos_qualif(id: int, pseudo: str, premier: str, second: str, troisieme: str):
+    
+    country = country_fonction()
+    file_path = f'data/pronos_{country}_qualifs.json'
+
+    if not os.path.exists(file_path):
+        pronos_database = {
+            str(id): {
+                "Pseudo": pseudo,
+                "1": premier,
+                "2": second,
+                "3": troisieme,
+                "Modif":False
+            }
+        }
+    else:
+        with open(file_path, 'r', encoding='utf-8') as f:
+            pronos_database = json.load(f)
+            if str(id) not in pronos_database:
+                pronos_database[str(id)] = {
+                    "Pseudo": pseudo,
+                    "1": premier,
+                    "2": second,
+                    "3": troisieme,
+                    "Modif":False
+                }
+            else:
+                if pronos_database[str(id)]["Modif"]:
+                    return False
+                else:
+
+                    pronos_database[str(id)]["Pseudo"] = pseudo
+                    pronos_database[str(id)]["1"] = premier
+                    pronos_database[str(id)]["2"] = second
+                    pronos_database[str(id)]["3"] = troisieme
+                    pronos_database[str(id)]["Modif"] = True
+
+
+    with open(file_path, 'w', encoding='utf-8') as f:
+        json.dump(pronos_database, f, ensure_ascii=False, indent=4)
+    return True
+
+
+       
 
 
 

app/__pycache__/config.cpython-313.pyc

@@ -8,6 +8,10 @@
 import json
 from error_embed import info_embed, no_prono
 
+def ensure_file_exists(path):
+    if not os.path.exists(path):
+        with open(path, 'w', encoding='utf-8') as f:
+            json.dump({}, f)
 
 
 async def help(interaction: discord.Interaction):