Skip to content

Abort if a flow is removed during a step #142138

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Apr 9, 2025
Merged

Conversation

emontnemery
Copy link
Contributor

@emontnemery emontnemery commented Apr 3, 2025

Proposed change

Abort early if a flow is removed during a step, instead of going ahead creating an entry

To still give flow managers the opportunity to clean up, a new method FlowManager.async_flow_removed is added, which is meant to be implemented by classed derived from FlowManager.

In this PR, an async_flow_removed method is added to ConfigEntriesFlowManager, the method cleans up reauth issues which ensures reauth issues are always cleaned up when a reauth flow is aborted, not only if the reauth flow returns abort.

Type of change

  • Dependency upgrade
  • Bugfix (non-breaking change which fixes an issue)
  • New integration (thank you!)
  • New feature (which adds functionality to an existing integration)
  • Deprecation (breaking change to happen in the future)
  • Breaking change (fix/feature causing existing functionality to break)
  • Code quality improvements to existing code or addition of tests

Additional information

  • This PR fixes or closes issue: fixes #
  • This PR is related to issue:
  • Link to documentation pull request:
  • Link to developer documentation pull request:
  • Link to frontend pull request:

Checklist

  • The code change is tested and works locally.
  • Local tests pass. Your PR cannot be merged unless tests pass
  • There is no commented out code in this PR.
  • I have followed the development checklist
  • I have followed the perfect PR recommendations
  • The code has been formatted using Ruff (ruff format homeassistant tests)
  • Tests have been added to verify that the new code works.

If user exposed functionality or configuration variables are added/changed:

If the code communicates with devices, web services, or third-party tools:

  • The manifest file has all fields filled out correctly.
    Updated and included derived files by running: python3 -m script.hassfest.
  • New or updated dependencies have been added to requirements_all.txt.
    Updated by running python3 -m script.gen_requirements_all.
  • For the updated dependencies - a link to the changelog, or at minimum a diff between library versions is added to the PR description.

To help with the load of incoming pull requests:

@emontnemery emontnemery requested a review from a team as a code owner April 3, 2025 06:17
@emontnemery emontnemery marked this pull request as draft April 3, 2025 06:17
@emontnemery emontnemery force-pushed the data_entry_flow_end_on_abort branch from a3bd027 to f52d0fb Compare April 9, 2025 09:33
Comment on lines +1506 to +1511
@callback
def async_flow_removed(
self,
flow: data_entry_flow.FlowHandler[ConfigFlowContext, ConfigFlowResult],
result: ConfigFlowResult,
) -> ConfigFlowResult:
"""Finish a config flow and add an entry.

This method is called when a flow step returns FlowResultType.ABORT or
FlowResultType.CREATE_ENTRY.
"""
) -> None:
"""Handle a removed config flow."""
Copy link
Contributor Author

@emontnemery emontnemery Apr 9, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The point of this method is to unconditionally remove reauth issues when a reauth flow is removed. We could optionally also include updating of the discovery keys, but I don't think that's needed.

@bdraco
Copy link
Member

bdraco commented Apr 9, 2025

Latest design looks a lot cleaner. 👍

@emontnemery
Copy link
Contributor Author

emontnemery commented Apr 9, 2025

This needs tests of the new method and changed behavior before merge

Edit: Tests have been added, I think this can be merged now.

@emontnemery emontnemery marked this pull request as ready for review April 9, 2025 14:54
@bdraco
Copy link
Member

bdraco commented Apr 9, 2025

Thanks @emontnemery

@bdraco bdraco merged commit f344314 into dev Apr 9, 2025
48 checks passed
@bdraco bdraco deleted the data_entry_flow_end_on_abort branch April 9, 2025 17:04
@github-actions github-actions bot locked and limited conversation to collaborators Apr 10, 2025
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants