evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update coolmaster to use CoordinatorEntity (#39465)

Browse source
This commit is contained in:
J. Nick Koston 2020-08-30 13:18:19 -05:00 • committed by GitHub
parent 80910b4f9a
commit c4f5a05b27
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
homeassistant/components/coolmaster/climate.py
View file

@ -14,6 +14,7 @@ from homeassistant.components.climate.const import (
SUPPORT_TARGET_TEMPERATURE, SUPPORT_TARGET_TEMPERATURE,
) )
from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_SUPPORTED_MODES, DATA_COORDINATOR, DATA_INFO, DOMAIN from .const import CONF_SUPPORTED_MODES, DATA_COORDINATOR, DATA_INFO, DOMAIN
@ -54,44 +55,27 @@ async def async_setup_entry(hass, config_entry, async_add_devices):
async_add_devices(all_devices) async_add_devices(all_devices)
class CoolmasterClimate(ClimateEntity): class CoolmasterClimate(CoordinatorEntity, ClimateEntity):
"""Representation of a coolmaster climate device.""" """Representation of a coolmaster climate device."""
def __init__(self, coordinator, unit_id, unit, supported_modes, info): def __init__(self, coordinator, unit_id, unit, supported_modes, info):
"""Initialize the climate device.""" """Initialize the climate device."""
self._coordinator = coordinator super().__init__(coordinator)
self._unit_id = unit_id self._unit_id = unit_id
self._unit = unit self._unit = unit
self._hvac_modes = supported_modes self._hvac_modes = supported_modes
self._info = info self._info = info
@property
def should_poll(self):
"""No need to poll. Coordinator notifies entity of updates."""
return False
@property
def available(self):
"""Return if entity is available."""
return self._coordinator.last_update_success
def _refresh_from_coordinator(self): def _refresh_from_coordinator(self):
self._unit = self._coordinator.data[self._unit_id] self._unit = self.coordinator.data[self._unit_id]
self.async_write_ha_state() self.async_write_ha_state()
async def async_added_to_hass(self): async def async_added_to_hass(self):
"""When entity is added to hass.""" """When entity is added to hass."""
self.async_on_remove( self.async_on_remove(
self._coordinator.async_add_listener(self._refresh_from_coordinator) self.coordinator.async_add_listener(self._refresh_from_coordinator)
) )
async def async_update(self):
"""Update the entity.
Only used by the generic entity update service.
"""
await self._coordinator.async_request_refresh()
@property @property
def device_info(self): def device_info(self):
"""Return device info for this device.""" """Return device info for this device."""