entity.py

Go to the documentation of this file.

"""Entity representing a Skybell HD Doorbell."""
 
from __future__ import annotations
 
from aioskybell import SkybellDevice
 
from homeassistant.const import ATTR_CONNECTIONS
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import DEFAULT_NAME, DOMAIN
from .coordinator import SkybellDataUpdateCoordinator
 
 
class SkybellEntity(CoordinatorEntity[SkybellDataUpdateCoordinator]):
    """An HA implementation for Skybell entity."""
 
    _attr_attribution = "Data provided by Skybell.com"
    _attr_has_entity_name = True
 
    def __init__(
        self, coordinator: SkybellDataUpdateCoordinator, description: EntityDescription
    ) -> None:
        """Initialize a SkyBell entity."""
        super().__init__(coordinator)
        self.entity_descriptionentity_description = description
        self._attr_unique_id_attr_unique_id = f"{self._device.device_id}_{description.key}"
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, self._device_device.device_id)},
            manufacturer=DEFAULT_NAME,
            model=self._device_device.type,
            name=self._device_device.name.capitalize(),
            sw_version=self._device_device.firmware_ver,
        )
        if self._device_device.mac:
            self._attr_device_info_attr_device_info[ATTR_CONNECTIONS] = {
                (dr.CONNECTION_NETWORK_MAC, self._device_device.mac)
            }
 
    @property
    def _device(self) -> SkybellDevice:
        """Return the device."""
        return self.coordinator.device
 
    async def async_added_to_hass(self) -> None:
        """When entity is added to hass."""
        await super().async_added_to_hass()
        self._handle_coordinator_update()