diff --git a/homeassistant/components/otbr/websocket_api.py b/homeassistant/components/otbr/websocket_api.py index 7c69a8d0a2d..497fc285d5f 100644 --- a/homeassistant/components/otbr/websocket_api.py +++ b/homeassistant/components/otbr/websocket_api.py @@ -3,12 +3,7 @@ from typing import TYPE_CHECKING import python_otbr_api -from homeassistant.components.websocket_api import ( - ActiveConnection, - async_register_command, - async_response, - websocket_command, -) +from homeassistant.components import websocket_api from homeassistant.core import HomeAssistant, callback from homeassistant.exceptions import HomeAssistantError @@ -21,18 +16,19 @@ if TYPE_CHECKING: @callback def async_setup(hass: HomeAssistant) -> None: """Set up the OTBR Websocket API.""" - async_register_command(hass, websocket_info) - async_register_command(hass, websocket_create_network) + websocket_api.async_register_command(hass, websocket_info) + websocket_api.async_register_command(hass, websocket_create_network) -@websocket_command( +@websocket_api.websocket_command( { "type": "otbr/info", } ) -@async_response +@websocket_api.require_admin +@websocket_api.async_response async def websocket_info( - hass: HomeAssistant, connection: ActiveConnection, msg: dict + hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict ) -> None: """Get OTBR info.""" if DOMAIN not in hass.data: @@ -56,14 +52,15 @@ async def websocket_info( ) -@websocket_command( +@websocket_api.websocket_command( { "type": "otbr/create_network", } ) -@async_response +@websocket_api.require_admin +@websocket_api.async_response async def websocket_create_network( - hass: HomeAssistant, connection: ActiveConnection, msg: dict + hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict ) -> None: """Create a new Thread network.""" if DOMAIN not in hass.data: