mirror of
https://github.com/pyMC-dev/pyMC_Repeater.git
synced 2026-07-03 08:22:30 +02:00
Lloyd
f8661a2c10
- Added `SQLiteHandler` for managing packet and advert storage in SQLite database. - Implemented `RRDToolHandler` for creating and updating RRD databases for metrics. - Developed `MQTTHandler` for publishing data to MQTT broker. - Created `StorageCollector` to integrate SQLite, RRDTool, and MQTT functionalities. - Added methods for recording packets, adverts, and noise floor data. - Implemented data retrieval methods for packet statistics, recent packets, and noise floor history. - Established database schema with appropriate tables and indices for efficient data access. - Included error handling and logging for database operations and MQTT communications.
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
import json
|
|
import logging
|
|
from typing import Dict, Any
|
|
|
|
try:
|
|
import paho.mqtt.client as mqtt
|
|
MQTT_AVAILABLE = True
|
|
except ImportError:
|
|
MQTT_AVAILABLE = False
|
|
|
|
logger = logging.getLogger("MQTTHandler")
|
|
|
|
|
|
class MQTTHandler:
|
|
def __init__(self, mqtt_config: dict, node_name: str = "unknown"):
|
|
self.mqtt_config = mqtt_config
|
|
self.node_name = node_name
|
|
self.client = None
|
|
self.available = MQTT_AVAILABLE
|
|
self._init_client()
|
|
|
|
def _init_client(self):
|
|
if not self.available or not self.mqtt_config.get("enabled", False):
|
|
logger.info("MQTT disabled or not available")
|
|
return
|
|
|
|
try:
|
|
self.client = mqtt.Client()
|
|
|
|
username = self.mqtt_config.get("username")
|
|
password = self.mqtt_config.get("password")
|
|
if username:
|
|
self.client.username_pw_set(username, password)
|
|
|
|
broker = self.mqtt_config.get("broker", "localhost")
|
|
port = self.mqtt_config.get("port", 1883)
|
|
|
|
self.client.connect(broker, port, 60)
|
|
self.client.loop_start()
|
|
|
|
logger.info(f"MQTT client connected to {broker}:{port}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to initialize MQTT: {e}")
|
|
self.client = None
|
|
|
|
def publish(self, record: dict, record_type: str):
|
|
if not self.client:
|
|
return
|
|
|
|
try:
|
|
base_topic = self.mqtt_config.get("base_topic", "meshcore/repeater")
|
|
topic = f"{base_topic}/{self.node_name}/{record_type}"
|
|
payload = {k: v for k, v in record.items() if v is not None}
|
|
message = json.dumps(payload, default=str)
|
|
self.client.publish(topic, message, qos=0, retain=False)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to publish to MQTT: {e}")
|
|
|
|
def close(self):
|
|
if self.client:
|
|
self.client.loop_stop()
|
|
self.client.disconnect()
|
|
logger.info("MQTT client disconnected") |