Remise en état des fichiers

app/mqtt_logger.py

@@ -0,0 +1,48 @@
+import argparse
+import paho.mqtt.client as mqtt_client
+from dotenv import load_dotenv
+import logging
+from app.logger_config import setup_logger
+from utils_db import connect_to_mysql
+from functools import partial
+
+def on_message(table_sql, _client, _userdata, msg):
+    try:
+        logging.info(f"Message reçu sur {msg.topic}: {msg.payload.decode()}")
+        cursor = mydb.cursor()
+        sonde_name = '/'.join(msg.topic.split('/')[1:])
+        sql = f"INSERT INTO {table_sql} (Sonde, Temperature) VALUES (%s, %s)"
+        val = (sonde_name, msg.payload.decode())
+        cursor.execute(sql, val)
+        mydb.commit()
+        logging.info(f"Insertion réussie : {val}")
+    except Exception as e:
+        logging.error(f"Erreur lors de l'insertion du message : {e}")
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--log", required=True, help="Nom du fichier de log")
+    parser.add_argument("--table", required=True, help="Nom complet de la table SQL")
+    parser.add_argument("--topic", required=True, help="Topic MQTT à écouter")
+    args = parser.parse_args()
+
+    # 📋 Initialiser le logger
+    setup_logger(args.log)
+
+    # 🔑 Charger les variables d'environnement
+    load_dotenv()
+
+    # 🔌 Connexion MySQL
+    mydb = connect_to_mysql()
+
+    # 📡 Connexion MQTT
+    try:
+        client = mqtt_client.Client()
+        client.username_pw_set("Bwps", "scJ5ACj2keRfI^")
+        client.on_message = partial(on_message, args.table)
+        client.connect("54.36.188.119", 1883, 60)
+        client.subscribe(args.topic)
+        logging.info(f"Connexion MQTT réussie et abonnement au topic '{args.topic}'.")
+        client.loop_forever()
+    except Exception as err:
+        logging.error(f"Erreur MQTT : {err}")