entity.py

Go to the documentation of this file.

"""The Z-Wave-Me WS integration."""
 
from zwave_me_ws import ZWaveMeData
 
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
 
 
class ZWaveMeEntity(Entity):
    """Representation of a ZWaveMe device."""
 
    def __init__(self, controller, device):
        """Initialize the device."""
        self.controllercontroller = controller
        self.devicedevice = device
        self._attr_name_attr_name = device.title
        self._attr_unique_id: str = (
            f"{self.controller.config.unique_id}-{self.device.id}"
        )
        self._attr_should_poll_attr_should_poll = False
 
    @property
    def device_info(self) -> DeviceInfo:
        """Return device specific attributes."""
        return DeviceInfo(
            identifiers={(DOMAIN, self.devicedevice.deviceIdentifier)},
            name=self._attr_name_attr_name,
            manufacturer=self.devicedevice.manufacturer,
            sw_version=self.devicedevice.firmware,
            suggested_area=self.devicedevice.locationName,
        )
 
    async def async_added_to_hass(self) -> None:
        """Connect to an updater."""
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(
                self.hasshass, f"ZWAVE_ME_INFO_{self.device.id}", self.get_new_dataget_new_data
            )
        )
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(
                self.hasshass,
                f"ZWAVE_ME_UNAVAILABLE_{self.device.id}",
                self.set_unavailable_statusset_unavailable_status,
            )
        )
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(
                self.hasshass, f"ZWAVE_ME_DESTROY_{self.device.id}", self.delete_entitydelete_entity
            )
        )
 
    @callback
    def get_new_data(self, new_data: ZWaveMeData) -> None:
        """Update info in the HAss."""
        self.devicedevice = new_data
        self._attr_available_attr_available = not new_data.isFailed
        self.async_write_ha_stateasync_write_ha_state()
 
    @callback
    def set_unavailable_status(self):
        """Update status in the HAss."""
        self._attr_available_attr_available = False
        self.async_write_ha_stateasync_write_ha_state()
 
    @callback
    def delete_entity(self) -> None:
        """Remove this entity."""
        self.hasshass.async_create_task(self.async_removeasync_remove(force_remove=True))