From 9d3661647767666fdbbadb393ddda336e8f510aa Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 15 Dec 2021 12:12:54 +0100 Subject: [PATCH] Use new enums in kostal_plenticore (#61871) Co-authored-by: epenet --- .../components/kostal_plenticore/const.py | 182 +++++++++--------- 1 file changed, 93 insertions(+), 89 deletions(-) diff --git a/homeassistant/components/kostal_plenticore/const.py b/homeassistant/components/kostal_plenticore/const.py index ebed1ddcb74..d4e1fa47a8b 100644 --- a/homeassistant/components/kostal_plenticore/const.py +++ b/homeassistant/components/kostal_plenticore/const.py @@ -3,18 +3,13 @@ from typing import NamedTuple from homeassistant.components.sensor import ( ATTR_STATE_CLASS, - STATE_CLASS_MEASUREMENT, - STATE_CLASS_TOTAL_INCREASING, + SensorDeviceClass, + SensorStateClass, ) from homeassistant.const import ( ATTR_DEVICE_CLASS, ATTR_ICON, ATTR_UNIT_OF_MEASUREMENT, - DEVICE_CLASS_BATTERY, - DEVICE_CLASS_CURRENT, - DEVICE_CLASS_ENERGY, - DEVICE_CLASS_POWER, - DEVICE_CLASS_VOLTAGE, ELECTRIC_CURRENT_AMPERE, ELECTRIC_POTENTIAL_VOLT, ENERGY_KILO_WATT_HOUR, @@ -48,9 +43,9 @@ SENSOR_PROCESS_DATA = [ "Solar Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, ATTR_ENABLED_DEFAULT: True, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -60,9 +55,9 @@ SENSOR_PROCESS_DATA = [ "Grid Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, ATTR_ENABLED_DEFAULT: True, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -70,7 +65,10 @@ SENSOR_PROCESS_DATA = [ "devices:local", "HomeBat_P", "Home Power from Battery", - {ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER}, + { + ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + }, "format_round", ), ( @@ -79,8 +77,8 @@ SENSOR_PROCESS_DATA = [ "Home Power from Grid", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -90,8 +88,8 @@ SENSOR_PROCESS_DATA = [ "Home Power from Own", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -101,8 +99,8 @@ SENSOR_PROCESS_DATA = [ "Home Power from PV", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -112,8 +110,8 @@ SENSOR_PROCESS_DATA = [ "Home Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -123,9 +121,9 @@ SENSOR_PROCESS_DATA = [ "AC Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, ATTR_ENABLED_DEFAULT: True, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -135,8 +133,8 @@ SENSOR_PROCESS_DATA = [ "DC1 Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -146,8 +144,8 @@ SENSOR_PROCESS_DATA = [ "DC1 Voltage", { ATTR_UNIT_OF_MEASUREMENT: ELECTRIC_POTENTIAL_VOLT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_VOLTAGE, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.VOLTAGE, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -157,8 +155,8 @@ SENSOR_PROCESS_DATA = [ "DC1 Current", { ATTR_UNIT_OF_MEASUREMENT: ELECTRIC_CURRENT_AMPERE, - ATTR_DEVICE_CLASS: DEVICE_CLASS_CURRENT, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.CURRENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_float", ), @@ -168,8 +166,8 @@ SENSOR_PROCESS_DATA = [ "DC2 Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -179,8 +177,8 @@ SENSOR_PROCESS_DATA = [ "DC2 Voltage", { ATTR_UNIT_OF_MEASUREMENT: ELECTRIC_POTENTIAL_VOLT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_VOLTAGE, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.VOLTAGE, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -190,8 +188,8 @@ SENSOR_PROCESS_DATA = [ "DC2 Current", { ATTR_UNIT_OF_MEASUREMENT: ELECTRIC_CURRENT_AMPERE, - ATTR_DEVICE_CLASS: DEVICE_CLASS_CURRENT, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.CURRENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_float", ), @@ -201,8 +199,8 @@ SENSOR_PROCESS_DATA = [ "DC3 Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -212,8 +210,8 @@ SENSOR_PROCESS_DATA = [ "DC3 Voltage", { ATTR_UNIT_OF_MEASUREMENT: ELECTRIC_POTENTIAL_VOLT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_VOLTAGE, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.VOLTAGE, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -223,8 +221,8 @@ SENSOR_PROCESS_DATA = [ "DC3 Current", { ATTR_UNIT_OF_MEASUREMENT: ELECTRIC_CURRENT_AMPERE, - ATTR_DEVICE_CLASS: DEVICE_CLASS_CURRENT, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.CURRENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_float", ), @@ -234,8 +232,8 @@ SENSOR_PROCESS_DATA = [ "PV to Battery Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -250,7 +248,7 @@ SENSOR_PROCESS_DATA = [ "devices:local:battery", "Cycles", "Battery Cycles", - {ATTR_ICON: "mdi:recycle", ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT}, + {ATTR_ICON: "mdi:recycle", ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT}, "format_round", ), ( @@ -259,8 +257,8 @@ SENSOR_PROCESS_DATA = [ "Battery Power", { ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, - ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER, - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -268,7 +266,10 @@ SENSOR_PROCESS_DATA = [ "devices:local:battery", "SoC", "Battery SoC", - {ATTR_UNIT_OF_MEASUREMENT: PERCENTAGE, ATTR_DEVICE_CLASS: DEVICE_CLASS_BATTERY}, + { + ATTR_UNIT_OF_MEASUREMENT: PERCENTAGE, + ATTR_DEVICE_CLASS: SensorDeviceClass.BATTERY, + }, "format_round", ), ( @@ -292,7 +293,7 @@ SENSOR_PROCESS_DATA = [ { ATTR_UNIT_OF_MEASUREMENT: PERCENTAGE, ATTR_ICON: "mdi:chart-donut", - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -324,7 +325,7 @@ SENSOR_PROCESS_DATA = [ { ATTR_UNIT_OF_MEASUREMENT: PERCENTAGE, ATTR_ICON: "mdi:chart-donut", - ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT, + ATTR_STATE_CLASS: SensorStateClass.MEASUREMENT, }, "format_round", ), @@ -341,7 +342,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -351,7 +352,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -361,7 +362,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -371,8 +372,8 @@ SENSOR_PROCESS_DATA = [ "Home Consumption Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -382,7 +383,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Battery Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -392,7 +393,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Battery Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -402,7 +403,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Battery Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -412,8 +413,8 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Battery Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -423,7 +424,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Grid Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -433,7 +434,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Grid Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -443,7 +444,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Grid Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -453,8 +454,8 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from Grid Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -464,7 +465,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from PV Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -474,7 +475,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from PV Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -484,7 +485,7 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from PV Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -494,8 +495,8 @@ SENSOR_PROCESS_DATA = [ "Home Consumption from PV Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -505,7 +506,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV1 Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -515,7 +516,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV1 Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -525,7 +526,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV1 Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -535,8 +536,8 @@ SENSOR_PROCESS_DATA = [ "Energy PV1 Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -546,7 +547,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV2 Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -556,7 +557,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV2 Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -566,7 +567,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV2 Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -576,8 +577,8 @@ SENSOR_PROCESS_DATA = [ "Energy PV2 Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -587,7 +588,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV3 Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -597,7 +598,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV3 Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -607,7 +608,7 @@ SENSOR_PROCESS_DATA = [ "Energy PV3 Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -617,8 +618,8 @@ SENSOR_PROCESS_DATA = [ "Energy PV3 Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -628,7 +629,7 @@ SENSOR_PROCESS_DATA = [ "Energy Yield Day", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, ATTR_ENABLED_DEFAULT: True, }, "format_energy", @@ -639,7 +640,7 @@ SENSOR_PROCESS_DATA = [ "Energy Yield Month", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -649,7 +650,7 @@ SENSOR_PROCESS_DATA = [ "Energy Yield Year", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, }, "format_energy", ), @@ -659,8 +660,8 @@ SENSOR_PROCESS_DATA = [ "Energy Yield Total", { ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR, - ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY, - ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING, + ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY, + ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING, }, "format_energy", ), @@ -679,7 +680,10 @@ SENSOR_SETTINGS_DATA = [ "devices:local", "Battery:MinHomeComsumption", "Battery min Home Consumption", - {ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER}, + { + ATTR_UNIT_OF_MEASUREMENT: POWER_WATT, + ATTR_DEVICE_CLASS: SensorDeviceClass.POWER, + }, "format_round", ), (