Fixed issue when cached schools will hang initialization.

custom_components/bakalari/config_flow.py

@@ -40,6 +40,7 @@ def __init__(self) -> None:
         self._schools: Schools | None = None
         self._loading_task = None
         self._reauth_data: dict | None = None
+        self._loading_error: str | None = None
 
     async def _load_schools(self) -> bool:
         """Load schools from cache or fetch new ones from server."""
@@ -63,6 +64,7 @@ async def _load_schools(self) -> bool:
                     self.__class__.__name__,
                     __name__,
                 )
+                self._loading_error = "cannot_connect"
                 return False
 
             await schools_storage.async_save(schools_cache.school_list)
@@ -71,17 +73,24 @@ async def _load_schools(self) -> bool:
                 self.__class__.__name__,
                 __name__,
             )
-        else:
-            _LOGGER.info(
-                "[class=%s module=%s] Schools loaded from cache.",
-                self.__class__.__name__,
-                __name__,
-            )
-        return False
+            return True
+
+        _LOGGER.info(
+            "[class=%s module=%s] Schools loaded from cache.",
+            self.__class__.__name__,
+            __name__,
+        )
+        return True
 
     async def async_step_user(self, user_input=None) -> config_entries.ConfigFlowResult:
         """Handle the initial step."""
 
+        # After progress finishes, HA will re-enter this step; at that point we can
+        # either continue or show an error with a retry button.
+        if user_input is not None and user_input.get("retry"):
+            self._loading_task = None
+            self._loading_error = None
+
         if self._loading_task is None:
             self._loading_task = self.hass.async_create_task(self._load_schools())
 
@@ -91,6 +100,34 @@ async def async_step_user(self, user_input=None) -> config_entries.ConfigFlowRes
                 progress_task=self._loading_task,
             )
 
+        # Task exists -> either still running or already finished.
+        if not self._loading_task.done():
+            return self.async_show_progress(
+                step_id="user",
+                progress_action="loading_schools",
+                progress_task=self._loading_task,
+            )
+
+        try:
+            ok = bool(self._loading_task.result())
+        except Exception:  # noqa: BLE001 - surface as HA form error
+            _LOGGER.exception(
+                "[class=%s module=%s] Unexpected error while loading schools",
+                self.__class__.__name__,
+                __name__,
+            )
+            ok = False
+            self._loading_error = "unknown"
+
+        if not ok:
+            # Reset task so user can retry.
+            self._loading_task = None
+            return self.async_show_form(
+                step_id="user",
+                data_schema=vol.Schema({vol.Optional("retry", default=True): bool}),
+                errors={"base": self._loading_error or "cannot_connect"},
+            )
+
         return self.async_show_progress_done(next_step_id="complete")
 
     async def async_step_complete(

custom_components/bakalari/translations/cs.json

@@ -4,7 +4,8 @@
     "error": {
       "loading_failed": "Nepodařilo se načíst školy z serveru ani s cache!",
       "invalid_auth": "Neplatné přihlašovací údaje!",
-      "cannot_connect": "Nelze se připojit k serveru!"
+      "cannot_connect": "Nelze se připojit k serveru!",
+      "unknown": "Neznámá chyba."
     },
     "abort": {
       "reauth_entry_not_found": "Nepodařilo se najít dítě s daným přihlašovacím jménem.",