fusion commande pronos en une fonctionnelle

Author: Oreshout

app/__pycache__/f1api.cpython-313.pyc

@@ -49,15 +49,18 @@ async def helping_tools(interaction: discord.Interaction):
     best_lap="Meilleur tour (uniquement pour Course et Sprint, optionnel pour Qualif)"
 )
 async def submit(interaction: discord.Interaction,
-                 statue: str,
+                  statue: Literal["Course", "Qualif", "Sprint", "Sprint_Qualif"],
                  premier: str,
                  deuxieme: str,
                  troisieme: str,
                  best_lap: str = None):
     await interaction.response.defer(ephemeral=True)
+    STATUSES_VALIDES = ["qualif", "course", "sprint", "sprint_qualif"]
     statue_lower = statue.lower()
-    if statue_lower not in ["Qualif", "Course", "Sprint", "Sprint_Qualif"]:
-        await embed.Error(interaction, "❌ Statue invalide. Choisis parmi : Qualif, Course, Sprint.")
+
+    if statue_lower not in STATUSES_VALIDES:
+        readable = ", ".join([s.capitalize() for s in STATUSES_VALIDES])
+        await embed.Error(interaction, f"❌ Statue invalide. Choisis parmi : {readable}.")
         return
 
     if statue_lower in ["Course", "Sprint"] and not best_lap:
@@ -70,14 +73,13 @@ async def submit(interaction: discord.Interaction,
 
     if command_enabled:
         try:
-            success = pr.pronos_race(
-                interaction.user.id,
-                str(interaction.user),
+            success = pr.pronos_generic(
+                str(interaction.user),  # pseudo
                 premier,
                 deuxieme,
                 troisieme,
-                best_lap,
-                statue_lower
+                statue_lower,
+                best_lap
             )
         except Exception as e:
             logger.exception("Erreur pendant l'enregistrement du prono")
@@ -95,61 +97,6 @@ async def submit(interaction: discord.Interaction,
 # _______________________________________________________________________________________________________________________________
 
 
-@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="pronos_sprint", 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", best_lap="Meilleur Tour")
-async def submit_sprint(interaction: discord.Interaction, premier: str, deuxieme: str, troisieme: str, best_lap: str):
-    await interaction.response.defer(ephemeral=True)
-    if command_enabled:
-        try:
-            success = pr.pronos_sprint(
-                interaction.user.id,
-                str(interaction.user),
-                premier, deuxieme, troisieme, best_lap
-            )
-        except Exception as e:
-            logger.exception("Erreur pendant l'enregistrement du prono sprint")
-            await embed.Error(interaction, f"❌ Erreur pendant l'enregistrement du prono : `{e}`")
-            return
-
-        if success:
-            await interaction.followup.send("✅ Ton prono sprint a bien été pris en compte !", ephemeral=True)
-            logger.info(f'{interaction.user} à fais son pronos course sprint')
-        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")
 async def visu(interaction: discord.Interaction):
     await interaction.response.defer(ephemeral=True)

app/__pycache__/pronos.cpython-313.pyc

@@ -3,74 +3,89 @@
 import pandas as pd
 import json
 from config import logger
+import os
 
+os.makedirs('cache_fastf1', exist_ok=True)
+f1.Cache.enable_cache('cache_fastf1')
 
 def getNextEvent():
     year = datetime.now(timezone.utc).year
     calendar = f1.get_event_schedule(year)
-    time = datetime.now(timezone.utc)
+    now = datetime.now(timezone.utc)
+    
+    print("Colonnes du calendrier :", calendar.columns)  # print des colonnes pour vérifier
+
     for row in calendar.itertuples():
-        if (pd.isna(row.Session4DateUtc)):
-            logger.info("Pas de Date pour la qualif à "+row.Location)
-        elif (time > row.Session4Date):
-            logger.info(row.Location + " est passé (Qualif)")
-        else:
-            logger.info("Prochaine Session Qualif à "+row.Location)
-            session = {
-                "Round": str(row.Index),
-                "Country": row.Country,
-                "Location": row.Location,
-                "Session": 'Q',
-                "Date": row.Session4DateUtc.strftime("%d/%m/%Y,%H:%M:%S"),
-                "Saison": datetime.now(timezone.utc).year
-            }
-            with open('data/Session.json', 'w', encoding='utf-8') as f:
-                json.dump(session, f, ensure_ascii=False, indent=4)
-            break
-        if (pd.isna(row.Session5DateUtc)):
-            logger.info("Pas de Date pour la course à "+row.Location)
-        elif (time > row.Session5Date):
-            logger.info(row.Location + " est passé (Course)")
-        else:
-            logger.info("Prochaine Course à "+row.Location)
-            session = {
-                "Round": row.Index,
-                "Country": row.Country,
-                "Location": row.Location,
-                "Session": 'R',
-                "Date": row.Session5DateUtc.strftime("%d/%m/%Y,%H:%M:%S"),
-                "Saison": datetime.now(timezone.utc).year
-            }
-            with open('data/Session.json', 'w', encoding='utf-8') as f:
-                json.dump(session, f, ensure_ascii=False, indent=4)
-            break
-        if (pd.isna(row.Session6DateUtc)):
-            logger.info("Pas de Date pour la course Sprint à "+row.Location)
-        elif (time > row.Session6Date):
-            logger.info(row.Location + " est passé (Course)")
-        else:
-            logger.info("Prochaine Course Sprint à "+row.Location)
-            session = {
-                "Round": row.Index,
-                "Country": row.Country,
-                "Location": row.Location,
-                "Session": 'S',
-                "Date": row.Session5DateUtc.strftime("%d/%m/%Y,%H:%M:%S"),
-                "Saison": datetime.now(timezone.utc).year
+        round_number = row.Index
+        country = row.Country
+        location = row.Location
+
+        # On parcourt toutes les sessions et on cherche leur nom pour savoir laquelle est le sprint, qualif, etc.
+        sessions = []
+        for i in range(1, 6):  # Session1 à Session5
+            session_name = getattr(row, f'Session{i}', None)
+            session_date = getattr(row, f'Session{i}DateUtc', None)
+
+            if session_name is None or session_date is None:
+                continue
+
+            # On associe les codes aux noms des sessions
+            code = None
+            if "Race" in session_name:
+                code = "R"
+            elif "Sprint" in session_name:
+                code = "S"
+            elif "Qualifying" in session_name:
+                code = "Q"
+            elif "Practice" in session_name:
+                code = "P"
+
+            if code:
+                sessions.append((code, session_date))
+
+        for session_code, session_date in sessions:
+            if pd.isna(session_date):
+                logger.info(f"[{country}] Pas de date pour la session '{session_code}'")
+                continue
+
+            # ✅ Correction ici : forcer la timezone si manquante
+            if session_date.tzinfo is None:
+                session_date = session_date.replace(tzinfo=timezone.utc)
+
+            if now > session_date:
+                logger.info(f"[{country}] Session '{session_code}' déjà passée")
+                continue
+
+            session_data = {
+                "Round": round_number,
+                "Country": country,
+                "Location": location,
+                "Session": session_code,
+                "Date": session_date.strftime("%d/%m/%Y,%H:%M:%S"),
+                "Saison": year
             }
+
+            os.makedirs('data', exist_ok=True)
             with open('data/Session.json', 'w', encoding='utf-8') as f:
-                json.dump(session, f, ensure_ascii=False, indent=4)
-            break
+                json.dump(session_data, f, ensure_ascii=False, indent=4)
 
+            logger.info(f"[{country}] Prochaine session trouvée : {session_code} à {location}")
+            return
 
+    logger.warning("Aucune session future trouvée dans le calendrier.")
+    
 def getResults():
     with open('data/Session.json', 'r', encoding='utf-8') as f:
         data = json.load(f)
 
     country = data.get('Country', 'unknown').lower()
-    session_type = data.get("Session", "")  # garder la casse telle quelle
+    session_type = data.get("Session", "")  # garder la casse
 
-    session = f1.get_session(data['Saison'], data["Location"], data["Session"])
+    try:
+        session = f1.get_session(data['Saison'], data["Round"], session_type)
+    except Exception as e:
+        logger.error(f"Erreur lors de la récupération de la session : {e}")
+        return 5
 
     if not session.f1_api_support:
         logger.info("L'API FastF1 n'est pas disponible pour cette session.")
@@ -86,23 +101,27 @@ def getResults():
         logger.warning("Les résultats ne sont pas encore disponibles.")
         return 1
 
+    if len(session.results) < 3:
+        logger.warning("Pas assez de résultats pour un top 3.")
+        return 1
+
     try:
         number = session.laps.pick_fastest().DriverNumber
-        row = session.results.loc[session.results.DriverNumber == str(number)]
-        driver = row.FullName.values[0]
+        row = session.results[session.results.DriverNumber == str(number)]
+        driver = row.FullName.values[0] if not row.empty else "Inconnu"
 
         result = {
             "1": session.results.FullName.iloc[0],
             "2": session.results.FullName.iloc[1],
             "3": session.results.FullName.iloc[2],
-            "Best Lap": driver
+            "Best Lap": driver,
+            "timestamp": datetime.now().isoformat()
         }
 
-    except (ValueError, IndexError) as e:
+    except Exception as e:
         logger.warning(f"Erreur dans le traitement des résultats : {e}")
         return 1
 
-    # Choix du fichier selon le type exact de session
     if session_type == "Q":
         filename = f'data/Results_Qualif_{country}.json'
     elif session_type == "S":