core/homeassistant/components/skybell/entity.py

51 lines
1.8 KiB
Python

"""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_description = description
self._attr_unique_id = f"{self._device.device_id}_{description.key}"
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, self._device.device_id)},
manufacturer=DEFAULT_NAME,
model=self._device.type,
name=self._device.name.capitalize(),
sw_version=self._device.firmware_ver,
)
if self._device.mac:
self._attr_device_info[ATTR_CONNECTIONS] = {
(dr.CONNECTION_NETWORK_MAC, self._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()