Don't fail if message is received before websocket start (#9634)

blakeblackshear · Feb 5, 2024 · 00804a0 · 00804a0
1 parent a33f2f1
commit 00804a0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/frigate/comms/ws.py b/frigate/comms/ws.py
@@ -38,6 +38,7 @@ class WebSocketClient(Communicator):  # type: ignore[misc]
 
     def __init__(self, config: FrigateConfig) -> None:
         self.config = config
+        self.websocket_server = None
 
     def subscribe(self, receiver: Callable) -> None:
         self._dispatcher = receiver
@@ -98,6 +99,10 @@ def publish(self, topic: str, payload: str, _: bool) -> None:
             logger.debug(f"payload for {topic} wasn't text. Skipping...")
             return
 
+        if self.websocket_server is None:
+            logger.debug("Skipping message, websocket not connected yet")
+            return
+
         try:
             self.websocket_server.manager.broadcast(ws_message)
         except ConnectionResetError: