Make ConfigFlow.async_set_unique_id smarter

This commit is contained in:
Erik 2024-08-20 15:26:44 +02:00
parent a1e3e7f24f
commit 95b532df41

View File

@ -2385,12 +2385,14 @@ class ConfigFlow(ConfigEntryBaseFlow):
raise data_entry_flow.AbortFlow(error) raise data_entry_flow.AbortFlow(error)
async def async_set_unique_id( async def async_set_unique_id(
self, unique_id: str | None = None, *, raise_on_progress: bool = True self, unique_id: str | None = None, *, raise_on_progress: bool | None = None
) -> ConfigEntry | None: ) -> ConfigEntry | None:
"""Set a unique ID for the config flow. """Set a unique ID for the config flow.
Returns optionally existing config entry with same ID. Returns optionally existing config entry with same ID.
""" """
if raise_on_progress is None:
raise_on_progress = self.source != SOURCE_USER
if unique_id is None: if unique_id is None:
self.context["unique_id"] = None self.context["unique_id"] = None
return None return None