Use new reauth helpers in volvooncall (#128782)

This commit is contained in:
epenet 2024-10-19 19:22:20 +02:00 committed by GitHub
parent 990987ac92
commit 6af6b73c89
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -9,7 +9,7 @@ from typing import Any
import voluptuous as vol
from volvooncall import Connection
from homeassistant.config_entries import ConfigEntry, ConfigFlow, ConfigFlowResult
from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlow, ConfigFlowResult
from homeassistant.const import (
CONF_PASSWORD,
CONF_REGION,
@ -35,7 +35,6 @@ class VolvoOnCallConfigFlow(ConfigFlow, domain=DOMAIN):
"""VolvoOnCall config flow."""
VERSION = 1
_reauth_entry: ConfigEntry | None = None
async def async_step_user(
self, user_input: dict[str, Any] | None = None
@ -53,7 +52,7 @@ class VolvoOnCallConfigFlow(ConfigFlow, domain=DOMAIN):
if user_input is not None:
await self.async_set_unique_id(user_input[CONF_USERNAME])
if not self._reauth_entry:
if self.source != SOURCE_REAUTH:
self._abort_if_unique_id_configured()
try:
@ -64,21 +63,18 @@ class VolvoOnCallConfigFlow(ConfigFlow, domain=DOMAIN):
_LOGGER.exception("Unhandled exception in user step")
errors["base"] = "unknown"
if not errors:
if self._reauth_entry:
self.hass.config_entries.async_update_entry(
self._reauth_entry, data=self._reauth_entry.data | user_input
if self.source == SOURCE_REAUTH:
return self.async_update_reload_and_abort(
self._get_reauth_entry(), data_updates=user_input
)
await self.hass.config_entries.async_reload(
self._reauth_entry.entry_id
)
return self.async_abort(reason="reauth_successful")
return self.async_create_entry(
title=user_input[CONF_USERNAME], data=user_input
)
elif self._reauth_entry:
elif self.source == SOURCE_REAUTH:
reauth_entry = self._get_reauth_entry()
for key in defaults:
defaults[key] = self._reauth_entry.data.get(key)
defaults[key] = reauth_entry.data.get(key)
user_schema = vol.Schema(
{
@ -110,9 +106,6 @@ class VolvoOnCallConfigFlow(ConfigFlow, domain=DOMAIN):
self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult:
"""Perform reauth upon an API authentication error."""
self._reauth_entry = self.hass.config_entries.async_get_entry(
self.context["entry_id"]
)
return await self.async_step_user()
async def is_valid(self, user_input):