From ff3a1f2050662298b258ada2b9cebab0e83d1103 Mon Sep 17 00:00:00 2001 From: ErnstEeldert Date: Wed, 13 Jan 2021 13:45:11 +0100 Subject: [PATCH] Add device class attribute to tado humidity sensor state (#45084) * add device class attribute to humidity sensor state * * explict return none * use const for device class value * removed unnecessary icon definitions --- homeassistant/components/tado/sensor.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/tado/sensor.py b/homeassistant/components/tado/sensor.py index f9a924b52c6..6613de82bff 100644 --- a/homeassistant/components/tado/sensor.py +++ b/homeassistant/components/tado/sensor.py @@ -2,7 +2,12 @@ import logging from homeassistant.config_entries import ConfigEntry -from homeassistant.const import PERCENTAGE, TEMP_CELSIUS +from homeassistant.const import ( + DEVICE_CLASS_HUMIDITY, + DEVICE_CLASS_TEMPERATURE, + PERCENTAGE, + TEMP_CELSIUS, +) from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.entity import Entity @@ -126,12 +131,13 @@ class TadoZoneSensor(TadoZoneEntity, Entity): return None @property - def icon(self): - """Icon for the sensor.""" - if self.zone_variable == "temperature": - return "mdi:thermometer" + def device_class(self): + """Return the device class.""" if self.zone_variable == "humidity": - return "mdi:water-percent" + return DEVICE_CLASS_HUMIDITY + if self.zone_variable == "temperature": + return DEVICE_CLASS_TEMPERATURE + return None @callback def _async_update_callback(self):