From a1d48ad1ecbf983b0649953985180a04f93fb879 Mon Sep 17 00:00:00 2001 From: springstan <46536646+springstan@users.noreply.github.com> Date: Sun, 30 Aug 2020 16:38:15 +0200 Subject: [PATCH] Update juicenet to use CoordinatorEntity (#39424) --- homeassistant/components/juicenet/entity.py | 26 +++------------------ 1 file changed, 3 insertions(+), 23 deletions(-) diff --git a/homeassistant/components/juicenet/entity.py b/homeassistant/components/juicenet/entity.py index fe81f242cd0..759979c5f11 100644 --- a/homeassistant/components/juicenet/entity.py +++ b/homeassistant/components/juicenet/entity.py @@ -1,44 +1,24 @@ """Adapter to wrap the pyjuicenet api for home assistant.""" -from homeassistant.helpers.entity import Entity +from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN -class JuiceNetDevice(Entity): +class JuiceNetDevice(CoordinatorEntity): """Represent a base JuiceNet device.""" def __init__(self, device, sensor_type, coordinator): """Initialise the sensor.""" + super().__init__(coordinator) self.device = device self.type = sensor_type - self.coordinator = coordinator @property def name(self): """Return the name of the device.""" return self.device.name - @property - def should_poll(self): - """Return False, updates are controlled via coordinator.""" - return False - - @property - def available(self): - """Return True if entity is available.""" - return self.coordinator.last_update_success - - async def async_update(self): - """Update the entity.""" - await self.coordinator.async_request_refresh() - - async def async_added_to_hass(self): - """Subscribe to updates.""" - self.async_on_remove( - self.coordinator.async_add_listener(self.async_write_ha_state) - ) - @property def unique_id(self): """Return a unique ID."""