Fix Hass 2024.9 support #1434

AlexxIT · Sep 4, 2024 · a85b0df · a85b0df
1 parent 1c835f9
commit a85b0df
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/custom_components/xiaomi_gateway3/core/converters/silabs.py b/custom_components/xiaomi_gateway3/core/converters/silabs.py
@@ -2,7 +2,6 @@
 
 from zigpy.zcl import Cluster
 from zigpy.zcl.foundation import (
-    DATA_TYPES,
     GENERAL_COMMANDS,
     CommandSchema,
     TypeValue,
@@ -19,6 +18,15 @@
     Status as ZDOStatus,
 )
 
+try:
+    # for https://github.com/zigpy/zigpy/blob/0.66.0/zigpy/zcl/foundation.py
+    from zigpy.zcl.foundation import DataType
+
+    DATA_TYPES = {int(d.type_id): d for d in DataType}
+except ImportError:
+    # for https://github.com/zigpy/zigpy/blob/0.65.4/zigpy/zcl/foundation.py
+    from zigpy.zcl.foundation import DATA_TYPES
+
 _LOGGER = logging.getLogger(__name__)
 
 CLUSTERS = {}