Fix long-term statistics

AlexxIT · Apr 2, 2024 · 725c3c0 · 725c3c0
1 parent a306c7f
commit 725c3c0
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/custom_components/xiaomi_gateway3/hass/entity_description.py b/custom_components/xiaomi_gateway3/hass/entity_description.py
@@ -41,7 +41,7 @@
     "mode": "_attr_mode",  # NumberMode
     "name": "_attr_name",
     "poll": "_attr_should_poll",
-    "stats": "_attr_state_class",
+    "statistics": "_attr_state_class",
     "units": "_attr_native_unit_of_measurement",
     "visible": "_attr_entity_registry_visible_default",
 }
@@ -153,7 +153,7 @@
     },
     "sensor.energy": {
         "class": SENSOR.ENERGY,
-        "stats": SensorStateClass.TOTAL,
+        "statistics": SensorStateClass.TOTAL,
         "units": UnitOfEnergy.KILO_WATT_HOUR,
     },
     ##
@@ -229,4 +229,13 @@ def setup_entity_description(entity: Entity, conv: BaseConv) -> bool:
             v = EntityCategory(v)
         setattr(entity, ENTITY_KEYS.get(k) or k, v)
 
+    # sensor with unit_of_measurement and without state_class will be MEASUREMENT
+    # https://developers.home-assistant.io/docs/core/entity/sensor/#long-term-statistics
+    if (
+        conv.domain == "sensor"
+        and hasattr(entity, "_attr_native_unit_of_measurement")
+        and not hasattr(entity, "_attr_state_class")
+    ):
+        setattr(entity, "_attr_state_class", SensorStateClass.MEASUREMENT)
+
     return True