sv/tsun-gen3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add callback on mqtt/ha conn establishment

Browse Source
This commit is contained in:
Stefan Allius
2023-11-01 21:11:54 +01:00
parent 85be9072db
commit 7a2667767e
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/src/mqtt.py
View File

@@ -16,9 +16,11 @@ class Singleton(type):
class Mqtt(metaclass=Singleton): class Mqtt(metaclass=Singleton):
client = None client = None
cb_MqttIsUp = None
def __init__(self):
def __init__(self, cb_MqttIsUp):
logger_mqtt.debug(f'MQTT: __init__') logger_mqtt.debug(f'MQTT: __init__')
if cb_MqttIsUp: self.cb_MqttIsUp = cb_MqttIsUp
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
self.task = loop.create_task(self.__loop()) self.task = loop.create_task(self.__loop())
self.ha_restarts = 0 self.ha_restarts = 0
@@ -61,6 +63,11 @@ class Mqtt(metaclass=Singleton):
while True: while True:
try: try:
async with self.client: async with self.client:
logger_mqtt.info('MQTT broker connection established')
if self.cb_MqttIsUp:
await self.cb_MqttIsUp()
async with self.client.messages() as messages: async with self.client.messages() as messages:
await self.client.subscribe(f"{ha['auto_conf_prefix']}/status") await self.client.subscribe(f"{ha['auto_conf_prefix']}/status")
async for message in messages: async for message in messages:
@@ -68,6 +75,7 @@ class Mqtt(metaclass=Singleton):
logger_mqtt.info(f'Home-Assistant Status: {status}') logger_mqtt.info(f'Home-Assistant Status: {status}')
if status == 'online': if status == 'online':
self.ha_restarts += 1 self.ha_restarts += 1
await self.cb_MqttIsUp()
except aiomqtt.MqttError: except aiomqtt.MqttError:
logger_mqtt.info(f"Connection lost; Reconnecting in {interval} seconds ...") logger_mqtt.info(f"Connection lost; Reconnecting in {interval} seconds ...")