fix: handle connection errors with common exception (#201)

related to #196

Author: CFenner

custom_components/solakon_one/config_flow.py

@@ -11,7 +11,6 @@
 from homeassistant.const import CONF_HOST, CONF_PORT, CONF_SCAN_INTERVAL, UnitOfTime
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.data_entry_flow import FlowResult
-from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers import config_validation as cv, selector
 
 from .const import (
@@ -22,6 +21,7 @@
     DEFAULT_SCAN_INTERVAL,
     DOMAIN,
 )
+from .exceptions import CannotConnect
 from .modbus import get_modbus_hub
 
 _LOGGER = logging.getLogger(__name__)
@@ -152,7 +152,3 @@ async def async_step_init(
                 self._config_entry.options or self._config_entry.data,
             ),
         )
-
-
-class CannotConnect(HomeAssistantError):
-    """Error to indicate we cannot connect."""

custom_components/solakon_one/exceptions.py

@@ -0,0 +1,5 @@
+from homeassistant.exceptions import HomeAssistantError
+
+
+class CannotConnect(HomeAssistantError):
+    """Error to indicate we cannot connect."""

custom_components/solakon_one/modbus.py

@@ -21,6 +21,7 @@
     DEFAULT_SCAN_INTERVAL,
     REGISTERS,
 )
+from .exceptions import CannotConnect
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -95,7 +96,7 @@ async def async_setup(self) -> None:
                     _LOGGER.warning(f"Test read exception: {e}")
             else:
                 _LOGGER.error(f"Failed to connect to {self._host}:{self._port}")
-                raise ConnectionError(f"Failed to connect to {self._host}:{self._port}")
+                raise CannotConnect(f"Failed to connect to {self._host}:{self._port}")
 
         except Exception as err:
             _LOGGER.error(f"Connection setup error: {err}")