entity.py

Go to the documentation of this file.

"""An abstract class common to all EZVIZ entities."""
 
from __future__ import annotations
 
from typing import Any
 
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import DOMAIN, MANUFACTURER
from .coordinator import EzvizDataUpdateCoordinator
 
 
class EzvizEntity(CoordinatorEntity[EzvizDataUpdateCoordinator], Entity):
    """Generic entity encapsulating common features of EZVIZ device."""
 
    _attr_has_entity_name = True
 
    def __init__(
        self,
        coordinator: EzvizDataUpdateCoordinator,
        serial: str,
    ) -> None:
        """Initialize the entity."""
        super().__init__(coordinator)
        self._serial_serial = serial
        self._camera_name_camera_name = self.datadatadata["name"]
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, serial)},
            connections={
                (CONNECTION_NETWORK_MAC, self.datadatadata["mac_address"]),
            },
            manufacturer=MANUFACTURER,
            model=self.datadatadata["device_sub_category"],
            name=self.datadatadata["name"],
            sw_version=self.datadatadata["version"],
        )
 
    @property
    def data(self) -> dict[str, Any]:
        """Return coordinator data for this entity."""
        return self.coordinator.data[self._serial_serial]
 
 
class EzvizBaseEntity(Entity):
    """Generic entity for EZVIZ individual poll entities."""
 
    _attr_has_entity_name = True
 
    def __init__(
        self,
        coordinator: EzvizDataUpdateCoordinator,
        serial: str,
    ) -> None:
        """Initialize the entity."""
        self._serial_serial = serial
        self.coordinatorcoordinator = coordinator
        self._camera_name_camera_name = self.datadata["name"]
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, serial)},
            connections={
                (CONNECTION_NETWORK_MAC, self.datadata["mac_address"]),
            },
            manufacturer=MANUFACTURER,
            model=self.datadata["device_sub_category"],
            name=self.datadata["name"],
            sw_version=self.datadata["version"],
        )
 
    @property
    def data(self) -> dict[str, Any]:
        """Return coordinator data for this entity."""
        return self.coordinatorcoordinator.data[self._serial_serial]