From 7bed868d603192bf6629c16d8a452f8b1dd56f4b Mon Sep 17 00:00:00 2001
From: springstan <46536646+springstan@users.noreply.github.com>
Date: Sun, 30 Aug 2020 18:10:28 +0200
Subject: [PATCH] Update wolflink to use CoordinatorEntity (#39444)

---
 homeassistant/components/wolflink/sensor.py | 27 +++------------------
 1 file changed, 3 insertions(+), 24 deletions(-)

diff --git a/homeassistant/components/wolflink/sensor.py b/homeassistant/components/wolflink/sensor.py
index a9deced9e91..97f48e27988 100644
--- a/homeassistant/components/wolflink/sensor.py
+++ b/homeassistant/components/wolflink/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
     TEMP_CELSIUS,
     TIME_HOURS,
 )
-from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -55,12 +55,12 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     async_add_entities(entities, True)
 
 
-class WolfLinkSensor(Entity):
+class WolfLinkSensor(CoordinatorEntity):
     """Base class for all Wolf entities."""
 
     def __init__(self, coordinator, wolf_object: Parameter, device_id):
         """Initialize."""
-        self.coordinator = coordinator
+        super().__init__(coordinator)
         self.wolf_object = wolf_object
         self.device_id = device_id
 
@@ -88,27 +88,6 @@ class WolfLinkSensor(Entity):
         """Return a unique_id for this entity."""
         return f"{self.device_id}:{self.wolf_object.parameter_id}"
 
-    @property
-    def available(self):
-        """Return True if entity is available."""
-        return self.coordinator.last_update_success
-
-    @property
-    def should_poll(self):
-        """No need to poll. Coordinator notifies entity of updates."""
-        return False
-
-    async def async_added_to_hass(self):
-        """When entity is added to hass."""
-        self.async_on_remove(
-            self.coordinator.async_add_listener(self.async_write_ha_state)
-        )
-
-    async def async_update(self):
-        """Update the sensor."""
-        await self.coordinator.async_request_refresh()
-        _LOGGER.debug("Updating %s", self.coordinator.data[self.wolf_object.value_id])
-
 
 class WolfLinkHours(WolfLinkSensor):
     """Class for hour based entities."""