github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-08-15 18:25:44 +02:00
Code Releases 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_TYPE,
CONF_USERNAME,
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER,
DEVICE_CLASS_TEMPERATURE,
ENERGY_KILO_WATT_HOUR,
@ -204,6 +205,8 @@ class SAJsensor(SensorEntity):
"""Return the device class the sensor belongs to."""
if self.unit_of_measurement == POWER_WATT:
return DEVICE_CLASS_POWER
if self.unit_of_measurement == ENERGY_KILO_WATT_HOUR:
return DEVICE_CLASS_ENERGY
if (
self.unit_of_measurement == TEMP_CELSIUS
or self._sensor.unit == TEMP_FAHRENHEIT