jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 22:27:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for System Bridge service responses (#100055)

Browse Source 
* Add support for System Bridge service responses

* Update conversion to use dataclass

* Update debug statements

* Update debug message
This commit is contained in:
Aidan Timson 2024-03-05 13:30:53 +00:00 committed by GitHub
parent bf596562bf
commit b070bb25a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 41 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
homeassistant/components/system_bridge/__init__.py
View File

@ -2,6 +2,7 @@
from __future__ import annotations from __future__ import annotations
import asyncio import asyncio
from dataclasses import asdict
import logging import logging
from systembridgeconnector.exceptions import ( from systembridgeconnector.exceptions import (
@ -29,7 +30,12 @@ from homeassistant.const import (
CONF_URL, CONF_URL,
Platform, Platform,
) )
from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.core import (
HomeAssistant,
ServiceCall,
ServiceResponse,
SupportsResponse,
)
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
@ -194,52 +200,59 @@ async def async_setup_entry(
raise vol.Invalid(f"Could not find device {device}") from exception raise vol.Invalid(f"Could not find device {device}") from exception
raise vol.Invalid(f"Device {device} does not exist") raise vol.Invalid(f"Device {device} does not exist")
async def handle_open_path(call: ServiceCall) -> None: async def handle_open_path(service_call: ServiceCall) -> ServiceResponse:
"""Handle the open path service call.""" """Handle the open path service call."""
_LOGGER.debug("Open: %s", call.data) _LOGGER.debug("Open path: %s", service_call.data)
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][ coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][
call.data[CONF_BRIDGE] service_call.data[CONF_BRIDGE]
] ]
await coordinator.websocket_client.open_path( response = await coordinator.websocket_client.open_path(
OpenPath(path=call.data[CONF_PATH]) OpenPath(path=service_call.data[CONF_PATH])
) )
return asdict(response)
async def handle_power_command(call: ServiceCall) -> None: async def handle_power_command(service_call: ServiceCall) -> ServiceResponse:
"""Handle the power command service call.""" """Handle the power command service call."""
_LOGGER.debug("Power command: %s", call.data) _LOGGER.debug("Power command: %s", service_call.data)
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][ coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][
call.data[CONF_BRIDGE] service_call.data[CONF_BRIDGE]
] ]
await getattr( response = await getattr(
coordinator.websocket_client, coordinator.websocket_client,
POWER_COMMAND_MAP[call.data[CONF_COMMAND]], POWER_COMMAND_MAP[service_call.data[CONF_COMMAND]],
)() )()
return asdict(response)
async def handle_open_url(call: ServiceCall) -> None: async def handle_open_url(service_call: ServiceCall) -> ServiceResponse:
"""Handle the open url service call.""" """Handle the open url service call."""
_LOGGER.debug("Open: %s", call.data) _LOGGER.debug("Open URL: %s", service_call.data)
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][ coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][
call.data[CONF_BRIDGE] service_call.data[CONF_BRIDGE]
] ]
await coordinator.websocket_client.open_url(OpenUrl(url=call.data[CONF_URL])) response = await coordinator.websocket_client.open_url(
OpenUrl(url=service_call.data[CONF_URL])
)
return asdict(response)
async def handle_send_keypress(call: ServiceCall) -> None: async def handle_send_keypress(service_call: ServiceCall) -> ServiceResponse:
"""Handle the send_keypress service call.""" """Handle the send_keypress service call."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][ coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][
call.data[CONF_BRIDGE] service_call.data[CONF_BRIDGE]
] ]
await coordinator.websocket_client.keyboard_keypress( response = await coordinator.websocket_client.keyboard_keypress(
KeyboardKey(key=call.data[CONF_KEY]) KeyboardKey(key=service_call.data[CONF_KEY])
) )
return asdict(response)
async def handle_send_text(call: ServiceCall) -> None: async def handle_send_text(service_call: ServiceCall) -> ServiceResponse:
"""Handle the send_keypress service call.""" """Handle the send_keypress service call."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][ coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][
call.data[CONF_BRIDGE] service_call.data[CONF_BRIDGE]
] ]
await coordinator.websocket_client.keyboard_text( response = await coordinator.websocket_client.keyboard_text(
KeyboardText(text=call.data[CONF_TEXT]) KeyboardText(text=service_call.data[CONF_TEXT])
) )
return asdict(response)
hass.services.async_register( hass.services.async_register(
DOMAIN, DOMAIN,
@ -251,6 +264,7 @@ async def async_setup_entry(
vol.Required(CONF_PATH): cv.string, vol.Required(CONF_PATH): cv.string,
}, },
), ),
supports_response=SupportsResponse.ONLY,
) )
hass.services.async_register( hass.services.async_register(
@ -263,6 +277,7 @@ async def async_setup_entry(
vol.Required(CONF_COMMAND): vol.In(POWER_COMMAND_MAP), vol.Required(CONF_COMMAND): vol.In(POWER_COMMAND_MAP),
}, },
), ),
supports_response=SupportsResponse.ONLY,
) )
hass.services.async_register( hass.services.async_register(
@ -275,6 +290,7 @@ async def async_setup_entry(
vol.Required(CONF_URL): cv.string, vol.Required(CONF_URL): cv.string,
}, },
), ),
supports_response=SupportsResponse.ONLY,
) )
hass.services.async_register( hass.services.async_register(
@ -287,6 +303,7 @@ async def async_setup_entry(
vol.Required(CONF_KEY): cv.string, vol.Required(CONF_KEY): cv.string,
}, },
), ),
supports_response=SupportsResponse.ONLY,
) )
hass.services.async_register( hass.services.async_register(
@ -299,6 +316,7 @@ async def async_setup_entry(
vol.Required(CONF_TEXT): cv.string, vol.Required(CONF_TEXT): cv.string,
}, },
), ),
supports_response=SupportsResponse.ONLY,
) )
# Reload entry when its updated. # Reload entry when its updated.