Remise en état Technique

utils/db.py

@@ -0,0 +1,101 @@
+# utils/db.py
+
+import mysql.connector
+import os
+import sys
+import logging
+from datetime import datetime
+
+
+def connect_to_mysql(db_name: str = None):
+    """Connexion à MySQL via variables d'environnement."""
+    try:
+        mydb = mysql.connector.connect(
+            host=os.getenv("DB_HOST"),
+            user=os.getenv("DB_USER"),
+            password=os.getenv("DB_PASSWORD"),
+            database=db_name if db_name else os.getenv("DB_NAME")
+        )
+        return mydb
+    except mysql.connector.Error as err:
+        logging.error(f"Erreur de connexion MySQL : {err}")
+        sys.exit(1)
+
+
+def get_latest_chaufferie():
+    """Renvoie les dernières valeurs par sonde dans la table 'Chaufferie'."""
+    db = connect_to_mysql()
+    cursor = db.cursor(dictionary=True)
+    query = """
+        SELECT c1.*
+        FROM Chaufferie c1
+        INNER JOIN (
+            SELECT Sonde, MAX(Date) AS MaxDate
+            FROM Chaufferie
+            GROUP BY Sonde
+        ) c2 ON c1.Sonde = c2.Sonde AND c1.Date = c2.MaxDate
+        ORDER BY c1.Sonde;
+    """
+    cursor.execute(query)
+    result = cursor.fetchall()
+    cursor.close()
+    db.close()
+    return result
+
+
+def get_history_by_sonde(sonde: str, start: datetime, end: datetime):
+    """Retourne l’historique des températures d’une sonde entre deux dates."""
+    db = connect_to_mysql()
+    cursor = db.cursor(dictionary=True)
+    query = """
+        SELECT * FROM Chaufferie
+        WHERE Sonde = %s AND Date BETWEEN %s AND %s
+        ORDER BY Date;
+    """
+    cursor.execute(query, (sonde, start, end))
+    result = cursor.fetchall()
+    cursor.close()
+    db.close()
+    return result
+
+
+def verifier_utilisateur_commun(utilisateur: str, motdepasse: str):
+    """Vérifie si un utilisateur (non superviseur) existe dans la table MotsDePasse."""
+    db = connect_to_mysql()
+    cursor = db.cursor(dictionary=True)
+    query = """
+        SELECT * FROM MotsDePasse
+        WHERE utilisateur = %s AND motdepasse = %s AND role = 'utilisateur'
+    """
+    cursor.execute(query, (utilisateur, motdepasse))
+    result = cursor.fetchone()
+    cursor.close()
+    db.close()
+    return result
+
+
+def lire_alertes_sondes():
+    """Renvoie la liste des alertes non acquittées dans la table Alertes_Chaufferie."""
+    db = connect_to_mysql()
+    cursor = db.cursor(dictionary=True)
+    query = """
+        SELECT * FROM Alertes_Chaufferie
+        WHERE Etat != 'Acquitté'
+        ORDER BY Debut_defaut DESC
+    """
+    cursor.execute(query)
+    result = cursor.fetchall()
+    cursor.close()
+    db.close()
+    return result
+
+
+def acquitter_alerte(id_alerte: int):
+    """Met à jour une alerte comme acquittée dans la base."""
+    db = connect_to_mysql()
+    cursor = db.cursor()
+    query = "UPDATE Alertes_Chaufferie SET Etat = 'Acquitté' WHERE Id = %s"
+    cursor.execute(query, (id_alerte,))
+    db.commit()
+    cursor.close()
+    db.close()