🆕 Ajout de fonction 'Test sondes'.py

domo91.py

@@ -11,6 +11,11 @@ import random
 import datetime
 
 st.set_page_config(page_title="Domo91 - Surveillance", layout="wide")
+if "authenticated" not in st.session_state:
+    st.session_state["authenticated"] = False
+    st.session_state["role"] = None
+    st.session_state["lieu_autorise"] = None
+
 st.title("📡 Supervision Températures")
 
 # --- Configuration base de données ---
@@ -100,52 +105,85 @@ def generer_pdf(site, date_str):
 
     except Exception as e:
         st.error(f"Erreur lors de la génération du PDF : {e}")
-# --- Initialisation session ---
+# --- Initialisation des variables de session ---
 if "authenticated" not in st.session_state:
     st.session_state["authenticated"] = False
+if "role" not in st.session_state:
     st.session_state["role"] = None
+if "lieu_autorise" not in st.session_state:
     st.session_state["lieu_autorise"] = None
 
 # --- Sidebar (connexion + bouton PDF) ---
-with st.sidebar:
-    if st.sidebar.button("🔔 Forcer une alerte de test"):
-        now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        sonde_test = "TEST_ALERTE"
+
+# 🔐 Connexion (toujours affichée)
+st.sidebar.header("🔐 Connexion")
+if not st.session_state.get("authenticated"):
+    login = st.sidebar.text_input("Nom d'utilisateur")
+    password = st.sidebar.text_input("Mot de passe", type="password")
+    if st.sidebar.button("Se connecter"):
         try:
-            conn = mysql.connector.connect(**db_params)
-            cursor = conn.cursor()
-            table_alertes = f"Alertes_{site}"
-            cursor.execute(f"""
-                INSERT INTO {table_alertes} (Sonde, Debut_defaut, Etat)
-                VALUES (%s, %s, %s)
-            """, (sonde_test, now, "Test"))
-            conn.commit()
+            conn = mysql.connector.connect(**db_config)
+            cursor = conn.cursor(dictionary=True)
+            cursor.execute("SELECT * FROM MotsDePasse WHERE utilisateur = %s", (login,))
+            result = cursor.fetchone()
+            if result and result["mot_de_passe"] == password:
+                st.session_state["authenticated"] = True
+                st.session_state["role"] = result["role"]
+                st.session_state["lieu_autorise"] = result["Lieu"]
+                st.success(f"Connecté comme {result['role']} ({result['Lieu']})")
+            else:
+                st.sidebar.error("Identifiants invalides")
             cursor.close()
             conn.close()
-            st.success(f"Alerte de test créée pour {sonde_test} à {now}")
         except Exception as e:
-            st.error(f"Erreur lors de la création de l'alerte : {e}")
-    st.header("🔐 Connexion")
-    if not st.session_state.get("authenticated"):
-        login = st.text_input("Nom d'utilisateur")
-        password = st.text_input("Mot de passe", type="password")
-        if st.button("Se connecter"):
-            try:
-                conn = mysql.connector.connect(**db_config)
-                cursor = conn.cursor(dictionary=True)
-                cursor.execute("SELECT * FROM MotsDePasse WHERE utilisateur = %s", (login,))
-                result = cursor.fetchone()
-                if result and result["mot_de_passe"] == password:
-                    st.session_state["authenticated"] = True
-                    st.session_state["role"] = result["role"]
-                    st.session_state["lieu_autorise"] = result["Lieu"]
-                    st.success(f"Connecté comme {result['role']} ({result['Lieu']})")
-                else:
-                    st.error("Identifiants invalides")
-                cursor.close()
-                conn.close()
-            except Exception as e:
-                st.error(f"Erreur lors de la connexion à la base : {e}")
+            st.sidebar.error(f"Erreur lors de la connexion à la base : {e}")
+else:
+    st.sidebar.success(f"Connecté ({st.session_state['role']})")
+    if st.sidebar.button("🔓 Déconnexion"):
+        st.session_state["authenticated"] = False
+        st.session_state["role"] = None
+        st.session_state["lieu_autorise"] = None
+        st.rerun()
+
+# 🔔 Forcer une alerte de test dynamique avec choix de la sonde
+if st.session_state.get("authenticated"):
+    site_actuel = (
+        st.session_state.get("lieu_autorise")
+        if st.session_state.get("role") != "superviseur"
+        else st.session_state.get("selected_site")
+    )
+    if site_actuel:
+        try:
+            conn = mysql.connector.connect(**db_config)
+            cursor = conn.cursor()
+            cursor.execute(f"SELECT DISTINCT Sonde FROM `{site_actuel}` ORDER BY Sonde ASC")
+            sondes_dispo = [row[0] for row in cursor.fetchall()]
+            cursor.close()
+            conn.close()
+
+            if sondes_dispo:
+                st.sidebar.markdown("---")
+                st.sidebar.subheader("🧪 Test alerte manuelle")
+                sonde_test = st.sidebar.selectbox("Choisir une sonde :", sondes_dispo)
+                if st.sidebar.button("🔔 Forcer une alerte de test"):
+                    now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                    try:
+                        conn = mysql.connector.connect(**db_config)
+                        cursor = conn.cursor()
+                        table_alertes = f"Alertes_{site_actuel}"
+                        cursor.execute(f"""
+                            INSERT INTO {table_alertes} (Sonde, Debut_defaut, Status)
+                            VALUES (%s, %s, %s)
+                        """, (sonde_test, now, "Test"))
+                        conn.commit()
+                        cursor.close()
+                        conn.close()
+                        st.success(f"Alerte de test créée pour {sonde_test} à {now}")
+                    except Exception as e:
+                        st.error(f"Erreur lors de la création de l'alerte : {e}")
+        except Exception as e:
+            st.sidebar.warning(f"Erreur chargement des sondes : {e}")
+
     else:
         st.success(f"Connecté ({st.session_state['role']})")
         if st.button("🔓 Déconnexion"):
@@ -173,6 +211,7 @@ if st.session_state["authenticated"]:
         sites_possibles = ["Saclay", "Meudon"]
         if st.session_state["role"] == "superviseur":
             site_selectionne = st.selectbox("📍 Choisissez un site :", sites_possibles)
+            st.session_state["selected_site"] = site_selectionne
         else:
             site_selectionne = st.session_state["lieu_autorise"]
             st.info(f"Site imposé : {site_selectionne}")
@@ -281,7 +320,7 @@ if st.session_state["role"] == "superviseur":
                     cursor_add = conn_add.cursor()
 
                     cursor_add.execute(
-                        "INSERT INTO Chambres_froides (Lieu, Sonde, Temp_Max, Etat) VALUES (%s, %s, %s, %s)",
+                        "INSERT INTO Chambres_froides (Lieu, Sonde, Temp_Max, Status) VALUES (%s, %s, %s, %s)",
                         (site_selectionne, nouvelle_sonde, temp_max, "ON" if etat_on else "OFF")
                     )
                     conn_add.commit()