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

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

View File

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