summaryrefslogtreecommitdiff
path: root/src/home/mqtt/_mqtt.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/home/mqtt/_mqtt.py')
-rw-r--r--src/home/mqtt/_mqtt.py86
1 files changed, 0 insertions, 86 deletions
diff --git a/src/home/mqtt/_mqtt.py b/src/home/mqtt/_mqtt.py
deleted file mode 100644
index 746ae2e..0000000
--- a/src/home/mqtt/_mqtt.py
+++ /dev/null
@@ -1,86 +0,0 @@
-import os.path
-import paho.mqtt.client as mqtt
-import ssl
-import logging
-
-from ._config import MqttCreds, MqttConfig
-from typing import Optional
-
-
-class Mqtt:
- _connected: bool
- _is_server: bool
- _mqtt_config: MqttConfig
-
- def __init__(self,
- clean_session=True,
- client_id='',
- creds: Optional[MqttCreds] = None,
- is_server=False):
- if not client_id:
- raise ValueError('client_id must not be empty')
-
- self._client = mqtt.Client(client_id=client_id,
- protocol=mqtt.MQTTv311,
- clean_session=clean_session)
- self._client.on_connect = self.on_connect
- self._client.on_disconnect = self.on_disconnect
- self._client.on_message = self.on_message
- self._client.on_log = self.on_log
- self._client.on_publish = self.on_publish
- self._loop_started = False
- self._connected = False
- self._is_server = is_server
- self._mqtt_config = MqttConfig()
- self._logger = logging.getLogger(self.__class__.__name__)
-
- if not creds:
- creds = self._mqtt_config.creds() if not is_server else self._mqtt_config.server_creds()
-
- self._client.username_pw_set(creds.username, creds.password)
-
- def _configure_tls(self):
- ca_certs = os.path.realpath(os.path.join(
- os.path.dirname(os.path.realpath(__file__)),
- '..',
- '..',
- '..',
- 'assets',
- 'mqtt_ca.crt'
- ))
- self._client.tls_set(ca_certs=ca_certs,
- cert_reqs=ssl.CERT_REQUIRED,
- tls_version=ssl.PROTOCOL_TLSv1_2)
-
- def connect_and_loop(self, loop_forever=True):
- self._configure_tls()
- addr = self._mqtt_config.local_addr() if self._is_server else self._mqtt_config.remote_addr()
- self._client.connect(addr.host, addr.port, 60)
- if loop_forever:
- self._client.loop_forever()
- else:
- self._client.loop_start()
- self._loop_started = True
-
- def disconnect(self):
- self._client.disconnect()
- self._client.loop_write()
- self._client.loop_stop()
-
- def on_connect(self, client: mqtt.Client, userdata, flags, rc):
- self._logger.info("Connected with result code " + str(rc))
- self._connected = True
-
- def on_disconnect(self, client: mqtt.Client, userdata, rc):
- self._logger.info("Disconnected with result code " + str(rc))
- self._connected = False
-
- def on_log(self, client: mqtt.Client, userdata, level, buf):
- level = mqtt.LOGGING_LEVEL[level] if level in mqtt.LOGGING_LEVEL else logging.INFO
- self._logger.log(level, f'MQTT: {buf}')
-
- def on_message(self, client: mqtt.Client, userdata, msg):
- self._logger.debug(msg.topic + ": " + str(msg.payload))
-
- def on_publish(self, client: mqtt.Client, userdata, mid):
- self._logger.debug(f'publish done, mid={mid}')