jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 23:27:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small speed up to creating flows (#117896)

Browse Source 
Use a defaultdict instead of setdefault
This commit is contained in:
J. Nick Koston 2024-05-21 20:16:08 -10:00 committed by GitHub
parent 5abf77662a
commit 1985a2ad8b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
homeassistant/data_entry_flow.py
View File

@ -4,6 +4,7 @@ from __future__ import annotations
import abc import abc
import asyncio import asyncio
from collections import defaultdict
from collections.abc import Callable, Container, Iterable, Mapping from collections.abc import Callable, Container, Iterable, Mapping
from contextlib import suppress from contextlib import suppress
import copy import copy
@ -203,12 +204,12 @@ class FlowManager(abc.ABC, Generic[_FlowResultT, _HandlerT]):
self.hass = hass self.hass = hass
self._preview: set[_HandlerT] = set() self._preview: set[_HandlerT] = set()
self._progress: dict[str, FlowHandler[_FlowResultT, _HandlerT]] = {} self._progress: dict[str, FlowHandler[_FlowResultT, _HandlerT]] = {}
self._handler_progress_index: dict[ self._handler_progress_index: defaultdict[
_HandlerT, set[FlowHandler[_FlowResultT, _HandlerT]] _HandlerT, set[FlowHandler[_FlowResultT, _HandlerT]]
] = {} ] = defaultdict(set)
self._init_data_process_index: dict[ self._init_data_process_index: defaultdict[
type, set[FlowHandler[_FlowResultT, _HandlerT]] type, set[FlowHandler[_FlowResultT, _HandlerT]]
] = {} ] = defaultdict(set)
@abc.abstractmethod @abc.abstractmethod
async def async_create_flow( async def async_create_flow(
@ -295,7 +296,7 @@ class FlowManager(abc.ABC, Generic[_FlowResultT, _HandlerT]):
return self._async_flow_handler_to_flow_result( return self._async_flow_handler_to_flow_result(
( (
progress progress
for progress in self._init_data_process_index.get(init_data_type, set()) for progress in self._init_data_process_index.get(init_data_type, ())
if matcher(progress.init_data) if matcher(progress.init_data)
), ),
include_uninitialized, include_uninitialized,
@ -471,10 +472,9 @@ class FlowManager(abc.ABC, Generic[_FlowResultT, _HandlerT]):
) -> None: ) -> None:
"""Add a flow to in progress.""" """Add a flow to in progress."""
if flow.init_data is not None: if flow.init_data is not None:
init_data_type = type(flow.init_data) self._init_data_process_index[type(flow.init_data)].add(flow)
self._init_data_process_index.setdefault(init_data_type, set()).add(flow)
self._progress[flow.flow_id] = flow self._progress[flow.flow_id] = flow
self._handler_progress_index.setdefault(flow.handler, set()).add(flow) self._handler_progress_index[flow.handler].add(flow)
@callback @callback
def _async_remove_flow_from_index( def _async_remove_flow_from_index(