fix: adiing broken pipe exception log

Author: humbertogontijo

roborock/exceptions.py

@@ -6,6 +6,9 @@ class RoborockException(BaseException):
 class RoborockTimeout(RoborockException):
     """Class for Roborock timeout exceptions."""
 
+class RoborockConnectionException(RoborockException):
+    """Class for Roborock connection exceptions."""
+
 class RoborockBackoffException(RoborockException):
     """Class for Roborock exceptions when many retries were made."""
 

roborock/local_api.py

@@ -10,7 +10,7 @@
 
 from roborock.api import RoborockClient, SPECIAL_COMMANDS
 from roborock.containers import RoborockLocalDeviceInfo
-from roborock.exceptions import RoborockTimeout, CommandVacuumError
+from roborock.exceptions import RoborockTimeout, CommandVacuumError, RoborockConnectionException
 from roborock.typing import RoborockCommand
 from roborock.util import get_running_loop_or_create_one
 
@@ -105,18 +105,23 @@ async def _main_coro(self):
                     await self.on_message(self.device_id, message)
                 except Exception as e:
                     _LOGGER.exception(e)
-            except BrokenPipeError:
+            except BrokenPipeError as e:
+                _LOGGER.exception(e)
                 await self.disconnect()
 
     async def connect(self):
         async with self._mutex:
             if not self.is_connected or self.socket.is_closed:
                 self.socket = RoborockSocket(socket.AF_INET, socket.SOCK_STREAM)
                 self.socket.setblocking(False)
-                async with async_timeout.timeout(self.timeout):
-                    _LOGGER.info(f"Connecting to {self.ip}")
-                    await self.loop.sock_connect(self.socket, (self.ip, 58867))
-                    self.is_connected = True
+                try:
+                    async with async_timeout.timeout(self.timeout):
+                        _LOGGER.info(f"Connecting to {self.ip}")
+                        await self.loop.sock_connect(self.socket, (self.ip, 58867))
+                        self.is_connected = True
+                except Exception as e:
+                    await self.disconnect()
+                    raise RoborockConnectionException(f"Failed connecting to {self.ip}") from e
                 self.loop.create_task(self._main_coro())
 
     async def disconnect(self):
@@ -134,6 +139,7 @@ async def send_message(self, data: bytes):
             raise RoborockTimeout(
                 f"Timeout after {self.timeout} seconds waiting for response"
             ) from None
-        except BrokenPipeError:
+        except BrokenPipeError as e:
+            _LOGGER.exception(e)
             await self.disconnect()
         return response