From dd86de3255485c7904c7d82ad467b6b4fb0e5c18 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 11 Aug 2020 19:42:10 +0200 Subject: [PATCH] Add energy device class to Toon sensors (#38686) --- homeassistant/components/toon/const.py | 24 ++++++++++++++---------- 1 file changed, 14 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/toon/const.py b/homeassistant/components/toon/const.py index c814134f767..a5f57f95cd4 100644 --- a/homeassistant/components/toon/const.py +++ b/homeassistant/components/toon/const.py @@ -5,7 +5,11 @@ from homeassistant.components.binary_sensor import ( DEVICE_CLASS_CONNECTIVITY, DEVICE_CLASS_PROBLEM, ) -from homeassistant.components.sensor import DEVICE_CLASS_POWER, DEVICE_CLASS_TEMPERATURE +from homeassistant.components.sensor import ( + DEVICE_CLASS_ENERGY, + DEVICE_CLASS_POWER, + DEVICE_CLASS_TEMPERATURE, +) from homeassistant.const import ( ATTR_DEVICE_CLASS, ATTR_ICON, @@ -192,7 +196,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "day_average", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: False, }, @@ -210,7 +214,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "day_usage", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: True, }, @@ -219,7 +223,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "meter_high", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: False, }, @@ -228,7 +232,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "meter_low", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: False, }, @@ -246,7 +250,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "meter_produced_high", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: False, }, @@ -255,7 +259,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "meter_produced_low", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: False, }, @@ -291,7 +295,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "day_produced_solar", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:solar-power", ATTR_DEFAULT_ENABLED: True, }, @@ -300,7 +304,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "day_to_grid_usage", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:solar-power", ATTR_DEFAULT_ENABLED: False, }, @@ -309,7 +313,7 @@ SENSOR_ENTITIES = { ATTR_SECTION: "power_usage", ATTR_MEASUREMENT: "day_from_grid_usage", ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: None, + ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, ATTR_ICON: "mdi:power-plug", ATTR_DEFAULT_ENABLED: False, },