Use _get_reauth/reconfigure_entry in nam (#127303)

This commit is contained in:
epenet 2024-10-02 16:13:24 +02:00 committed by GitHub
parent e2eb986c7c
commit c6fa160c02
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5,7 +5,7 @@ from __future__ import annotations
from collections.abc import Mapping from collections.abc import Mapping
from dataclasses import dataclass from dataclasses import dataclass
import logging import logging
from typing import TYPE_CHECKING, Any from typing import Any
from aiohttp.client_exceptions import ClientConnectorError from aiohttp.client_exceptions import ClientConnectorError
from nettigo_air_monitor import ( from nettigo_air_monitor import (
@ -72,11 +72,9 @@ class NAMFlowHandler(ConfigFlow, domain=DOMAIN):
VERSION = 1 VERSION = 1
def __init__(self) -> None: _config: NamConfig
"""Initialize flow.""" entry: ConfigEntry
self.host: str host: str
self.entry: ConfigEntry
self._config: NamConfig
async def async_step_user( async def async_step_user(
self, user_input: dict[str, Any] | None = None self, user_input: dict[str, Any] | None = None
@ -189,8 +187,7 @@ class NAMFlowHandler(ConfigFlow, domain=DOMAIN):
self, entry_data: Mapping[str, Any] self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult: ) -> ConfigFlowResult:
"""Handle configuration by re-auth.""" """Handle configuration by re-auth."""
if entry := self.hass.config_entries.async_get_entry(self.context["entry_id"]): self.entry = self._get_reauth_entry()
self.entry = entry
self.host = entry_data[CONF_HOST] self.host = entry_data[CONF_HOST]
self.context["title_placeholders"] = {"host": self.host} self.context["title_placeholders"] = {"host": self.host}
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()
@ -229,13 +226,8 @@ class NAMFlowHandler(ConfigFlow, domain=DOMAIN):
self, entry_data: Mapping[str, Any] self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult: ) -> ConfigFlowResult:
"""Handle a reconfiguration flow initialized by the user.""" """Handle a reconfiguration flow initialized by the user."""
entry = self.hass.config_entries.async_get_entry(self.context["entry_id"]) self.entry = self._get_reconfigure_entry()
self.host = self.entry.data[CONF_HOST]
if TYPE_CHECKING:
assert entry is not None
self.host = entry.data[CONF_HOST]
self.entry = entry
return await self.async_step_reconfigure_confirm() return await self.async_step_reconfigure_confirm()