Fix configuration migration process

elad-bar · Oct 13, 2022 · aaa2cc8 · aaa2cc8
1 parent e5807fe
commit aaa2cc8
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## 2.0.9
+
+- Fix configuration migration process
+
 ## 2.0.8
 
 - Removed port from configuration as it's not being used

diff --git a/custom_components/edgeos/component/managers/home_assistant.py b/custom_components/edgeos/component/managers/home_assistant.py
@@ -143,9 +143,12 @@ async def async_initialize_data_providers(self):
         updated = False
 
         if self._entry is not None:
-            entry_options = self._entry.options.__dict__
+            migration_data = {}
+            entry_options = self._entry.options
 
-            migration_data = {} if entry_options is None else entry_options
+            if entry_options is not None:
+                for option_key in entry_options:
+                    migration_data[option_key] = entry_options.get(option_key)
 
             if self._entry.data is not None:
                 migration_data[STORAGE_DATA_UNIT] = self._entry.data.get(STORAGE_DATA_UNIT)

diff --git a/custom_components/edgeos/manifest.json b/custom_components/edgeos/manifest.json
@@ -7,6 +7,6 @@
   "codeowners": ["@elad-bar"],
   "requirements": ["aiohttp"],
   "config_flow": true,
-  "version": "2.0.8",
+  "version": "2.0.9",
   "iot_class": "local_polling"
 }