Add files via upload

Compatibility fix for HA 2025.1.1
rospogrigio · Jan 10, 2025 · 24010e6 · 24010e6
1 parent 2b1f76b
commit 24010e6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/custom_components/airbnk_mqtt/__init__.py b/custom_components/airbnk_mqtt/__init__.py
@@ -5,7 +5,8 @@
 import voluptuous as vol
 
 from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
-from homeassistant.helpers.typing import HomeAssistantType
+from homeassistant.helpers.service import async_register_admin_service
+from homeassistant.core import HomeAssistant
 from homeassistant.const import CONF_TOKEN, SERVICE_RELOAD
 
 from .airbnk_api import AirbnkApi
@@ -57,7 +58,8 @@ async def _handle_reload(service):
         await asyncio.gather(*reload_tasks)
         _LOGGER.debug("RELOAD DONE")
 
-    hass.helpers.service.async_register_admin_service(
+    async_register_admin_service(
+        hass,
         DOMAIN,
         SERVICE_RELOAD,
         _handle_reload,
@@ -112,7 +114,7 @@ async def async_migrate_entry(hass, config_entry: ConfigEntry):
     return True
 
 
-async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry):
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     """Establish connection with Airbnk."""
 
     device_configs = entry.data[CONF_DEVICE_CONFIGS]
@@ -131,7 +133,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry):
     hass.data[DOMAIN] = {AIRBNK_DEVICES: lock_devices}
 
     for component in COMPONENT_TYPES:
-        await hass.config_entries.async_forward_entry_setup(entry, component)
+        await hass.config_entries.async_forward_entry_setups(entry, [component])
     return True
 
 

diff --git a/custom_components/airbnk_mqtt/binary_sensor.py b/custom_components/airbnk_mqtt/binary_sensor.py
@@ -5,9 +5,7 @@
 from homeassistant.components.binary_sensor import (
     BinarySensorEntity,
 )
-from homeassistant.components.sensor import (
-    SensorDeviceClass
-)
+from homeassistant.components.sensor import SensorDeviceClass
 
 from .const import (
     DOMAIN as AIRBNK_DOMAIN,

diff --git a/custom_components/airbnk_mqtt/const.py b/custom_components/airbnk_mqtt/const.py
@@ -69,24 +69,24 @@
     SENSOR_TYPE_BATTERY: {
         CONF_NAME: "Battery",
         CONF_TYPE: SENSOR_TYPE_BATTERY,
-        CONF_DEVICE_CLASS: DEVICE_CLASS_BATTERY,
+        CONF_DEVICE_CLASS: SensorDeviceClass.BATTERY,
         CONF_UNIT_OF_MEASUREMENT: PERCENTAGE,
     },
     SENSOR_TYPE_VOLTAGE: {
         CONF_NAME: "Battery voltage",
         CONF_TYPE: SENSOR_TYPE_VOLTAGE,
-        CONF_DEVICE_CLASS: DEVICE_CLASS_VOLTAGE,
-        CONF_UNIT_OF_MEASUREMENT: ELECTRIC_POTENTIAL_VOLT,
+        CONF_DEVICE_CLASS: SensorDeviceClass.VOLTAGE,
+        CONF_UNIT_OF_MEASUREMENT: UnitOfElectricPotential.VOLT,
     },
     SENSOR_TYPE_LAST_ADVERT: {
         CONF_NAME: "Time from last advert",
         CONF_TYPE: SENSOR_TYPE_LAST_ADVERT,
-        CONF_UNIT_OF_MEASUREMENT: TIME_SECONDS,
+        CONF_UNIT_OF_MEASUREMENT: UnitOfTime.SECONDS,
     },
     SENSOR_TYPE_SIGNAL_STRENGTH: {
         CONF_NAME: "Signal strength",
         CONF_TYPE: SENSOR_TYPE_SIGNAL_STRENGTH,
-        CONF_DEVICE_CLASS: DEVICE_CLASS_SIGNAL_STRENGTH,
+        CONF_DEVICE_CLASS: SensorDeviceClass.SIGNAL_STRENGTH,
         CONF_UNIT_OF_MEASUREMENT: SIGNAL_STRENGTH_DECIBELS,
     },
     SENSOR_TYPE_LOCK_EVENTS: {