瀏覽代碼

constrain websockets to frigate topics

Blake Blackshear 4 年之前
父節點
當前提交
57ced2c284
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      frigate/http.py

+ 2 - 2
frigate/http.py

@@ -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: