From a12f5da7e6ed1467c3b97c7080dacb86c0eca531 Mon Sep 17 00:00:00 2001 From: Erik Date: Wed, 9 Apr 2025 15:36:37 +0200 Subject: [PATCH] Use helper setup.async_wait_component --- homeassistant/components/websocket_api/commands.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/websocket_api/commands.py b/homeassistant/components/websocket_api/commands.py index aad053e033c..ddcdd4f1cf8 100644 --- a/homeassistant/components/websocket_api/commands.py +++ b/homeassistant/components/websocket_api/commands.py @@ -60,9 +60,9 @@ from homeassistant.loader import ( async_get_integrations, ) from homeassistant.setup import ( - DATA_SETUP_DONE, async_get_loaded_integrations, async_get_setup_timings, + async_wait_component, ) from homeassistant.util.json import format_unserializable_data @@ -943,12 +943,6 @@ async def handle_integration_wait( """Handle wait for integration command.""" domain: str = msg["domain"] - setup_done = hass.data.get(DATA_SETUP_DONE, {}) - - # Wait for the integration to be set up - if setup_future := setup_done.get(domain): - await setup_future - connection.send_result( - msg["id"], {"integration_loaded": domain in hass.config.components} + msg["id"], {"integration_loaded": await async_wait_component(hass, domain)} )