Update for latest home assistant core

Author: werthdavid

custom_components/pulsatrix_local_mqtt/definitions/sensor.py

@@ -7,16 +7,12 @@
 import pytz
 from datetime import datetime
 from homeassistant.components.sensor import (
-    STATE_CLASS_MEASUREMENT,
-    STATE_CLASS_TOTAL_INCREASING,
+    SensorStateClass,
     SensorDeviceClass,
-    SensorEntityDescription,
+    SensorEntityDescription
 )
 from homeassistant.const import (
-    ELECTRIC_CURRENT_AMPERE,
-    ELECTRIC_POTENTIAL_VOLT,
-    FREQUENCY_HERTZ,
-    POWER_KILO_WATT, UnitOfEnergy,
+   UnitOfEnergy,UnitOfPower,UnitOfFrequency,UnitOfElectricCurrent,UnitOfElectricPotential
 )
 from homeassistant.helpers.entity import EntityCategory
 
@@ -102,7 +98,7 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="activePower",
         initial_value=0,
         device_class=SensorDeviceClass.POWER,
-        native_unit_of_measurement=POWER_KILO_WATT,
+        native_unit_of_measurement=UnitOfPower.KILO_WATT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=True,
         disabled=False,
@@ -115,7 +111,7 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="lastMeterValue",
         initial_value=0,
         device_class=SensorDeviceClass.ENERGY,
-        state_class=STATE_CLASS_TOTAL_INCREASING,
+        state_class=SensorStateClass.TOTAL_INCREASING,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=True,
@@ -128,7 +124,7 @@ def map_state_to_datetime(value, key, index) -> str:
         state=extract_json_amperage_kilo_watts,
         attribute="effectiveAmperageLimit",
         device_class=SensorDeviceClass.POWER,
-        native_unit_of_measurement=POWER_KILO_WATT,
+        native_unit_of_measurement=UnitOfPower.KILO_WATT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=True,
         disabled=False,
@@ -168,8 +164,8 @@ def map_state_to_datetime(value, key, index) -> str:
         state=extract_json_float,
         attribute="frequency",
         icon="mdi:current-ac",
-        native_unit_of_measurement=FREQUENCY_HERTZ,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfFrequency.HERTZ,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,
@@ -182,8 +178,8 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="voltage",
         attribute_index=0,
         device_class=SensorDeviceClass.VOLTAGE,
-        native_unit_of_measurement=ELECTRIC_POTENTIAL_VOLT,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,
@@ -196,8 +192,8 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="voltage",
         attribute_index=1,
         device_class=SensorDeviceClass.VOLTAGE,
-        native_unit_of_measurement=ELECTRIC_POTENTIAL_VOLT,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,
@@ -210,8 +206,8 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="voltage",
         attribute_index=2,
         device_class=SensorDeviceClass.VOLTAGE,
-        native_unit_of_measurement=ELECTRIC_POTENTIAL_VOLT,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,
@@ -224,8 +220,8 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="amperage",
         attribute_index=0,
         device_class=SensorDeviceClass.CURRENT,
-        native_unit_of_measurement=ELECTRIC_CURRENT_AMPERE,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,
@@ -238,8 +234,8 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="amperage",
         attribute_index=1,
         device_class=SensorDeviceClass.CURRENT,
-        native_unit_of_measurement=ELECTRIC_CURRENT_AMPERE,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,
@@ -252,8 +248,8 @@ def map_state_to_datetime(value, key, index) -> str:
         attribute="amperage",
         attribute_index=2,
         device_class=SensorDeviceClass.CURRENT,
-        native_unit_of_measurement=ELECTRIC_CURRENT_AMPERE,
-        state_class=STATE_CLASS_MEASUREMENT,
+        native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
+        state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         disabled=False,