Source code for juham.mqtt.jpaho2

from typing import override
import paho.mqtt.client as paho
import os

from juham.base import JMqtt




class JPaho2(paho.Client, JMqtt):
    """MQTT broker implementation based on paho.mqtt Python package.
    Uses multi-inheritance by subclassing paho.CLient and JMqtt super classes.
    """

    _class_id = None
    configured = False
    paho_version = 1
    _client_ids: list = []



    @classmethod
    def register_client_id(cls, name_pid):
        if name_pid in cls._client_ids:
            raise ValueError(f"Duplicate MQTT client ID detected {name_pid}")
        cls._client_ids.append(name_pid)


    def __init__(self, name="paho"):
        name_pid = name + str(os.getpid())
        self.register_client_id(name_pid)
        if self.paho_version == 2:
            super().__init__(paho.CallbackAPIVersion.VERSION1, name_pid)
        else:
            super().__init__(name_pid)
        self.name = name_pid



    @override
    def connect_to_server(
        self, host: str = "localhost", port: int = 1883, keepalive: int = 60
    ):
        # FIXME: is this a bug in the new paho version 2? Subclassing paho client with the
        # new paho version 2 throws invalid host
        # exception even if the host is perfectly ok. Worked fine with the previous paho release.
        # The problem is that the private _host instance variable don't get set by is empty
        # even if self.host is set to perfectly valid value. So we set _host = self.host, which fixes paho client.
        self._host = host
        return super().connect(host, port, keepalive)


    # def on_message(self, mth):
    #    self.on_message = mth

    # def on_connect(self, mth):
    #    self.on_connect = mth

    # def on_disconnect(self, mth):
    #    self.on_disconnect = mth



    @classmethod
    def register(cls):
        if cls._class_id is None:
            JMqtt.register()
            cls.initialize_class()