Merge pull request #114 from explosivo22/dev

Config Flow Migration

Author: explosivo22

custom_components/rinnaicontrolr-ha/__init__.py

@@ -83,3 +83,21 @@ async def async_remove_config_entry_device(
 ) -> bool:
     """Remove a config entry from a device."""
     return True
+
+async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry):
+    """Migrate old entry."""
+    _LOGGER.debug("Migrating from version %s", config_entry.version)
+
+    if config_entry.version == 1:
+        data = {**config_entry.data}
+
+        if not data.get(CONF_ACCESS_TOKEN):
+            data[CONF_ACCESS_TOKEN] = config_entry.data[CONF_REFRESH_TOKEN]
+        if not data.get(CONF_REFRESH_TOKEN):
+            data[CONF_REFRESH_TOKEN] = config_entry.data[CONF_REFRESH_TOKEN]
+
+        hass.config_entries.async_update_entry(config_entry, data=data, version=2)
+
+    _LOGGER.info("Migration to version %s successful", config_entry.version)
+
+    return True

custom_components/rinnaicontrolr-ha/config_flow.py

@@ -23,7 +23,7 @@
 class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
     """Handle a config flow for Rinnai."""
 
-    VERSION = 1
+    VERSION = 2
 
     entry: config_entries.ConfigEntry | None
 

custom_components/rinnaicontrolr-ha/manifest.json

@@ -5,6 +5,6 @@
     "documentation": "https://github.com/explosivo22/rinnaicontrolr-ha/",
     "codeowners": [ "@explosivo22" ],
     "requirements": [ "aiorinnai==0.4.0a2" ],
-    "version": "1.5.4",
+    "version": "1.5.5",
     "iot_class":  "cloud_polling"
 }