93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
"""Reolink parent entity class."""
|
|
from __future__ import annotations
|
|
|
|
from typing import TypeVar
|
|
|
|
from reolink_aio.api import DUAL_LENS_MODELS
|
|
|
|
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
|
|
from homeassistant.helpers.entity import DeviceInfo
|
|
from homeassistant.helpers.update_coordinator import (
|
|
CoordinatorEntity,
|
|
DataUpdateCoordinator,
|
|
)
|
|
|
|
from . import ReolinkData
|
|
from .const import DOMAIN
|
|
|
|
_T = TypeVar("_T")
|
|
|
|
|
|
class ReolinkBaseCoordinatorEntity(CoordinatorEntity[DataUpdateCoordinator[_T]]):
|
|
"""Parent class fo Reolink entities."""
|
|
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(
|
|
self,
|
|
reolink_data: ReolinkData,
|
|
coordinator: DataUpdateCoordinator[_T],
|
|
) -> None:
|
|
"""Initialize ReolinkBaseCoordinatorEntity."""
|
|
super().__init__(coordinator)
|
|
|
|
self._host = reolink_data.host
|
|
|
|
http_s = "https" if self._host.api.use_https else "http"
|
|
self._conf_url = f"{http_s}://{self._host.api.host}:{self._host.api.port}"
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, self._host.unique_id)},
|
|
connections={(CONNECTION_NETWORK_MAC, self._host.api.mac_address)},
|
|
name=self._host.api.nvr_name,
|
|
model=self._host.api.model,
|
|
manufacturer=self._host.api.manufacturer,
|
|
hw_version=self._host.api.hardware_version,
|
|
sw_version=self._host.api.sw_version,
|
|
configuration_url=self._conf_url,
|
|
)
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return True if entity is available."""
|
|
return self._host.api.session_active and super().available
|
|
|
|
|
|
class ReolinkHostCoordinatorEntity(ReolinkBaseCoordinatorEntity[None]):
|
|
"""Parent class for entities that control the Reolink NVR itself, without a channel.
|
|
|
|
A camera connected directly to HomeAssistant without using a NVR is in the reolink API
|
|
basically a NVR with a single channel that has the camera connected to that channel.
|
|
"""
|
|
|
|
def __init__(self, reolink_data: ReolinkData) -> None:
|
|
"""Initialize ReolinkHostCoordinatorEntity."""
|
|
super().__init__(reolink_data, reolink_data.device_coordinator)
|
|
|
|
|
|
class ReolinkChannelCoordinatorEntity(ReolinkHostCoordinatorEntity):
|
|
"""Parent class for Reolink hardware camera entities connected to a channel of the NVR."""
|
|
|
|
def __init__(
|
|
self,
|
|
reolink_data: ReolinkData,
|
|
channel: int,
|
|
) -> None:
|
|
"""Initialize ReolinkChannelCoordinatorEntity for a hardware camera connected to a channel of the NVR."""
|
|
super().__init__(reolink_data)
|
|
|
|
self._channel = channel
|
|
|
|
dev_ch = channel
|
|
if self._host.api.model in DUAL_LENS_MODELS:
|
|
dev_ch = 0
|
|
|
|
if self._host.api.is_nvr:
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, f"{self._host.unique_id}_ch{dev_ch}")},
|
|
via_device=(DOMAIN, self._host.unique_id),
|
|
name=self._host.api.camera_name(dev_ch),
|
|
model=self._host.api.camera_model(dev_ch),
|
|
manufacturer=self._host.api.manufacturer,
|
|
configuration_url=self._conf_url,
|
|
)
|