Add logging to the custom sleep function (#115)

Author: DeerMaximum

custom_components/ta_cmi/__init__.py

@@ -114,6 +114,11 @@ async def custom_sleep(delay: int) -> None:
             await asyncio.sleep(delay)
         except asyncio.CancelledError:
             elapsed = time.time() - start
+            _LOGGER.debug(
+                "Sleep cancelled after %s. Sleep remaining time: %s",
+                elapsed,
+                delay - elapsed,
+            )
             await asyncio.sleep(delay - elapsed)
 
     async def _async_update_data(self) -> dict[str, Any]:

custom_components/ta_cmi/config_flow.py

@@ -2,10 +2,10 @@
 from __future__ import annotations
 
 import asyncio
+from copy import deepcopy
 from datetime import timedelta
 import time
 from typing import Any
-from copy import deepcopy
 
 from aiohttp import ClientSession
 from homeassistant import config_entries
@@ -103,7 +103,7 @@ def __init__(self) -> None:
         self.start_time: float = ConfigFlow.init_start_time
 
     async def async_step_user(
-            self, user_input: dict[str, Any] | None = None
+        self, user_input: dict[str, Any] | None = None
     ) -> FlowResult:
         """Handle the initial step."""
         errors: dict[str, Any] = {}
@@ -145,7 +145,7 @@ async def async_step_user(
         )
 
     async def async_step_devices(
-            self, user_input: dict[str, Any] | None = None
+        self, user_input: dict[str, Any] | None = None
     ) -> FlowResult:
         """Step for setup devices."""
         errors: dict[str, Any] = {}
@@ -203,14 +203,17 @@ async def async_step_devices(
                 else:
                     errors["base"] = "unknown"
                     _LOGGER.exception("Unexpected exception: %s", err)
+                    break
             except RateLimitError:
                 errors["base"] = "rate_limit"
+                break
             except InvalidDeviceError:
                 errors["base"] = "invalid_device"
                 _LOGGER.warning("Invalid device: %s", dev.id)
             except Exception as err:  # pylint: disable=broad-except
                 _LOGGER.exception("Unexpected exception: %s", err)
                 errors["base"] = "unknown"
+                break
             else:
                 devices_list[dev.id] = str(dev)
 
@@ -232,7 +235,7 @@ def _generate_channel_types(self) -> list[str]:
         return [x.title() for x in DEVICE_TYPE_STRING_MAP.values()]
 
     async def async_step_channel(
-            self, user_input: dict[str, Any] | None = None
+        self, user_input: dict[str, Any] | None = None
     ) -> FlowResult:
         """Step for setup channels."""
 
@@ -294,7 +297,7 @@ async def async_step_finish(self) -> FlowResult:
     @staticmethod
     @callback
     def async_get_options_flow(
-            config_entry: config_entries.ConfigEntry,
+        config_entry: config_entries.ConfigEntry,
     ) -> OptionsFlowHandler:
         """Get the options flow for this handler."""
         return OptionsFlowHandler(config_entry)
@@ -327,7 +330,7 @@ def __init__(self, config_entry: config_entries.ConfigEntry) -> None:
         self.data = dict(self.config_entry.data)
 
     async def async_step_init(
-            self, user_input: dict[str, Any] | None = None
+        self, user_input: dict[str, Any] | None = None
     ) -> FlowResult:
         """Handle options flow."""
 
@@ -343,9 +346,7 @@ async def async_step_init(
                 try:
                     tmp = deepcopy(self.data)
                     tmp[CONF_HOST] = user_input[CONF_HOST]
-                    await validate_login(
-                        tmp, async_get_clientsession(self.hass)
-                    )
+                    await validate_login(tmp, async_get_clientsession(self.hass))
                 except CannotConnect:
                     errors["base"] = "cannot_connect"
                 except InvalidAuth: