use consts for units

pail23 · Feb 3, 2024 · 29fa7e0 · 29fa7e0
1 parent 1c8cd01
commit 29fa7e0
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/custom_components/stiebel_eltron_isg/sensor.py b/custom_components/stiebel_eltron_isg/sensor.py
@@ -14,6 +14,8 @@
     UnitOfTemperature,
     UnitOfPressure,
     UnitOfEnergy,
+    UnitOfFrequency,
+    UnitOfVolumeFlowRate,
 )
 
 from .const import (
@@ -343,31 +345,33 @@ def create_volume_stream_entity_description(name, key):
         name="Ventilation air actual fan speed",
         icon="mdi:fan",
         has_entity_name=True,
-        native_unit_of_measurement="Hz",
+        native_unit_of_measurement=UnitOfFrequency.HERTZ,
+        device_class=SensorDeviceClass.FREQUENCY,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     SensorEntityDescription(
         VENTILATION_AIR_TARGET_FLOW_RATE,
         name="Ventilation air target fan speed",
         icon="mdi:fan",
         has_entity_name=True,
-        native_unit_of_measurement="m3/h",
+        native_unit_of_measurement=UnitOfVolumeFlowRate.CUBIC_METERS_PER_HOUR,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     SensorEntityDescription(
         EXTRACT_AIR_ACTUAL_FAN_SPEED,
         name="Extract air actual fan speed",
         icon="mdi:fan",
         has_entity_name=True,
-        native_unit_of_measurement="Hz",
+        native_unit_of_measurement=UnitOfFrequency.HERTZ,
+        device_class=SensorDeviceClass.FREQUENCY,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     SensorEntityDescription(
         EXTRACT_AIR_TARGET_FLOW_RATE,
         name="Extract air target fan speed",
         icon="mdi:fan",
         has_entity_name=True,
-        native_unit_of_measurement="m3/h",
+        native_unit_of_measurement=UnitOfVolumeFlowRate.CUBIC_METERS_PER_HOUR,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     create_humidity_entity_description("Extract air humidity", EXTRACT_AIR_HUMIDITY),