From aa995fb590b359b69dad1f7b1b7ef47724ed8240 Mon Sep 17 00:00:00 2001 From: David Knowles Date: Wed, 31 Dec 2025 04:47:48 -0500 Subject: [PATCH] Use WATER device_class for Hydrawise sensors (#160018) --- homeassistant/components/hydrawise/sensor.py | 12 +++++------ .../hydrawise/snapshots/test_sensor.ambr | 20 +++++++++---------- 2 files changed, 16 insertions(+), 16 deletions(-) diff --git a/homeassistant/components/hydrawise/sensor.py b/homeassistant/components/hydrawise/sensor.py index 3a04a587bb4..19fcd0295a2 100644 --- a/homeassistant/components/hydrawise/sensor.py +++ b/homeassistant/components/hydrawise/sensor.py @@ -67,21 +67,21 @@ FLOW_CONTROLLER_SENSORS: tuple[HydrawiseSensorEntityDescription, ...] = ( HydrawiseSensorEntityDescription( key="daily_total_water_use", translation_key="daily_total_water_use", - device_class=SensorDeviceClass.VOLUME, + device_class=SensorDeviceClass.WATER, suggested_display_precision=1, value_fn=lambda sensor: _get_water_use(sensor).total_use, ), HydrawiseSensorEntityDescription( key="daily_active_water_use", translation_key="daily_active_water_use", - device_class=SensorDeviceClass.VOLUME, + device_class=SensorDeviceClass.WATER, suggested_display_precision=1, value_fn=lambda sensor: _get_water_use(sensor).total_active_use, ), HydrawiseSensorEntityDescription( key="daily_inactive_water_use", translation_key="daily_inactive_water_use", - device_class=SensorDeviceClass.VOLUME, + device_class=SensorDeviceClass.WATER, suggested_display_precision=1, value_fn=lambda sensor: _get_water_use(sensor).total_inactive_use, ), @@ -91,7 +91,7 @@ FLOW_ZONE_SENSORS: tuple[SensorEntityDescription, ...] = ( HydrawiseSensorEntityDescription( key="daily_active_water_use", translation_key="daily_active_water_use", - device_class=SensorDeviceClass.VOLUME, + device_class=SensorDeviceClass.WATER, suggested_display_precision=1, value_fn=lambda sensor: float( _get_water_use(sensor).active_use_by_zone_id.get(sensor.zone.id, 0.0) @@ -204,7 +204,7 @@ class HydrawiseSensor(HydrawiseEntity, SensorEntity): @property def native_unit_of_measurement(self) -> str | None: """Return the unit_of_measurement of the sensor.""" - if self.entity_description.device_class != SensorDeviceClass.VOLUME: + if self.entity_description.device_class != SensorDeviceClass.WATER: return self.entity_description.native_unit_of_measurement return ( UnitOfVolume.GALLONS @@ -217,7 +217,7 @@ class HydrawiseSensor(HydrawiseEntity, SensorEntity): """Icon of the entity based on the value.""" if ( self.entity_description.key in FLOW_MEASUREMENT_KEYS - and self.entity_description.device_class == SensorDeviceClass.VOLUME + and self.entity_description.device_class == SensorDeviceClass.WATER and round(self.state, 2) == 0.0 ): return "mdi:water-outline" diff --git a/tests/components/hydrawise/snapshots/test_sensor.ambr b/tests/components/hydrawise/snapshots/test_sensor.ambr index e2e97da120c..9a552db3984 100644 --- a/tests/components/hydrawise/snapshots/test_sensor.ambr +++ b/tests/components/hydrawise/snapshots/test_sensor.ambr @@ -28,7 +28,7 @@ 'suggested_unit_of_measurement': , }), }), - 'original_device_class': , + 'original_device_class': , 'original_icon': None, 'original_name': 'Daily active water use', 'platform': 'hydrawise', @@ -44,7 +44,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ 'attribution': 'Data provided by hydrawise.com', - 'device_class': 'volume', + 'device_class': 'water', 'friendly_name': 'Home Controller Daily active water use', 'unit_of_measurement': , }), @@ -139,7 +139,7 @@ 'suggested_unit_of_measurement': , }), }), - 'original_device_class': , + 'original_device_class': , 'original_icon': None, 'original_name': 'Daily inactive water use', 'platform': 'hydrawise', @@ -155,7 +155,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ 'attribution': 'Data provided by hydrawise.com', - 'device_class': 'volume', + 'device_class': 'water', 'friendly_name': 'Home Controller Daily inactive water use', 'unit_of_measurement': , }), @@ -196,7 +196,7 @@ 'suggested_unit_of_measurement': , }), }), - 'original_device_class': , + 'original_device_class': , 'original_icon': None, 'original_name': 'Daily total water use', 'platform': 'hydrawise', @@ -212,7 +212,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ 'attribution': 'Data provided by hydrawise.com', - 'device_class': 'volume', + 'device_class': 'water', 'friendly_name': 'Home Controller Daily total water use', 'unit_of_measurement': , }), @@ -253,7 +253,7 @@ 'suggested_unit_of_measurement': , }), }), - 'original_device_class': , + 'original_device_class': , 'original_icon': None, 'original_name': 'Daily active water use', 'platform': 'hydrawise', @@ -269,7 +269,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ 'attribution': 'Data provided by hydrawise.com', - 'device_class': 'volume', + 'device_class': 'water', 'friendly_name': 'Zone One Daily active water use', 'unit_of_measurement': , }), @@ -464,7 +464,7 @@ 'suggested_unit_of_measurement': , }), }), - 'original_device_class': , + 'original_device_class': , 'original_icon': 'mdi:water-outline', 'original_name': 'Daily active water use', 'platform': 'hydrawise', @@ -480,7 +480,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ 'attribution': 'Data provided by hydrawise.com', - 'device_class': 'volume', + 'device_class': 'water', 'friendly_name': 'Zone Two Daily active water use', 'icon': 'mdi:water-outline', 'unit_of_measurement': ,