Add options/possible states as extra state attributes

vincentwolsink · Nov 26, 2024 · ff880ea · ff880ea
1 parent 3ae1974
commit ff880ea
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 9 deletions.
diff --git a/custom_components/aguaiot/const.py b/custom_components/aguaiot/const.py
@@ -264,15 +264,15 @@ class AguaIOTCanalizationEntityDescription(ClimateEntityDescription):
         icon="mdi:fire",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
+        device_class=SensorDeviceClass.ENUM,
     ),
     AguaIOTSensorEntityDescription(
         key="alarms_get",
         name="Alarm",
         icon="mdi:alert-outline",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
+        device_class=SensorDeviceClass.ENUM,
         force_enabled=True,
     ),
     AguaIOTSensorEntityDescription(
@@ -281,17 +281,17 @@ class AguaIOTCanalizationEntityDescription(ClimateEntityDescription):
         icon="mdi:gauge",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
         force_enabled=True,
+        device_class=SensorDeviceClass.ENUM,
     ),
     AguaIOTSensorEntityDescription(
         key="real_power_wood_get",
         name="Real Wood Power",
         icon="mdi:gauge",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
         force_enabled=True,
+        device_class=SensorDeviceClass.ENUM,
         hybrid_only=True,
     ),
     AguaIOTSensorEntityDescription(
@@ -300,23 +300,23 @@ class AguaIOTCanalizationEntityDescription(ClimateEntityDescription):
         icon="mdi:fan",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
+        device_class=SensorDeviceClass.ENUM,
     ),
     AguaIOTSensorEntityDescription(
         key="vent_rear2_get",
         name="Real Vent Rear",
         icon="mdi:fan",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
+        device_class=SensorDeviceClass.ENUM,
     ),
     AguaIOTSensorEntityDescription(
         key="type_combustible_get",
         name="Fuel",
         icon="mdi:gas-burner",
         native_unit_of_measurement=None,
         state_class=None,
-        device_class=None,
+        device_class=SensorDeviceClass.ENUM,
     ),
     AguaIOTSensorEntityDescription(
         key="pres_h2o_get",

diff --git a/custom_components/aguaiot/sensor.py b/custom_components/aguaiot/sensor.py
@@ -4,7 +4,7 @@
     CoordinatorEntity,
     DataUpdateCoordinator,
 )
-from homeassistant.components.sensor import SensorEntity
+from homeassistant.components.sensor import SensorEntity, SensorDeviceClass
 from homeassistant.helpers.entity import DeviceInfo
 from .const import SENSORS, DOMAIN
 
@@ -69,5 +69,25 @@ def extra_state_attributes(self):
             return {
                 "raw_value": self._device.get_register_value(
                     self.entity_description.key
-                )
+                ),
             }
+
+    @property
+    def options(self):
+        if self.entity_description.device_class == SensorDeviceClass.ENUM:
+            options = sorted(
+                list(
+                    set(
+                        self._device.get_register_value_options(
+                            self.entity_description.key
+                        ).values()
+                    )
+                )
+            )
+            cur_value = self._device.get_register_value_description(
+                self.entity_description.key
+            )
+            if cur_value not in options:
+                options.append(cur_value)
+
+            return options