From d43538785d902b5e01ab96417b9b1713152baa20 Mon Sep 17 00:00:00 2001
From: Torma <tormakristof@tormakristof.eu>
Date: Sat, 3 Oct 2020 13:59:24 +0200
Subject: [PATCH] login to mqtt broker

---
 docker-compose.yml | 2 ++
 src/mqtt_helper.py | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index b4b311c..6fe11af 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -24,6 +24,8 @@ services:
     restart: always
     networks:
       - guard
+    ports:
+      - "127.0.0.1:1883:1883"
 
   guard-service:
     image: registry.kmlabz.com/birbnetes/guard-service
diff --git a/src/mqtt_helper.py b/src/mqtt_helper.py
index 33f6930..d3e6628 100644
--- a/src/mqtt_helper.py
+++ b/src/mqtt_helper.py
@@ -19,7 +19,7 @@ class MQTT:
     """
 
     def __init__(self, host=config.MQTT_HOSTNAME, port=config.MQTT_PORT, client_id=config.CLIENT_ID, qos=2,
-                 retain=True):
+                 retain=True, username=config.MQTT_USERNAME, password=config.MQTT_PASSWORD):
         """
         Init variables
         :param host:
@@ -35,6 +35,8 @@ class MQTT:
         self._topic = None
         self.qos = qos
         self.retain = retain
+        self.username = username
+        self.password = password
 
     def get_topic(self) -> str:
         """
@@ -61,6 +63,7 @@ class MQTT:
         logging.info("Connecting to MQTT")
         self.client = mqtt.Client(client_id=self.client_id, clean_session=True, userdata=None, protocol=mqtt.MQTTv311,
                                   transport="tcp")
+        self.client.username_pw_set(username=self.username,password=self.password)
         self.port = int(self.port)
         self.client.connect(host=self.host, port=self.port, keepalive=60)