64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
"""BaseEntity to support multiple LinkPlay platforms."""
|
|
|
|
from collections.abc import Callable, Coroutine
|
|
from typing import Any, Concatenate
|
|
|
|
from linkplay.bridge import LinkPlayBridge
|
|
from linkplay.manufacturers import MANUFACTURER_GENERIC, get_info_from_project
|
|
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from . import DOMAIN, LinkPlayRequestException
|
|
|
|
|
|
def exception_wrap[_LinkPlayEntityT: LinkPlayBaseEntity, **_P, _R](
|
|
func: Callable[Concatenate[_LinkPlayEntityT, _P], Coroutine[Any, Any, _R]],
|
|
) -> Callable[Concatenate[_LinkPlayEntityT, _P], Coroutine[Any, Any, _R]]:
|
|
"""Define a wrapper to catch exceptions and raise HomeAssistant errors."""
|
|
|
|
async def _wrap(self: _LinkPlayEntityT, *args: _P.args, **kwargs: _P.kwargs) -> _R:
|
|
try:
|
|
return await func(self, *args, **kwargs)
|
|
except LinkPlayRequestException as err:
|
|
raise HomeAssistantError(
|
|
f"Exception occurred when communicating with API {func}: {err}"
|
|
) from err
|
|
|
|
return _wrap
|
|
|
|
|
|
class LinkPlayBaseEntity(Entity):
|
|
"""Representation of a LinkPlay base entity."""
|
|
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(self, bridge: LinkPlayBridge) -> None:
|
|
"""Initialize the LinkPlay media player."""
|
|
|
|
self._bridge = bridge
|
|
|
|
manufacturer, model = get_info_from_project(bridge.device.properties["project"])
|
|
model_id = None
|
|
if model != MANUFACTURER_GENERIC:
|
|
model_id = bridge.device.properties["project"]
|
|
|
|
connections: set[tuple[str, str]] = set()
|
|
if "MAC" in bridge.device.properties:
|
|
connections.add(
|
|
(dr.CONNECTION_NETWORK_MAC, bridge.device.properties["MAC"])
|
|
)
|
|
|
|
self._attr_device_info = dr.DeviceInfo(
|
|
configuration_url=bridge.endpoint,
|
|
connections=connections,
|
|
hw_version=bridge.device.properties["hardware"],
|
|
identifiers={(DOMAIN, bridge.device.uuid)},
|
|
manufacturer=manufacturer,
|
|
model=model,
|
|
model_id=model_id,
|
|
name=bridge.device.name,
|
|
sw_version=bridge.device.properties["firmware"],
|
|
)
|