Fix deprecated consts and error linked to async call within thread (#46)

ssenart · May 7, 2024 · b316d3d · b316d3d
1 parent 36c2e4f
commit b316d3d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/custom_components/gazpar/sensor.py b/custom_components/gazpar/sensor.py
@@ -15,7 +15,7 @@
 import voluptuous as vol
 
 from homeassistant.components.sensor import PLATFORM_SCHEMA
-from homeassistant.const import CONF_NAME, CONF_PASSWORD, CONF_USERNAME, CONF_SCAN_INTERVAL, ENERGY_KILO_WATT_HOUR
+from homeassistant.const import CONF_NAME, CONF_PASSWORD, CONF_USERNAME, CONF_SCAN_INTERVAL, UnitOfEnergy
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.entity import Entity
 from homeassistant.helpers.event import track_time_interval, call_later
@@ -108,7 +108,7 @@ def __init__(self, hass, name: str, username: str, password: str, pceIdentifier:
         self._errorMessages = []
 
         self.sensors.append(
-            GazparSensor(name, PropertyName.ENERGY.value, ENERGY_KILO_WATT_HOUR, self))
+            GazparSensor(name, PropertyName.ENERGY.value, UnitOfEnergy.KILO_WATT_HOUR, self))
 
         if hass is not None:
             call_later(hass, 5, self.update_gazpar_data)
@@ -150,7 +150,7 @@ def update_gazpar_data(self, event_time):
 
         if event_time is not None:
             for sensor in self.sensors:
-                sensor.async_schedule_update_ha_state(True)
+                sensor.schedule_update_ha_state(True)
             _LOGGER.debug("HA notified that new data are available")
 
     # ----------------------------------

diff --git a/custom_components/gazpar/util.py b/custom_components/gazpar/util.py
@@ -1,6 +1,6 @@
 from pygazpar.enum import PropertyName, Frequency
-from homeassistant.const import CONF_USERNAME, ATTR_ATTRIBUTION, ATTR_UNIT_OF_MEASUREMENT, ATTR_FRIENDLY_NAME, ATTR_ICON, ATTR_DEVICE_CLASS, ENERGY_KILO_WATT_HOUR, DEVICE_CLASS_ENERGY
-from homeassistant.components.sensor import ATTR_STATE_CLASS, STATE_CLASS_TOTAL_INCREASING
+from homeassistant.const import CONF_USERNAME, ATTR_ATTRIBUTION, ATTR_UNIT_OF_MEASUREMENT, ATTR_FRIENDLY_NAME, ATTR_ICON, ATTR_DEVICE_CLASS, UnitOfEnergy
+from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass, SensorDeviceClass
 from typing import Any, Union
 
 from custom_components.gazpar.manifest import Manifest
@@ -57,11 +57,11 @@ def toAttributes(username: str, pceIdentifier: str, pygazparData: dict[str, list
             ATTR_VERSION: Manifest.version(),
             CONF_USERNAME: username,
             ATTR_PCE: pceIdentifier,
-            ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR,
+            ATTR_UNIT_OF_MEASUREMENT: UnitOfEnergy.KILO_WATT_HOUR,
             ATTR_FRIENDLY_NAME: SENSOR_FRIENDLY_NAME,
             ATTR_ICON: ICON_GAS,
-            ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY,
-            ATTR_STATE_CLASS: STATE_CLASS_TOTAL_INCREASING,
+            ATTR_DEVICE_CLASS: SensorDeviceClass.ENERGY,
+            ATTR_STATE_CLASS: SensorStateClass.TOTAL_INCREASING,
             ATTR_ERROR_MESSAGES: errorMessages,
             str(Frequency.HOURLY): {},
             str(Frequency.DAILY): {},