Handle WSServerHandshakeError (#594)

Kane610 · Feb 23, 2024 · f86cd67 · f86cd67
1 parent 7647002
commit f86cd67
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/aiounifi/interfaces/connectivity.py b/aiounifi/interfaces/connectivity.py
@@ -196,6 +196,13 @@ async def websocket(self, callback: Callable[[bytes], None]) -> None:
             err.add_note("Error connecting to UniFi websocket")
             raise
 
+        except aiohttp.WSServerHandshakeError as err:
+            LOGGER.error(
+                "Server handshake error connecting to UniFi websocket: '%s'", err
+            )
+            err.add_note("Server handshake error connecting to UniFi websocket")
+            raise
+
         except WebsocketError:
             raise