Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Streamline discovery flow callback (#76666)

pull/76710/head
Paulus Schoutsen 2022-08-12 19:42:41 -04:00 committed by GitHub
parent 2e40fc7288
commit 2f29f38ec6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/helpers/discovery_flow.py
View File

@ -60,17 +60,14 @@ class FlowDispatcher:
@callback
def async_setup(self) -> None:
"""Set up the flow disptcher."""
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STARTED, self.async_start)
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STARTED, self._async_start)
@callback
def async_start(self, event: Event) -> None:
async def _async_start(self, event: Event) -> None:
"""Start processing pending flows."""
self.hass.data.pop(DISCOVERY_FLOW_DISPATCHER)
self.hass.async_create_task(self._async_process_pending_flows())
async def _async_process_pending_flows(self) -> None:
"""Process any pending discovery flows."""
init_coros = [_async_init_flow(self.hass, *flow) for flow in self.pending_flows]
await gather_with_concurrency(
FLOW_INIT_LIMIT,
*[init_coro for init_coro in init_coros if init_coro is not None],