Fix "Migration handler not found" after upgrading to 3.0.0 (3.0.1)

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+*.pyc

custom_components/pid_departures/__init__.py

@@ -1,20 +1,56 @@
 """Prague Departure Board integration."""
 from __future__ import annotations
 
+import logging
+
 from homeassistant.const import CONF_ID
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
 
 from .const import DOMAIN
 from .coordinator import PIDConfigEntry, PIDDepartureUpdateCoordinator
 
+_LOGGER = logging.getLogger(__name__)
+
 PLATFORMS: list[str] = ["sensor", "binary_sensor", "calendar"]
 
 
+async def async_migrate_entry(hass: HomeAssistant, entry: PIDConfigEntry) -> bool:
+    """Migrate old config entries.
+
+    1.1: releases before 3.0.0 declared the config flow version as 0.1. The stored
+         data is unchanged since then, so only the version number is raised.
+    1.2: entries created before unique IDs were introduced get the stop ID as
+         unique ID.
+    """
+    _LOGGER.debug(
+        "Migrating entry %s from version %s.%s",
+        entry.title,
+        entry.version,
+        entry.minor_version,
+    )
+    if entry.version > 1:
+        # Downgraded from a newer version with an incompatible format.
+        return False
+
+    if entry.version < 1:
+        hass.config_entries.async_update_entry(entry, version=1, minor_version=1)
+
+    if entry.minor_version < 2:
+        _async_set_missing_unique_id(hass, entry)
+        hass.config_entries.async_update_entry(entry, minor_version=2)
+
+    _LOGGER.debug(
+        "Migrated entry %s to version %s.%s",
+        entry.title,
+        entry.version,
+        entry.minor_version,
+    )
+    return True
+
+
 async def async_setup_entry(hass: HomeAssistant, entry: PIDConfigEntry) -> bool:
     """Set up Departure Board from a config entry."""
-    _async_set_missing_unique_id(hass, entry)
-
     coordinator = PIDDepartureUpdateCoordinator(hass, entry)
     await coordinator.async_config_entry_first_refresh()
 

custom_components/pid_departures/config_flow.py

@@ -59,6 +59,7 @@ async def validate_api_key(hass: HomeAssistant, api_key: str, stop_id: str) -> N
 class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
 
     VERSION = 1
+    MINOR_VERSION = 2
 
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult:
         # Check for any previous instance of the integration

custom_components/pid_departures/manifest.json

@@ -9,5 +9,5 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/dvejsada/PID_integration/issues",
   "requirements": [],
-  "version": "3.0.0"
+  "version": "3.0.1"
 }