mirror of https://github.com/home-assistant/core
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
"""Reolink parent entity class."""
|
|
from __future__ import annotations
|
|
|
|
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
|
|
|
|
|
|
class ReolinkBaseCoordinatorEntity(CoordinatorEntity):
|
|
"""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.
|
|
"""
|
|
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(
|
|
self,
|
|
reolink_data: ReolinkData,
|
|
coordinator: DataUpdateCoordinator | None = None,
|
|
) -> None:
|
|
"""Initialize ReolinkBaseCoordinatorEntity for a NVR entity without a channel."""
|
|
if coordinator is None:
|
|
coordinator = reolink_data.device_coordinator
|
|
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 ReolinkCoordinatorEntity(ReolinkBaseCoordinatorEntity):
|
|
"""Parent class for Reolink hardware camera entities connected to a channel of the NVR."""
|
|
|
|
def __init__(
|
|
self,
|
|
reolink_data: ReolinkData,
|
|
channel: int,
|
|
coordinator: DataUpdateCoordinator | None = None,
|
|
) -> None:
|
|
"""Initialize ReolinkCoordinatorEntity for a hardware camera connected to a channel of the NVR."""
|
|
super().__init__(reolink_data, coordinator)
|
|
|
|
self._channel = channel
|
|
|
|
if self._host.api.is_nvr:
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, f"{self._host.unique_id}_ch{self._channel}")},
|
|
via_device=(DOMAIN, self._host.unique_id),
|
|
name=self._host.api.camera_name(self._channel),
|
|
model=self._host.api.camera_model(self._channel),
|
|
manufacturer=self._host.api.manufacturer,
|
|
configuration_url=self._conf_url,
|
|
)
|