diff --git a/homeassistant/data_entry_flow.py b/homeassistant/data_entry_flow.py index b02fcbfcd1f..b7b1a68e792 100644 --- a/homeassistant/data_entry_flow.py +++ b/homeassistant/data_entry_flow.py @@ -46,6 +46,15 @@ RESULT_TYPE_MENU = "menu" # Event that is fired when a flow is progressed via external or progress source. EVENT_DATA_ENTRY_FLOW_PROGRESSED = "data_entry_flow_progressed" +FLOW_NOT_COMPLETE_STEPS = { + FlowResultType.FORM, + FlowResultType.EXTERNAL_STEP, + FlowResultType.EXTERNAL_STEP_DONE, + FlowResultType.SHOW_PROGRESS, + FlowResultType.SHOW_PROGRESS_DONE, + FlowResultType.MENU, +} + @dataclass(slots=True) class BaseServiceInfo: @@ -407,14 +416,7 @@ class FlowManager(abc.ABC): error_if_core=False, ) - if result["type"] in ( - FlowResultType.FORM, - FlowResultType.EXTERNAL_STEP, - FlowResultType.EXTERNAL_STEP_DONE, - FlowResultType.SHOW_PROGRESS, - FlowResultType.SHOW_PROGRESS_DONE, - FlowResultType.MENU, - ): + if result["type"] in FLOW_NOT_COMPLETE_STEPS: self._raise_if_step_does_not_exist(flow, result["step_id"]) flow.cur_step = result return result