Updating deprecated constants (#29)

augustas2 · Jan 25, 2024 · 138aaeb · 138aaeb
1 parent 7f8f5f4
commit 138aaeb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/custom_components/eldes_alarm/alarm_control_panel.py b/custom_components/eldes_alarm/alarm_control_panel.py
@@ -1,13 +1,12 @@
 """Interfaces with Eldes control panels."""
 import logging
 
-from homeassistant.components.alarm_control_panel import AlarmControlPanelEntity
+from homeassistant.components.alarm_control_panel import (
+    AlarmControlPanelEntity,
+    AlarmControlPanelEntityFeature
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
-from homeassistant.components.alarm_control_panel.const import (
-    SUPPORT_ALARM_ARM_AWAY,
-    SUPPORT_ALARM_ARM_HOME
-)
 from homeassistant.const import (
     STATE_ALARM_ARMED_AWAY,
     STATE_ALARM_ARMED_HOME,
@@ -71,7 +70,7 @@ def extra_state_attributes(self):
     @property
     def supported_features(self):
         """Return the list of supported features."""
-        return SUPPORT_ALARM_ARM_AWAY | SUPPORT_ALARM_ARM_HOME
+        return AlarmControlPanelEntityFeature.ARM_AWAY | AlarmControlPanelEntityFeature.ARM_HOME
 
     async def async_alarm_disarm(self, code=None):
         """Send disarm command."""

diff --git a/custom_components/eldes_alarm/sensor.py b/custom_components/eldes_alarm/sensor.py
@@ -5,7 +5,10 @@
 from homeassistant.components.sensor import SensorEntity, SensorDeviceClass
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
-from homeassistant.const import PERCENTAGE, TEMP_CELSIUS
+from homeassistant.const import (
+    PERCENTAGE,
+    UnitOfTemperature
+)
 
 from .const import (
     DATA_CLIENT,
@@ -141,7 +144,7 @@ def device_class(self):
     @property
     def native_unit_of_measurement(self):
         """Return the unit of measurement."""
-        return TEMP_CELSIUS
+        return UnitOfTemperature.CELSIUS
 
     @property
     def native_value(self):