-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
6dbc421
commit cb4cfe1
Showing
4 changed files
with
193 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -9,5 +9,5 @@ bluepy | |
| pygobject | ||
| dbus-python | ||
| bleak | ||
| paho-mqtt | ||
| paho-mqtt==1.6.1 | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,131 @@ | ||
| #!/usr/bin/env python3 | ||
| # coding=utf-8 | ||
| """ | ||
| MQTT Manager | ||
| """ | ||
|
|
||
| # region import | ||
| import threading | ||
| import time | ||
| import paho.mqtt.client as mqtt | ||
| import json | ||
|
|
||
| # endregion | ||
|
|
||
|
|
||
| class MqttManager(threading.Thread): | ||
| def __init__( | ||
| self, | ||
| username: str, | ||
| password: str, | ||
| server: str, | ||
| port: int, | ||
| client: str, | ||
| subscribe: str, | ||
| message_callback=None, | ||
| ): | ||
| """ | ||
| Instance MQTT connections | ||
| """ | ||
| threading.Thread.__init__(self) | ||
| self.server = server | ||
| self.port = port | ||
| self.client = mqtt.Client(client) | ||
| self.__subscribe = subscribe | ||
| self.client.username_pw_set(username=username, password=password) | ||
| self.thread = threading.Thread(target=self._start) | ||
| self.status = False | ||
| self.retry = 3 | ||
|
|
||
| def _start(self): | ||
| """ | ||
| Start process mqtt manager | ||
| :rtype: object | ||
| """ | ||
| print(f"Server: {self.server} Port: {self.port}") | ||
| self.client.on_connect = self.on_connect | ||
| self.client.on_message = self.on_message | ||
|
|
||
| while True: | ||
| try: | ||
| # Connect MQTT BROCKER | ||
| mqttt_status = self.client.connect(host=self.server, port=self.port) | ||
| if mqttt_status == 0: | ||
| self.status = True | ||
| else: | ||
| self.status = False | ||
| print("Running server") | ||
| # Block loop | ||
| self.client.loop_forever() | ||
|
|
||
| except Exception as error: | ||
| print(f"Fail Server: {self.server} Error:{error}") | ||
| self.status = False | ||
| # Add sleep in case down network | ||
| time.sleep(0.3) | ||
| pass | ||
|
|
||
| def publish(self, topic, payload): | ||
| """ | ||
| Publish message in broker | ||
| :param topic: Topic | ||
| :param payload: Payload | ||
| :return: | ||
| """ | ||
|
|
||
| (rc, mid) = (-1, -1) | ||
|
|
||
| try: | ||
| retry = self.retry | ||
| while retry > 0: | ||
| (rc, mid) = self.client.publish(topic=topic, payload=payload, qos=1) | ||
| print(f"Status publish: {(rc, mid)}") | ||
| if rc == 0: | ||
| return rc | ||
| time.sleep(0.3) | ||
| except Exception as error: | ||
| print(f"Fail publish error {error}") | ||
| print("fail publish") | ||
| return rc | ||
|
|
||
| def on_connect(self, client, userdata, flags, rc): | ||
| """ | ||
| The callback for when the client receives a CONNACK response from the server. | ||
| :param client: Client | ||
| :param userdata: Userdata | ||
| :param flags: flags | ||
| :param rc: rc | ||
| """ | ||
| try: | ||
| print(f"Connected with result code {str(rc)}") | ||
| # Subscribing in on_connect() means that if we lose the connection and | ||
| # reconnect then subscriptions will be renewed. | ||
| client.subscribe(self.__subscribe) | ||
| except Exception as error: | ||
| print(f"Fail on connect error {error}") | ||
|
|
||
| def on_message(self, client, userdata, msg): | ||
| """ | ||
| The callback for when a PUBLISH message is received from the server. | ||
| :param client: Client | ||
| :param userdata: User data | ||
| :param msg: msg | ||
| """ | ||
| content = json.loads(msg.payload) | ||
| if self.on_message_callback: | ||
| self.on_message_callback(content) # Chama a função de callback com o conteúdo da mensagem | ||
|
|
||
| def stop(self): | ||
| """ | ||
| Stop process | ||
| """ | ||
| print("Stop mqtt client") | ||
| self.client.loop_stop() | ||
| # self.thread.join() | ||
|
|
||
| def run(self): | ||
| """ | ||
| Run process | ||
| """ | ||
| self.thread.daemon = True | ||
| self.thread.start() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters