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

Rewrite flow handler to flow result conversion as a list comp (#118269)

Browse Source
This commit is contained in:
J. Nick Koston 2024-05-27 15:39:59 -10:00 committed by GitHub
parent 6f248acfd5
commit a5644c8ddb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
homeassistant/data_entry_flow.py
View File

@ -608,19 +608,22 @@ class FlowManager(abc.ABC, Generic[_FlowResultT, _HandlerT]):
include_uninitialized: bool,
) -> list[_FlowResultT]:
"""Convert a list of FlowHandler to a partial FlowResult that can be serialized."""
results = []
for flow in flows:
if not include_uninitialized and flow.cur_step is None:
continue
result = self._flow_result(
return [
self._flow_result(
flow_id=flow.flow_id,
handler=flow.handler,
context=flow.context,
step_id=flow.cur_step["step_id"],
)
if flow.cur_step
else self._flow_result(
flow_id=flow.flow_id,
handler=flow.handler,
context=flow.context,
)
if flow.cur_step:
result["step_id"] = flow.cur_step["step_id"]
results.append(result)
return results
for flow in flows
if include_uninitialized or flow.cur_step is not None
]
class FlowHandler(Generic[_FlowResultT, _HandlerT]):