jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing device class to SAJ energy sensors (#54048)

Browse Source
This commit is contained in:
Franck Nijhof 2021-08-05 12:12:06 +02:00 committed by GitHub
parent 29e604bd22
commit 1bc3c743db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
homeassistant/components/saj/sensor.py
View File

@ -18,6 +18,7 @@ from homeassistant.const import (
CONF_PASSWORD, CONF_PASSWORD,
CONF_TYPE, CONF_TYPE,
CONF_USERNAME, CONF_USERNAME,
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER, DEVICE_CLASS_POWER,
DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_TEMPERATURE,
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
@ -204,6 +205,8 @@ class SAJsensor(SensorEntity):
"""Return the device class the sensor belongs to.""" """Return the device class the sensor belongs to."""
if self.unit_of_measurement == POWER_WATT: if self.unit_of_measurement == POWER_WATT:
return DEVICE_CLASS_POWER return DEVICE_CLASS_POWER
if self.unit_of_measurement == ENERGY_KILO_WATT_HOUR:
return DEVICE_CLASS_ENERGY
if ( if (
self.unit_of_measurement == TEMP_CELSIUS self.unit_of_measurement == TEMP_CELSIUS
or self._sensor.unit == TEMP_FAHRENHEIT or self._sensor.unit == TEMP_FAHRENHEIT