61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
"""Base class for Squeezebox Sensor entities."""
|
|
|
|
from homeassistant.helpers.device_registry import (
|
|
CONNECTION_NETWORK_MAC,
|
|
DeviceInfo,
|
|
format_mac,
|
|
)
|
|
from homeassistant.helpers.entity import EntityDescription
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import DOMAIN, STATUS_QUERY_UUID
|
|
from .coordinator import (
|
|
LMSStatusDataUpdateCoordinator,
|
|
SqueezeBoxPlayerUpdateCoordinator,
|
|
)
|
|
|
|
|
|
class SqueezeboxEntity(CoordinatorEntity[SqueezeBoxPlayerUpdateCoordinator]):
|
|
"""Base entity class for Squeezebox entities."""
|
|
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(self, coordinator: SqueezeBoxPlayerUpdateCoordinator) -> None:
|
|
"""Initialize the SqueezeBox entity."""
|
|
super().__init__(coordinator)
|
|
self._player = coordinator.player
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, format_mac(self._player.player_id))},
|
|
connections={(CONNECTION_NETWORK_MAC, format_mac(self._player.player_id))},
|
|
)
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return True if entity is available."""
|
|
# super().available refers to CoordinatorEntity.available (self.coordinator.last_update_success)
|
|
# self.coordinator.available is the custom availability flag from SqueezeBoxPlayerUpdateCoordinator
|
|
return self.coordinator.available and super().available
|
|
|
|
|
|
class LMSStatusEntity(CoordinatorEntity[LMSStatusDataUpdateCoordinator]):
|
|
"""Defines a base status sensor entity."""
|
|
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: LMSStatusDataUpdateCoordinator,
|
|
description: EntityDescription,
|
|
) -> None:
|
|
"""Initialize status sensor entity."""
|
|
super().__init__(coordinator)
|
|
self.entity_description = description
|
|
self._attr_translation_key = description.key.replace(" ", "_")
|
|
self._attr_unique_id = (
|
|
f"{coordinator.data[STATUS_QUERY_UUID]}_{description.key}"
|
|
)
|
|
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, coordinator.data[STATUS_QUERY_UUID])},
|
|
)
|