From dd86de3255485c7904c7d82ad467b6b4fb0e5c18 Mon Sep 17 00:00:00 2001
From: Franck Nijhof <git@frenck.dev>
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,
     },