class TBClient(threading.Thread):
def __init__(self, config):
super().__init__()
self.setName('Connection thread.')
self.daemon = True
self.__config = config
self.__host = config["host"]
self.__port = config.get("port", 1883)
self.__default_quality_of_service = config.get("qos", 1)
credentials = config["security"]
self.__min_reconnect_delay = 1
self.__tls = bool(
credentials.get('tls', False) or credentials.get('caCert', False))
self.__ca_cert = None
self.__private_key = None
self.__cert = None
self.__token = None
self.__is_connected = False
self.__stopped = False
self.__paused = False
if credentials.get("accessToken") is not None:
self.__token = str(credentials["accessToken"])
if self.__tls:
self.__ca_cert = credentials.get("caCert")
self.__private_key = credentials.get("privateKey")
self.__cert = credentials.get("cert")
self.client = TBGatewayMqttClient(
self.__host,
self.__port,
self.__token,
self,
quality_of_service=self.__default_quality_of_service)
if self.__tls and self.__ca_cert is None and self.__private_key is None and self.__cert is None:
# pylint: disable=protected-access
self.client._client.tls_set_context(SSLContext(PROTOCOL_TLSv1_2))
# Adding callbacks
self.client._client._on_connect = self._on_connect
self.client._client._on_disconnect = self._on_disconnect
# self.client._client._on_log = self._on_log
self.start()
# def _on_log(self, *args):
# if "exception" in args[-1]:
# log.exception(args)
# else:
# log.debug(args)
def pause(self):
self.__paused = True
def unpause(self):
self.__paused = False
def is_connected(self):
return self.__is_connected
def _on_connect(self, client, userdata, flags, result_code, *extra_params):
log.debug('TB client %s connected to ThingsBoard', str(client))
if result_code == 0:
self.__is_connected = True
# pylint: disable=protected-access
self.client._on_connect(client, userdata, flags, result_code,
*extra_params)
def _on_disconnect(self, client, userdata, result_code):
# pylint: disable=protected-access
if self.client._client != client:
log.info(
"TB client %s has been disconnected. Current client for connection is: %s",
str(client), str(self.client._client))
client.disconnect()
client.loop_stop()
else:
self.__is_connected = False
self.client._on_disconnect(client, userdata, result_code)
def stop(self):
# self.disconnect()
self.client.stop()
self.__stopped = True
def disconnect(self):
self.__paused = True
self.unsubscribe('*')
self.client.disconnect()
def unsubscribe(self, subsription_id):
self.client.gw_unsubscribe(subsription_id)
self.client.unsubscribe_from_attribute(subsription_id)
def connect(self, min_reconnect_delay=10):
self.__paused = False
self.__stopped = False
self.__min_reconnect_delay = min_reconnect_delay
def run(self):
keep_alive = self.__config.get("keep_alive", 60)
try:
while not self.client.is_connected() and not self.__stopped:
if not self.__paused:
if self.__stopped:
break
log.debug("connecting to ThingsBoard")
try:
self.client.connect(
tls=self.__tls,
ca_certs=self.__ca_cert,
cert_file=self.__cert,
key_file=self.__private_key,
keepalive=keep_alive,
min_reconnect_delay=self.__min_reconnect_delay)
except ConnectionRefusedError:
pass
except Exception as e:
log.exception(e)
time.sleep(1)
except Exception as e:
log.exception(e)
time.sleep(10)
while not self.__stopped:
try:
if not self.__stopped:
time.sleep(1)
else:
break
except KeyboardInterrupt:
self.__stopped = True
except Exception as e:
log.exception(e)
class TBClient(threading.Thread):
def __init__(self, config):
super().__init__()
self.setName('Connection thread.')
self.daemon = True
self.__config = config
self.__host = config["host"]
self.__port = config.get("port", 1883)
credentials = config["security"]
self.__min_reconnect_delay = 10
self.__tls = True if credentials.get('caCert', False) else False
self.__ca_cert = None
self.__private_key = None
self.__cert = None
self.__token = None
self.__is_connected = False
self.__stopped = False
self.__paused = False
if credentials.get("accessToken") is not None:
self.__token = str(credentials["accessToken"])
if self.__tls:
self.__ca_cert = credentials.get("caCert")
self.__private_key = credentials.get("privateKey")
self.__cert = credentials.get("cert")
self.client = TBGatewayMqttClient(self.__host, self.__port,
self.__token, self)
# Adding callbacks
self.client._client._on_connect = self._on_connect
self.client._client._on_disconnect = self._on_disconnect
self.client._client._on_log = self._on_log
self.start()
def _on_log(self, *args):
if "exception" in args[-1]:
log.exception(args)
else:
log.debug(args)
def pause(self):
self.__paused = True
def unpause(self):
self.__paused = False
def is_connected(self):
return self.client.is_connected()
def _on_connect(self, client, userdata, flags, rc, *extra_params):
log.debug('TB client %s connected to ThingsBoard', str(client))
self.client._on_connect(client, userdata, flags, rc, *extra_params)
def _on_disconnect(self, client, userdata, rc):
log.info("TB client %s has been disconnected.", str(client))
self.client._on_disconnect(client, userdata, rc)
def stop(self):
self.disconnect()
self.__stopped = True
def disconnect(self):
self.__paused = True
self.client.disconnect()
def unsubscribe(self, subsription_id):
self.client.gw_unsubscribe(subsription_id)
self.client.unsubscribe_from_attribute(subsription_id)
def connect(self, min_reconnect_delay=10):
self.__paused = False
self.__stopped = False
self.__min_reconnect_delay = min_reconnect_delay
def run(self):
keep_alive = self.__config.get("keep_alive", 60)
try:
while not self.client.is_connected() and not self.__stopped:
log.debug("connecting to ThingsBoard")
if not self.__paused:
try:
self.client.connect(
tls=self.__tls,
ca_certs=self.__ca_cert,
cert_file=self.__cert,
key_file=self.__private_key,
keepalive=keep_alive,
min_reconnect_delay=self.__min_reconnect_delay)
except ConnectionRefusedError:
pass
time.sleep(1)
except Exception as e:
log.exception(e)
time.sleep(10)
while not self.__stopped:
try:
if not self.__stopped:
time.sleep(1)
else:
break
except KeyboardInterrupt:
self.__stopped = True
except Exception as e:
log.exception(e)