b8fdebd05c
* Add aircleaner and humidify service to nexia climate * These were removed from the original merge to reduce review scope * Additional tests for binary_sensor, sensor, and climate states * Switch to signals for services Get rid of everywhere we call device and change to zone or thermostat as it was too confusing Renames to make it clear that zone and thermostat are tightly coupled * Make scene activation responsive * no need to use update for only one key/value * stray comma * use async_call_later * its async, need ()s * cleaner * merge entity platform services testing branch
133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
"""The nexia integration base entity."""
|
|
|
|
from homeassistant.const import ATTR_ATTRIBUTION
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from .const import (
|
|
ATTRIBUTION,
|
|
DOMAIN,
|
|
MANUFACTURER,
|
|
SIGNAL_THERMOSTAT_UPDATE,
|
|
SIGNAL_ZONE_UPDATE,
|
|
)
|
|
|
|
|
|
class NexiaEntity(Entity):
|
|
"""Base class for nexia entities."""
|
|
|
|
def __init__(self, coordinator, name, unique_id):
|
|
"""Initialize the entity."""
|
|
super().__init__()
|
|
self._unique_id = unique_id
|
|
self._name = name
|
|
self._coordinator = coordinator
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return True if entity is available."""
|
|
return self._coordinator.last_update_success
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the unique id."""
|
|
return self._unique_id
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name."""
|
|
return self._name
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the device specific state attributes."""
|
|
return {
|
|
ATTR_ATTRIBUTION: ATTRIBUTION,
|
|
}
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Return False, updates are controlled via coordinator."""
|
|
return False
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Subscribe to updates."""
|
|
self._coordinator.async_add_listener(self.async_write_ha_state)
|
|
|
|
async def async_will_remove_from_hass(self):
|
|
"""Undo subscription."""
|
|
self._coordinator.async_remove_listener(self.async_write_ha_state)
|
|
|
|
|
|
class NexiaThermostatEntity(NexiaEntity):
|
|
"""Base class for nexia devices attached to a thermostat."""
|
|
|
|
def __init__(self, coordinator, thermostat, name, unique_id):
|
|
"""Initialize the entity."""
|
|
super().__init__(coordinator, name, unique_id)
|
|
self._thermostat = thermostat
|
|
self._thermostat_update_subscription = None
|
|
|
|
@property
|
|
def device_info(self):
|
|
"""Return the device_info of the device."""
|
|
return {
|
|
"identifiers": {(DOMAIN, self._thermostat.thermostat_id)},
|
|
"name": self._thermostat.get_name(),
|
|
"model": self._thermostat.get_model(),
|
|
"sw_version": self._thermostat.get_firmware(),
|
|
"manufacturer": MANUFACTURER,
|
|
}
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Listen for signals for services."""
|
|
await super().async_added_to_hass()
|
|
self._thermostat_update_subscription = async_dispatcher_connect(
|
|
self.hass,
|
|
f"{SIGNAL_THERMOSTAT_UPDATE}-{self._thermostat.thermostat_id}",
|
|
self.async_write_ha_state,
|
|
)
|
|
|
|
async def async_will_remove_from_hass(self):
|
|
"""Unsub from signals for services."""
|
|
await super().async_will_remove_from_hass()
|
|
if self._thermostat_update_subscription:
|
|
self._thermostat_update_subscription()
|
|
|
|
|
|
class NexiaThermostatZoneEntity(NexiaThermostatEntity):
|
|
"""Base class for nexia devices attached to a thermostat."""
|
|
|
|
def __init__(self, coordinator, zone, name, unique_id):
|
|
"""Initialize the entity."""
|
|
super().__init__(coordinator, zone.thermostat, name, unique_id)
|
|
self._zone = zone
|
|
self._zone_update_subscription = None
|
|
|
|
@property
|
|
def device_info(self):
|
|
"""Return the device_info of the device."""
|
|
data = super().device_info
|
|
data.update(
|
|
{
|
|
"identifiers": {(DOMAIN, self._zone.zone_id)},
|
|
"name": self._zone.get_name(),
|
|
"via_device": (DOMAIN, self._zone.thermostat.thermostat_id),
|
|
}
|
|
)
|
|
return data
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Listen for signals for services."""
|
|
await super().async_added_to_hass()
|
|
self._zone_update_subscription = async_dispatcher_connect(
|
|
self.hass,
|
|
f"{SIGNAL_ZONE_UPDATE}-{self._zone.zone_id}",
|
|
self.async_write_ha_state,
|
|
)
|
|
|
|
async def async_will_remove_from_hass(self):
|
|
"""Unsub from signals for services."""
|
|
await super().async_will_remove_from_hass()
|
|
if self._zone_update_subscription:
|
|
self._zone_update_subscription()
|