diff --git a/mqtt_datalogger.py b/mqtt_datalogger.py index 2ab2af2..2aad238 100644 --- a/mqtt_datalogger.py +++ b/mqtt_datalogger.py @@ -2,11 +2,16 @@ import paho.mqtt.client as mqtt import psycopg2 import json import time +import logging import os import uvicorn from fastapi import FastAPI, HTTPException from pydantic import BaseModel +# Configurazione Logging +logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') +logger = logging.getLogger(__name__) + # Configurazione connessione PostgreSQL DB_CONFIG = { "dbname": os.getenv("DB_NAME"), @@ -44,7 +49,7 @@ def connect_db(): ) return connection except Exception as e: - print(f"Errore di connessione al database: {e}") + logger.error(f"Errore di connessione al database: {e}") return None def setup_database(): @@ -72,8 +77,9 @@ def setup_database(): ); """) connection.commit() + logger.info("Database configurato correttamente.") except Exception as e: - print(f"Errore nella configurazione del database: {e}") + logger.error(f"Errore nella configurazione del database: {e}") finally: cursor.close() connection.close() @@ -88,8 +94,9 @@ def insert_device_status(payload): (payload["deviceEui"], payload["events"][0]["name"], json.dumps(payload)) ) connection.commit() + logger.info(f"Stato del dispositivo inserito: {payload['deviceEui']}") except Exception as e: - print(f"Errore durante l'inserimento dello stato del dispositivo: {e}") + logger.error(f"Errore durante l'inserimento dello stato del dispositivo: {e}") finally: cursor.close() connection.close() @@ -107,9 +114,10 @@ def insert_sensor_data(payload): (payload["deviceEui"], value["channel"], json.dumps(value["measurements"]), value["measureTime"]) ) connection.commit() + logger.info(f"Dati del sensore inseriti per il canale {value['channel']} del dispositivo {payload['deviceEui']}") cursor.close() except Exception as e: - print(f"Errore durante l'inserimento dei dati dei sensori: {e}") + logger.error(f"Errore durante l'inserimento dei dati dei sensori: {e}") finally: connection.close() @@ -136,9 +144,9 @@ def send_downlink_command(client, device_eui, command_id, value=None): payload = json.dumps(command) client.publish(topic, payload, qos=0, retain=True) - print(f"Comando inviato a {topic}: {payload}") + logger.info(f"Comando inviato a {topic}: {payload}") except Exception as e: - print(f"Errore durante l'invio del comando: {e}") + logger.error(f"Errore durante l'invio del comando: {e}") # Endpoint API per inviare comandi@app.post("/send_command") def api_send_command(command: Command): @@ -146,6 +154,7 @@ def api_send_command(command: Command): send_downlink_command(client, command.device_eui, command.command_id, command.value) return {"status": "success", "message": "Comando inviato con successo"} except Exception as e: + logger.error(f"Errore nell'invio del comando: {e}") raise HTTPException(status_code=500, detail=f"Errore nell'invio del comando: {e}") @app.get("/device_status") @@ -156,8 +165,10 @@ def get_device_status(): cursor = connection.cursor() cursor.execute(f"SELECT * FROM {DB_CONFIG['dbschema']}.{DB_CONFIG['table_dev_status']} ORDER BY timestamp DESC LIMIT 50;") results = cursor.fetchall() + logger.info("Stato dei dispositivi recuperato.") return {"status": "success", "data": results} except Exception as e: + logger.error(f"Errore durante il recupero dello stato dei dispositivi: {e}") raise HTTPException(status_code=500, detail=f"Errore durante il recupero dello stato dei dispositivi: {e}") finally: cursor.close() @@ -172,8 +183,10 @@ def get_sensor_data(): cursor = connection.cursor() cursor.execute(f"SELECT * FROM {DB_CONFIG['dbschema']}.{DB_CONFIG['table_sensor_data']} ORDER BY timestamp DESC LIMIT 50;") results = cursor.fetchall() + logger.info("Dati dei sensori recuperati.") return {"status": "success", "data": results} except Exception as e: + logger.error(f"Errore durante il recupero dei dati dei sensori: {e}") raise HTTPException(status_code=500, detail=f"Errore durante il recupero dei dati dei sensori: {e}") finally: cursor.close() @@ -181,17 +194,17 @@ def get_sensor_data(): raise HTTPException(status_code=500, detail="Errore durante il recupero dei dati") # Callback MQTT -def on_connect(client, userdata, flags, rc): +def on_connect(client, userdata, flags, rc, ppp): try: - print("Connesso al broker MQTT con codice di stato:", rc) - for topic in MQTT_TOPICS: + logger.info("Connesso al broker MQTT con codice di stato: %s", rc) + for topic in [MQTT_CONFIG['topics']]: client.subscribe(topic) - print(f"Sottoscritto al topic: {topic}") + logger.info(f"Sottoscritto al topic: {topic}") except Exception as e: - print(f"Errore durante la connessione al broker MQTT: {e}") + logger.error(f"Errore durante la connessione al broker MQTT: {e}") def on_message(client, userdata, msg): - print(f"Messaggio ricevuto su {msg.topic}: {msg.payload}") + logger.info(f"Messaggio ricevuto su {msg.topic}: {msg.payload}") try: payload = json.loads(msg.payload) if "events" in payload: @@ -200,11 +213,11 @@ def on_message(client, userdata, msg): elif payload["events"][0]["name"] == "measure-sensor": insert_sensor_data(payload) except Exception as e: - print(f"Errore nella gestione del messaggio: {e}") + logger.error(f"Errore nella gestione del messaggio: {e}") # Configurazione client MQTT -client = mqtt.Client() -client.username_pw_set(MQTT_USERNAME, MQTT_PASSWORD) +client = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2, protocol=mqtt.MQTTv5) +client.username_pw_set(MQTT_CONFIG['username'], MQTT_CONFIG['password']) client.on_connect = on_connect client.on_message = on_message @@ -212,11 +225,11 @@ client.on_message = on_message def main(): setup_database() try: - client.connect(MQTT_BROKER, MQTT_PORT, 60) + client.connect(MQTT_CONFIG['broker'], int(MQTT_CONFIG['port']), 60) client.loop_start() - print("Applicazione avviata. Accesso all'interfaccia web su http://127.0.0.1:8000") + logger.info("Applicazione avviata. Accesso all'interfaccia web su http://127.0.0.1:8000") except Exception as e: - print(f"Errore nella connessione al broker MQTT: {e}") + logger.error(f"Errore nella connessione al broker MQTT: {e}") if __name__ == "__main__": import uvicorn