75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
"""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
|
|
self._camera_name = self.data["name"]
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, serial)},
|
|
connections={
|
|
(CONNECTION_NETWORK_MAC, self.data["mac_address"]),
|
|
},
|
|
manufacturer=MANUFACTURER,
|
|
model=self.data["device_sub_category"],
|
|
name=self.data["name"],
|
|
sw_version=self.data["version"],
|
|
)
|
|
|
|
@property
|
|
def data(self) -> dict[str, Any]:
|
|
"""Return coordinator data for this entity."""
|
|
return self.coordinator.data[self._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
|
|
self.coordinator = coordinator
|
|
self._camera_name = self.data["name"]
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, serial)},
|
|
connections={
|
|
(CONNECTION_NETWORK_MAC, self.data["mac_address"]),
|
|
},
|
|
manufacturer=MANUFACTURER,
|
|
model=self.data["device_sub_category"],
|
|
name=self.data["name"],
|
|
sw_version=self.data["version"],
|
|
)
|
|
|
|
@property
|
|
def data(self) -> dict[str, Any]:
|
|
"""Return coordinator data for this entity."""
|
|
return self.coordinator.data[self._serial]
|