🔧 fix for config flow (#2010)

al-one · Dec 12, 2024 · 47ce8eb · 47ce8eb
1 parent 397f870
commit 47ce8eb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/custom_components/xiaomi_miot/config_flow.py b/custom_components/xiaomi_miot/config_flow.py
@@ -37,7 +37,7 @@
     in_china,
     async_analytics_track_event,
 )
-from .core.const import SUPPORTED_DOMAINS, CLOUD_SERVERS, CONF_XIAOMI_CLOUD
+from .core.const import SUPPORTED_DOMAINS, CLOUD_SERVERS, CONF_XIAOMI_CLOUD, HA_VERSION
 from .core.miot_spec import MiotSpec
 from .core.xiaomi_cloud import (
     MiotCloud,
@@ -585,8 +585,7 @@ async def async_step_customizing(self, user_input=None):
 
 class OptionsFlowHandler(config_entries.OptionsFlow, BaseFlowHandler):
     def __init__(self, config_entry: config_entries.ConfigEntry):
-        if not hasattr(self, 'config_entry'):
-            # hass v2024.12+
+        if HA_VERSION >= '2024.12':
             self.config_entry = config_entry
 
     @property

diff --git a/custom_components/xiaomi_miot/core/const.py b/custom_components/xiaomi_miot/core/const.py
@@ -1,12 +1,15 @@
 from enum import Enum
 from typing import Union
 
+from homeassistant.const import __version__ as HAVERSION  # noqa
+from awesomeversion import AwesomeVersion
 from .device_customizes import DEVICE_CUSTOMIZES, GLOBAL_CONVERTERS  # noqa
 from .miot_local_devices import MIOT_LOCAL_MODELS  # noqa
 from .translation_languages import TRANSLATION_LANGUAGES  # noqa
 
 DOMAIN = 'xiaomi_miot'
 DEFAULT_NAME = 'Xiaomi Miot'
+HA_VERSION = AwesomeVersion(HAVERSION)
 
 CONF_MODEL = 'model'
 CONF_XIAOMI_CLOUD = 'xiaomi_cloud'