From 00c05141277756b25575f8e2c17818e36888c202 Mon Sep 17 00:00:00 2001
From: Nick Mowen <nickmowen213@gmail.com>
Date: Tue, 22 Nov 2022 15:06:09 -0700
Subject: [PATCH] Fix ws mypy

---
 frigate/comms/ws.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/frigate/comms/ws.py b/frigate/comms/ws.py
index 797c08073..0a3aea169 100644
--- a/frigate/comms/ws.py
+++ b/frigate/comms/ws.py
@@ -4,6 +4,8 @@ import json
 import logging
 import threading
 
+from typing import Callable
+
 from wsgiref.simple_server import make_server
 from ws4py.server.wsgirefserver import (
     WebSocketWSGIHandler,
@@ -20,23 +22,23 @@ from frigate.config import FrigateConfig
 logger = logging.getLogger(__name__)
 
 
-class WebSocketClient(Communicator):
+class WebSocketClient(Communicator):  # type: ignore[misc]
     """Frigate wrapper for ws client."""
 
     def __init__(self, config: FrigateConfig) -> None:
         self.config = config
 
-    def subscribe(self, receiver) -> None:
+    def subscribe(self, receiver: Callable) -> None:
         self._dispatcher = receiver
         self.start()
 
-    def start(self):
+    def start(self) -> None:
         """Start the websocket client."""
 
-        class _WebSocketHandler(WebSocket):
+        class _WebSocketHandler(WebSocket):  # type: ignore[misc]
             receiver = self._dispatcher
 
-            def received_message(self, message):
+            def received_message(self, message: WebSocket.received_message) -> None:
                 try:
                     json_message = json.loads(message.data.decode("utf-8"))
                     json_message = {
@@ -72,7 +74,7 @@ class WebSocketClient(Communicator):
         )
         self.websocket_thread.start()
 
-    def publish(self, topic: str, payload: str, _) -> None:
+    def publish(self, topic: str, payload: str, _: bool) -> None:
         try:
             ws_message = json.dumps(
                 {
@@ -87,7 +89,7 @@ class WebSocketClient(Communicator):
 
         self.websocket_server.manager.broadcast(ws_message)
 
-    def stop(self):
+    def stop(self) -> None:
         self.websocket_server.manager.close_all()
         self.websocket_server.manager.stop()
         self.websocket_server.manager.join()