entity.py

Go to the documentation of this file.

"""Support for EcoNet products."""
 
from homeassistant.core import callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
 
from .const import DOMAIN, PUSH_UPDATE
 
 
class EcoNetEntity(Entity):
    """Define a base EcoNet entity."""
 
    _attr_should_poll = False
 
    def __init__(self, econet):
        """Initialize."""
        self._econet_econet = econet
        self._attr_name_attr_name = econet.device_name
        self._attr_unique_id_attr_unique_id = f"{econet.device_id}_{econet.device_name}"
 
    async def async_added_to_hass(self):
        """Subscribe to device events."""
        await super().async_added_to_hass()
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(self.hasshass, PUSH_UPDATE, self.on_update_receivedon_update_received)
        )
 
    @callback
    def on_update_received(self):
        """Update was pushed from the ecoent API."""
        self.async_write_ha_stateasync_write_ha_state()
 
    @property
    def available(self):
        """Return if the device is online or not."""
        return self._econet_econet.connected
 
    @property
    def device_info(self) -> DeviceInfo:
        """Return device registry information for this entity."""
        return DeviceInfo(
            identifiers={(DOMAIN, self._econet_econet.device_id)},
            manufacturer="Rheem",
            name=self._econet_econet.device_name,
        )