core/homeassistant/components/syncthru/entity.py

37 lines
1.4 KiB
Python

"""Base class for Syncthru entities."""
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 DOMAIN
from .coordinator import SyncthruCoordinator
class SyncthruEntity(CoordinatorEntity[SyncthruCoordinator]):
"""Base class for Syncthru entities."""
_attr_has_entity_name = True
def __init__(
self, coordinator: SyncthruCoordinator, entity_description: EntityDescription
) -> None:
"""Initialize the Syncthru entity."""
super().__init__(coordinator)
self.entity_description = entity_description
serial_number = coordinator.syncthru.serial_number()
assert serial_number is not None
self._attr_unique_id = f"{serial_number}_{entity_description.key}"
connections = set()
if mac := coordinator.syncthru.raw().get("identity", {}).get("mac_addr"):
connections.add((dr.CONNECTION_NETWORK_MAC, mac))
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, serial_number)},
connections=connections,
configuration_url=coordinator.syncthru.url,
manufacturer="Samsung",
model=coordinator.syncthru.model(),
name=coordinator.syncthru.hostname(),
)