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."""