mqtt.py

import paho.mqtt.client as mqtt_client


class Mqtt(mqtt_client.Client):
    def __init__(self, client_id, cfg):
        super().__init__(client_id=client_id, clean_session=True)
        self.enable_logger()
        self.username_pw_set(cfg["user"], cfg["passwd"])
        if cfg["use_ssl"]:
            self.tls_set()
        if cfg["use_ssl"] and not cfg["validate_cert"]:
            self.tls_insecure_set(True)
        self.on_connect = self._on_connect_callback
        self.connect(cfg["url"], cfg["port"])
        self.loop_start()
        self.subscriptions = []

    def __del__(self):
        self.disconnect()

    def _on_connect_callback(self, client, userdata, flags, rc):
        if len(self.subscriptions):
            self.subscribe(self.subscriptions)

    def persistent_subscribe(self, topic):
        self.subscriptions.append((topic, 0))
        self.subscribe(topic)