Update __init__.py (#323)

Fixing DATA_CUSTOMIZE hass 2024.11 issue
custom-components · Nov 7, 2024 · 576363b · 576363b
1 parent 0553046
commit 576363b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/remote_homeassistant/__init__.py b/custom_components/remote_homeassistant/__init__.py
@@ -19,7 +19,11 @@
 import homeassistant.components.websocket_api.auth as api
 import homeassistant.helpers.config_validation as cv
 import voluptuous as vol
-from homeassistant.config import DATA_CUSTOMIZE
+try:
+    # hass 2024.11
+    from homeassistant.core_config import DATA_CUSTOMIZE
+except (ModuleNotFoundError, ImportError):
+    from homeassistant.helpers.entity import DATA_CUSTOMIZE
 from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
 from homeassistant.const import (CONF_ABOVE, CONF_ACCESS_TOKEN, CONF_BELOW,
                                  CONF_DOMAINS, CONF_ENTITIES, CONF_ENTITY_ID,