134 lines
4.2 KiB
Python
134 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()
|