75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
import paho.mqtt.client as mqttClient
|
|
import mysql.connector
|
|
import sys
|
|
import os
|
|
import logging
|
|
from dotenv import load_dotenv
|
|
from datetime import datetime
|
|
|
|
# Charger les variables d'environnement
|
|
load_dotenv()
|
|
|
|
# 📁 Création du dossier de logs s'il n'existe pas
|
|
dossier_logs = "/var/log/Chaufferie_Saclay"
|
|
os.makedirs(dossier_logs, exist_ok=True)
|
|
|
|
# 📝 Configuration du logger
|
|
logfile = os.path.join(dossier_logs, "Chaufferie_Saclay.log")
|
|
logging.basicConfig(
|
|
filename=logfile,
|
|
level=logging.INFO,
|
|
format="%(asctime)s - %(levelname)s - %(message)s"
|
|
)
|
|
|
|
# Ajoute aussi l'affichage console
|
|
console = logging.StreamHandler()
|
|
console.setLevel(logging.INFO)
|
|
formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
|
|
console.setFormatter(formatter)
|
|
logging.getLogger('').addHandler(console)
|
|
|
|
# 🔌 Connexion MySQL
|
|
try:
|
|
sys.path.insert(0, "/myenv/lib/python3.11.2/site-packages")
|
|
mydb = mysql.connector.connect(
|
|
host=os.getenv("DB_HOST"),
|
|
user=os.getenv("DB_USER"),
|
|
password=os.getenv("DB_PASSWORD"),
|
|
database=os.getenv("DB_NAME")
|
|
)
|
|
logging.info("Connexion MySQL réussie.")
|
|
except mysql.connector.Error as err:
|
|
logging.error(f"Erreur de connexion MySQL : {err}")
|
|
sys.exit(1)
|
|
|
|
# 📥 Callback MQTT
|
|
def on_message(_client, _userdata, msg):
|
|
try:
|
|
logging.info(f"Message reçu sur {msg.topic}: {msg.payload.decode()}")
|
|
cursor = mydb.cursor()
|
|
|
|
topic_mqtt = msg.topic.split('/')[1] # ex: 'ECS'
|
|
frigo_name = msg.topic.split('/')[-1] # ex: 'Dietrich_1'
|
|
temperature = float(msg.payload.decode())
|
|
|
|
sql = "INSERT INTO Chaufferie (Sonde, Temperature, Date, Topic) VALUES (%s, %s, %s, %s)"
|
|
val = (frigo_name, temperature, datetime.now(), topic_mqtt)
|
|
|
|
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}")
|
|
|
|
# 📡 Connexion MQTT
|
|
try:
|
|
client = mqttClient.Client()
|
|
client.username_pw_set("Bwps", "scJ5ACj2keRfI^")
|
|
client.on_message = on_message
|
|
client.connect("54.36.188.119", 1883, 60)
|
|
client.subscribe("Chaufferie/#")
|
|
logging.info("Connexion MQTT réussie et abonnement au topic 'Chaufferie/#'.")
|
|
client.loop_forever()
|
|
except Exception as err:
|
|
logging.error(f"Erreur MQTT : {err}")
|