feat: websocket error raising

Lash-L · Apr 25, 2024 · 5f020db · 5f020db
2 parents 5611851 + cea2548
commit 5f020db
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -54,7 +54,7 @@ repos:
     hooks:
       - id: codespell
   - repo: https://github.com/PyCQA/flake8
-    rev: 6.0.0
+    rev: 7.0.0
     hooks:
       - id: flake8
   - repo: https://github.com/pre-commit/mirrors-mypy

diff --git a/poetry.lock b/poetry.lock
diff --git a/src/anova_wifi/websocket_handler.py b/src/anova_wifi/websocket_handler.py
@@ -3,8 +3,9 @@
 import logging
 from typing import Any
 
-from aiohttp import ClientSession, ClientWebSocketResponse
+from aiohttp import ClientSession, ClientWebSocketResponse, WebSocketError
 
+from . import WebsocketFailure
 from .web_socket_containers import (
     AnovaCommand,
     APCWifiDevice,
@@ -24,7 +25,10 @@ def __init__(self, firebase_jwt: str, jwt: str, session: ClientSession):
         self.ws: ClientWebSocketResponse | None = None
 
     async def connect(self) -> None:
-        self.ws = await self.session.ws_connect(self.url)
+        try:
+            self.ws = await self.session.ws_connect(self.url)
+        except WebSocketError as ex:
+            raise WebsocketFailure("Failed to connect to the websocket") from ex
         asyncio.ensure_future(self.message_listener())
 
     async def disconnect(self) -> None: