constrain websockets to frigate topics

This commit is contained in:
Blake Blackshear 2021-02-13 10:50:36 -06:00
parent 26a3491466
commit 57ced2c284

View File

@ -40,14 +40,14 @@ class MqttBackend():
def publish(self, message): def publish(self, message):
json_message = json.loads(message) json_message = json.loads(message)
self.mqtt_client.publish(json_message['topic'], json_message['payload'], retain=json_message['retain']) self.mqtt_client.publish(f"{self.topic_prefix}/{json_message['topic']}", json_message['payload'], retain=json_message['retain'])
def run(self): def run(self):
def send(client, userdata, message): def send(client, userdata, message):
"""Sends mqtt messages to clients.""" """Sends mqtt messages to clients."""
try: try:
ws_message = json.dumps({ ws_message = json.dumps({
'topic': message.topic, 'topic': message.topic.replace(f"{self.topic_prefix}/",""),
'payload': message.payload.decode() 'payload': message.payload.decode()
}) })
except: except: