evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ensure platform setup is awaited in zwave_me (#86581)

Browse source 
* Ensure platform setup is awaited in zwave_me

There was a race during setup since platform setup
was not being awaited and was being done in a task

Also migrates to using async_forward_entry_setups
instead of manually writing out async_forward_entry_setup

* Ensure platform setup is awaited in zwave_me

There was a race during setup since platform setup
was not being awaited and was being done in a task

Also migrates to using async_forward_entry_setups
instead of manually writing out async_forward_entry_setup
This commit is contained in:
J. Nick Koston 2023-01-24 11:49:02 -10:00 • committed by GitHub
parent 4b427ec02c
commit a851b20c97
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/zwave_me/__init__.py
View file

@ -1,5 +1,4 @@
"""The Z-Wave-Me WS integration."""
import asyncio
import logging
from zwave_me_ws import ZWaveMe, ZWaveMeData
@ -24,7 +23,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
hass.data.setdefault(DOMAIN, {})
controller = hass.data[DOMAIN][entry.entry_id] = ZWaveMeController(hass, entry)
if await controller.async_establish_connection():
hass.async_create_task(async_setup_platforms(hass, entry, controller))
await async_setup_platforms(hass, entry, controller)
registry = device_registry.async_get(hass)
controller.remove_stale_devices(registry)
return True
@ -98,12 +97,7 @@ async def async_setup_platforms(
hass: HomeAssistant, entry: ConfigEntry, controller: ZWaveMeController
) -> None:
"""Set up platforms."""
await asyncio.gather(
*[
hass.config_entries.async_forward_entry_setup(entry, platform)
for platform in PLATFORMS
]
)
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
controller.platforms_inited = True
await hass.async_add_executor_job(controller.zwave_api.get_devices)