Merge pull request #7 from ChristophCaina/patch-1

fix depricated ENERGY_KILO_WATT_HOUR

Author: nelbs

custom_components/solaredge_forecast/const.py

@@ -9,7 +9,7 @@
     SensorStateClass,
 )
 from homeassistant.const import (
-    ENERGY_KILO_WATT_HOUR
+    UnitOfEnergy
 )
 
 DOMAIN = "solaredge_forecast"
@@ -56,31 +56,31 @@ class SolaredgeForecastSensorEntityDescription(SensorEntityDescription):
         key="solaredge_forecast",
         name="Solar energy forecast",
         icon="mdi:solar-power",
-        native_unit_of_measurement=ENERGY_KILO_WATT_HOUR,
+        native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     SolaredgeForecastSensorEntityDescription(
         key="solaredge_produced",
         name="Solar energy produced",
         icon="mdi:solar-power",
-        native_unit_of_measurement=ENERGY_KILO_WATT_HOUR,
+        native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     SolaredgeForecastSensorEntityDescription(
         key="solaredge_estimated",
         name="Solar energy expected",
         icon="mdi:solar-power",
-        native_unit_of_measurement=ENERGY_KILO_WATT_HOUR,
+        native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     SolaredgeForecastSensorEntityDescription(
         key="solaredge_progress",
         name="Solar energy progress",
         icon="mdi:solar-power",
-        native_unit_of_measurement=ENERGY_KILO_WATT_HOUR,
+        native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),