Fix flux_led discovery running at shutdown (#88817)

This commit is contained in:
J. Nick Koston 2023-02-26 20:14:54 -06:00 committed by Paulus Schoutsen
parent 2fff836bd4
commit ab6bd75b70

View File

@ -87,14 +87,23 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
hass, STARTUP_SCAN_TIMEOUT hass, STARTUP_SCAN_TIMEOUT
) )
@callback
def _async_start_background_discovery(*_: Any) -> None:
"""Run discovery in the background."""
hass.async_create_background_task(_async_discovery(), "flux_led-discovery")
async def _async_discovery(*_: Any) -> None: async def _async_discovery(*_: Any) -> None:
async_trigger_discovery( async_trigger_discovery(
hass, await async_discover_devices(hass, DISCOVER_SCAN_TIMEOUT) hass, await async_discover_devices(hass, DISCOVER_SCAN_TIMEOUT)
) )
async_trigger_discovery(hass, domain_data[FLUX_LED_DISCOVERY]) async_trigger_discovery(hass, domain_data[FLUX_LED_DISCOVERY])
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STARTED, _async_discovery) hass.bus.async_listen_once(
async_track_time_interval(hass, _async_discovery, DISCOVERY_INTERVAL) EVENT_HOMEASSISTANT_STARTED, _async_start_background_discovery
)
async_track_time_interval(
hass, _async_start_background_discovery, DISCOVERY_INTERVAL
)
return True return True