move migrate entry

zweckj · Oct 30, 2023 · cc7e19f · cc7e19f
1 parent 0a53d83
commit cc7e19f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/custom_components/lamarzocco/__init__.py b/custom_components/lamarzocco/__init__.py
@@ -65,3 +65,8 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         hass.data[DOMAIN] = {}
 
     return unload_ok
+
+async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    if entry.version != 1:
+        entry.version = 2
+    return True
diff --git a/custom_components/lamarzocco/config_flow.py b/custom_components/lamarzocco/config_flow.py
@@ -92,11 +92,6 @@ def __init__(self) -> None:
         self._config: dict[str, Any] = {}
         self._machines: list[str] = []
 
-    async def async_migrate_entry(hass: core.HomeAssistant, config_entry: ConfigEntry) -> bool:
-        if config_entry.version == 1:
-            config_entry.version = 2
-        return True
-
     async def async_step_user(
         self, user_input: dict[str, Any] | None = None
     ) -> FlowResult:

diff --git a/custom_components/lamarzocco/manifest.json b/custom_components/lamarzocco/manifest.json
@@ -23,7 +23,7 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/zweckj/lamarzocco/issues",
   "loggers": ["lmcloud"],
-  "requirements": ["lmcloud==0.4.4b3"],
+  "requirements": ["lmcloud==0.4.4b4"],
   "ssdp": [],
   "version": "0.12.4"
 }