PyPI - homematicip - Versions diffs - 2.0.4.post1.dev2__tar.gz → 2.0.4.post1.dev4__tar.gz - Mend

@@ -17,6 +17,7 @@ class WebsocketHandler:
     """
     def __init__(self):
+        self.INITIAL_BACKOFF = 8
         self.url = None
         self._session = None
         self._ws = None
@@ -26,6 +27,7 @@ class WebsocketHandler:
         self._on_message_handlers: List[Callable] = []
         self._on_connected_handler: List[Callable] = []
         self._on_disconnected_handler: List[Callable] = []
+        self._on_reconnect_handler: List[Callable] = []
     def add_on_connected_handler(self, handler: Callable):
         """Adds a handler that is called when the connection is established."""
@@ -35,6 +37,10 @@ class WebsocketHandler:
         """Adds a handler that is called when the connection is closed."""
         self._on_disconnected_handler.append(handler)
+    def add_on_reconnect_handler(self, handler: Callable):
+        """Adds a handler that is called when the connection is re-established."""
+        self._on_reconnect_handler.append(handler)
     def add_on_message_handler(self, handler: Callable):
         """Adds a handler for incoming messages."""
         self._on_message_handlers.append(handler)
@@ -52,7 +58,7 @@ class WebsocketHandler:
                 LOGGER.error(f"Error in handler '{handler_name}': {e}", exc_info=True)
     async def _connect(self, context: ConnectionContext):
-        backoff = 1
+        backoff = self.INITIAL_BACKOFF
         max_backoff = 1800
         while not self._stop_event.is_set():
             try:
@@ -71,9 +77,10 @@ class WebsocketHandler:
                 )
                 LOGGER.info(f"WebSocket connection established to {context.websocket_url}.")
                 await self._call_handlers(self._on_connected_handler)
-                backoff = 1
+                backoff = self.INITIAL_BACKOFF
                 await self._listen()
             except Exception as e:
+                await self._call_handlers(self._on_reconnect_handler)
                 LOGGER.warning(f"Websocket lost connection: {e}. Retry in {backoff:.1f}s.")
                 await asyncio.sleep(backoff)
                 backoff = min(backoff * 2, max_backoff)
@@ -99,6 +106,8 @@ class WebsocketHandler:
                 await self._session.close()
             self._session = None
+        await self._call_handlers(self._on_disconnected_handler)
     async def start(self, context: ConnectionContext):
         async with self._task_lock:
             LOGGER.info("Start websocket client...")

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: homematicip
-Version: 2.0.4.post1.dev2
+Version: 2.0.4.post1.dev4
 Summary: An API for the homematicip cloud
 Author-email: Thomas Hahn <homematicip-rest-api@outlook.com>
 License: GPL-3.0-or-later

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
-__version__ = version = '2.0.4.post1.dev2'
-__version_tuple__ = version_tuple = (2, 0, 4, 'post1', 'dev2')
+__version__ = version = '2.0.4.post1.dev4'
+__version_tuple__ = version_tuple = (2, 0, 4, 'post1', 'dev4')

@@ -766,3 +766,8 @@ class AsyncHome(HomeMaticIPObject):
         if self._websocket_client:
             self._websocket_client.add_on_disconnected_handler(handler)
+    def set_on_reconnect_handler(self, handler: Callable):
+        """Sets a callback that is called when the WebSocket connection is trying to reconnect
+        after a disconnect."""
+        if self._websocket_client:
+            self._websocket_client.add_on_reconnect_handler(handler)

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: homematicip
-Version: 2.0.4.post1.dev2
+Version: 2.0.4.post1.dev4
 Summary: An API for the homematicip cloud
 Author-email: Thomas Hahn <homematicip-rest-api@outlook.com>
 License: GPL-3.0-or-later

homematicip 2.0.4.post1.dev2__tar.gz → 2.0.4.post1.dev4__tar.gz

homematicip 2.0.4.post1.dev2tar.gz → 2.0.4.post1.dev4tar.gz