Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify yolink service actions

pull/146753/head
epenet 2025-06-13 14:53:21 +00:00
parent d880ce6bb4
commit 20971c03b3
1 changed files with 37 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
homeassistant/components/yolink/services.py
View File

@ -25,45 +25,46 @@ _SPEAKER_HUB_PLAY_CALL_OPTIONAL_ATTRS = (
)
async def _handle_speaker_hub_play_call(service_call: ServiceCall) -> None:
"""Handle Speaker Hub audio play call."""
service_data = service_call.data
device_registry = dr.async_get(service_call.hass)
device_entry = device_registry.async_get(service_data[ATTR_TARGET_DEVICE])
if device_entry is not None:
for entry_id in device_entry.config_entries:
if (
entry := service_call.hass.config_entries.async_get_entry(entry_id)
) is None:
continue
if entry.domain == DOMAIN:
break
if entry is None or entry.state != ConfigEntryState.LOADED:
raise ServiceValidationError(
translation_domain=DOMAIN,
translation_key="invalid_config_entry",
)
home_store = service_call.hass.data[DOMAIN][entry.entry_id]
for identifier in device_entry.identifiers:
if (
device_coordinator := home_store.device_coordinators.get(identifier[1])
) is not None:
params = {
ATTR_TEXT_MESSAGE: service_data[ATTR_TEXT_MESSAGE],
ATTR_REPEAT: service_data[ATTR_REPEAT],
}
for attr, transform in _SPEAKER_HUB_PLAY_CALL_OPTIONAL_ATTRS:
if attr in service_data:
params[attr] = transform(service_data[attr])
play_request = ClientRequest("playAudio", params)
await device_coordinator.device.call_device(play_request)
@callback
def async_setup_services(hass: HomeAssistant) -> None:
"""Register services for YoLink integration."""
async def handle_speaker_hub_play_call(service_call: ServiceCall) -> None:
"""Handle Speaker Hub audio play call."""
service_data = service_call.data
device_registry = dr.async_get(hass)
device_entry = device_registry.async_get(service_data[ATTR_TARGET_DEVICE])
if device_entry is not None:
for entry_id in device_entry.config_entries:
if (entry := hass.config_entries.async_get_entry(entry_id)) is None:
continue
if entry.domain == DOMAIN:
break
if entry is None or entry.state != ConfigEntryState.LOADED:
raise ServiceValidationError(
translation_domain=DOMAIN,
translation_key="invalid_config_entry",
)
home_store = hass.data[DOMAIN][entry.entry_id]
for identifier in device_entry.identifiers:
if (
device_coordinator := home_store.device_coordinators.get(
identifier[1]
)
) is not None:
params = {
ATTR_TEXT_MESSAGE: service_data[ATTR_TEXT_MESSAGE],
ATTR_REPEAT: service_data[ATTR_REPEAT],
}
for attr, transform in _SPEAKER_HUB_PLAY_CALL_OPTIONAL_ATTRS:
if attr in service_data:
params[attr] = transform(service_data[attr])
play_request = ClientRequest("playAudio", params)
await device_coordinator.device.call_device(play_request)
hass.services.async_register(
domain=DOMAIN,
service=SERVICE_PLAY_ON_SPEAKER_HUB,
@ -80,5 +81,5 @@ def async_setup_services(hass: HomeAssistant) -> None:
),
},
),
service_func=handle_speaker_hub_play_call,
service_func=_handle_speaker_hub_play_call,
)