feat: set default sensor state_class

XiaoMi · topsworld · Jan 17, 2025 · Jan 7, 2025 · Jan 7, 2025 · Jan 8, 2025
commit e7564b01522d224e8da724b421574eb20ff04a20
diff --git a/custom_components/xiaomi_home/config_flow.py b/custom_components/xiaomi_home/config_flow.py
@@ -159,7 +159,7 @@ def __init__(self) -> None:
         self._area_name_rule = self.DEFAULT_AREA_NAME_RULE
         self._action_debug = False
         self._hide_non_standard_entities = False
-        self._display_binary_mode = ['text']
+        self._display_binary_mode = ['bool']
         self._display_devices_changed_notify = ['add', 'del', 'offline']
         self._auth_info = {}
         self._nick_name = DEFAULT_NICK_NAME

diff --git a/custom_components/xiaomi_home/sensor.py b/custom_components/xiaomi_home/sensor.py
@@ -53,7 +53,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.components.sensor import SensorEntity, SensorDeviceClass
-from homeassistant.components.sensor import DEVICE_CLASS_UNITS
+from homeassistant.components.sensor import DEVICE_CLASS_UNITS, SensorStateClass
 
 from .miot.miot_device import MIoTDevice, MIoTPropertyEntity
 from .miot.miot_spec import MIoTSpecProperty
@@ -115,6 +115,9 @@ def __init__(self, miot_device: MIoTDevice, spec: MIoTSpecProperty) -> None:
         # Set state_class
         if spec.state_class:
             self._attr_state_class = spec.state_class
+        else:
+            # Set default state_class
+            self._attr_state_class = SensorStateClass.MEASUREMENT
 
     @property
     def native_value(self) -> Any: