Merge pull request #12 from 0xAlon/update_deprecated_functionality

Fix deprecated functionality

Author: 0xAlon

custom_components/dolphin/__init__.py

@@ -39,11 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
         hass.data.setdefault(DOMAIN, {})
         hass.data[DOMAIN][entry.entry_id] = coordinator
 
-        # Set up all platforms for this device/entry.
-        for platform in PLATFORMS:
-            hass.async_create_task(
-                hass.config_entries.async_forward_entry_setup(entry, platform)
-            )
+        await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
     return True
 

custom_components/dolphin/climate.py

@@ -2,8 +2,9 @@
 from typing import Any
 from typing import Callable, List
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import UnitOfTemperature
 from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.typing import HomeAssistantType
+import homeassistant.core
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from .coordinator import UpdateCoordinator
 from homeassistant.helpers.entity import DeviceInfo
@@ -14,7 +15,6 @@
     HVACMode,
     ClimateEntity,
 )
-from homeassistant.const import TEMP_CELSIUS
 from homeassistant.components.climate.const import HVACMode
 
 OPERATION_LIST = [HVACMode.HEAT, HVACMode.OFF]
@@ -25,7 +25,7 @@
 
 
 async def async_setup_entry(
-        hass: HomeAssistantType,
+        hass: homeassistant.core.HomeAssistant,
         entry: ConfigEntry,
         async_add_entities: Callable[[List[Entity], bool], None],
 ) -> None:
@@ -40,7 +40,10 @@ async def async_setup_entry(
 
 class DolphinWaterHeater(CoordinatorEntity, ClimateEntity):
     _attr_hvac_modes = OPERATION_LIST
-    _attr_temperature_unit = TEMP_CELSIUS
+    _attr_temperature_unit = UnitOfTemperature.CELSIUS
+    _enable_turn_on_off_backwards_compatibility = False
+    _attr_supported_features = (ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.TURN_OFF |
+                                ClimateEntityFeature.TURN_ON)
 
     def __init__(self, hass, coordinator, device):
         CoordinatorEntity.__init__(self, coordinator)
@@ -49,7 +52,7 @@ def __init__(self, hass, coordinator, device):
 
         self._name = device
 
-        self._unit = TEMP_CELSIUS
+        self._unit = UnitOfTemperature.CELSIUS
         self._current_temperature = None
         self._attr_target_temperature_step = 1
 
@@ -132,6 +135,9 @@ def device_info(self) -> DeviceInfo:
             name=self.name,
         )
 
+    async def async_turn_off(self) -> None:
+        await self.async_set_hvac_mode(HVACMode.OFF)
+
     async def async_set_temperature(self, **kwargs: Any) -> None:
         """Set temperature."""
 

custom_components/dolphin/config_flow.py

@@ -5,6 +5,6 @@
   "documentation": "https://github.com/0xAlon/dolphin",
   "issue_tracker": "https://github.com/0xAlon/dolphin",
   "codeowners": ["@0xAlon"],
-  "version": "1.0.0",
+  "version": "1.0.1",
   "iot_class": "cloud_polling"
 }

custom_components/dolphin/const.py

@@ -3,7 +3,7 @@
 from homeassistant.components.sensor import SensorEntity
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.typing import HomeAssistantType
+import homeassistant.core
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from .coordinator import UpdateCoordinator
 from homeassistant.helpers.entity import DeviceInfo, async_generate_entity_id
@@ -17,7 +17,7 @@
 
 
 async def async_setup_entry(
-        hass: HomeAssistantType,
+        hass: homeassistant.core.HomeAssistant,
         entry: ConfigEntry,
         async_add_entities: Callable[[List[Entity], bool], None],
 ) -> None:

custom_components/dolphin/coordinator.py

@@ -3,7 +3,7 @@
 from homeassistant.components.switch import SwitchEntity
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.typing import HomeAssistantType
+import homeassistant.core
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from .coordinator import UpdateCoordinator
 from homeassistant.helpers.entity import DeviceInfo, async_generate_entity_id
@@ -17,7 +17,7 @@
 
 
 async def async_setup_entry(
-        hass: HomeAssistantType,
+        hass: homeassistant.core.HomeAssistant,
         entry: ConfigEntry,
         async_add_entities: Callable[[List[Entity], bool], None],
 ) -> None: