Use Mapping for async_step_reauth (p-s) (#72766)

This commit is contained in:
epenet 2022-06-02 14:28:14 +02:00 committed by GitHub
parent 52561ce076
commit fc8727454a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
12 changed files with 29 additions and 20 deletions

View File

@ -1,6 +1,7 @@
"""Config flow for Tesla Powerwall integration.""" """Config flow for Tesla Powerwall integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
import logging import logging
from typing import Any from typing import Any
@ -205,7 +206,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
errors=errors, errors=errors,
) )
async def async_step_reauth(self, data: dict[str, str]) -> FlowResult: async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
"""Handle configuration by re-auth.""" """Handle configuration by re-auth."""
self.reauth_entry = self.hass.config_entries.async_get_entry( self.reauth_entry = self.hass.config_entries.async_get_entry(
self.context["entry_id"] self.context["entry_id"]

View File

@ -1,6 +1,7 @@
"""Config flow to configure the PVOutput integration.""" """Config flow to configure the PVOutput integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from typing import Any from typing import Any
from pvo import PVOutput, PVOutputAuthenticationError, PVOutputError from pvo import PVOutput, PVOutputAuthenticationError, PVOutputError
@ -83,7 +84,7 @@ class PVOutputFlowHandler(ConfigFlow, domain=DOMAIN):
errors=errors, errors=errors,
) )
async def async_step_reauth(self, data: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
"""Handle initiation of re-authentication with PVOutput.""" """Handle initiation of re-authentication with PVOutput."""
self.reauth_entry = self.hass.config_entries.async_get_entry( self.reauth_entry = self.hass.config_entries.async_get_entry(
self.context["entry_id"] self.context["entry_id"]

View File

@ -1,6 +1,7 @@
"""Config flow to configure Renault component.""" """Config flow to configure Renault component."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from typing import TYPE_CHECKING, Any from typing import TYPE_CHECKING, Any
from renault_api.const import AVAILABLE_LOCALES from renault_api.const import AVAILABLE_LOCALES
@ -21,7 +22,7 @@ class RenaultFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
def __init__(self) -> None: def __init__(self) -> None:
"""Initialize the Renault config flow.""" """Initialize the Renault config flow."""
self._original_data: dict[str, Any] | None = None self._original_data: Mapping[str, Any] | None = None
self.renault_config: dict[str, Any] = {} self.renault_config: dict[str, Any] = {}
self.renault_hub: RenaultHub | None = None self.renault_hub: RenaultHub | None = None
@ -92,9 +93,9 @@ class RenaultFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
), ),
) )
async def async_step_reauth(self, user_input: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
"""Perform reauth upon an API authentication error.""" """Perform reauth upon an API authentication error."""
self._original_data = user_input.copy() self._original_data = data
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()
async def async_step_reauth_confirm( async def async_step_reauth_confirm(

View File

@ -1,6 +1,7 @@
"""Config flow for Ridwell integration.""" """Config flow for Ridwell integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from typing import TYPE_CHECKING, Any from typing import TYPE_CHECKING, Any
from aioridwell import async_get_client from aioridwell import async_get_client
@ -80,7 +81,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
data={CONF_USERNAME: self._username, CONF_PASSWORD: self._password}, data={CONF_USERNAME: self._username, CONF_PASSWORD: self._password},
) )
async def async_step_reauth(self, config: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, config: Mapping[str, Any]) -> FlowResult:
"""Handle configuration by re-auth.""" """Handle configuration by re-auth."""
self._username = config[CONF_USERNAME] self._username = config[CONF_USERNAME]
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()

View File

@ -1,9 +1,9 @@
"""Config flow for Samsung TV.""" """Config flow for Samsung TV."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from functools import partial from functools import partial
import socket import socket
from types import MappingProxyType
from typing import Any from typing import Any
from urllib.parse import urlparse from urllib.parse import urlparse
@ -525,7 +525,7 @@ class SamsungTVConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
) )
async def async_step_reauth( async def async_step_reauth(
self, data: MappingProxyType[str, Any] self, data: Mapping[str, Any]
) -> data_entry_flow.FlowResult: ) -> data_entry_flow.FlowResult:
"""Handle configuration by re-auth.""" """Handle configuration by re-auth."""
self._reauth_entry = self.hass.config_entries.async_get_entry( self._reauth_entry = self.hass.config_entries.async_get_entry(

View File

@ -1,6 +1,7 @@
"""Adds config flow for Sensibo integration.""" """Adds config flow for Sensibo integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from typing import Any from typing import Any
from pysensibo.exceptions import AuthenticationError from pysensibo.exceptions import AuthenticationError
@ -28,9 +29,7 @@ class SensiboConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
entry: config_entries.ConfigEntry | None entry: config_entries.ConfigEntry | None
async def async_step_reauth( async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
self, user_input: dict[str, Any] | None = None
) -> FlowResult:
"""Handle re-authentication with Sensibo.""" """Handle re-authentication with Sensibo."""
self.entry = self.hass.config_entries.async_get_entry(self.context["entry_id"]) self.entry = self.hass.config_entries.async_get_entry(self.context["entry_id"])

View File

@ -2,6 +2,7 @@
from __future__ import annotations from __future__ import annotations
import asyncio import asyncio
from collections.abc import Mapping
from typing import Any from typing import Any
import async_timeout import async_timeout
@ -97,7 +98,7 @@ class SimpliSafeFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
"""Define the config flow to handle options.""" """Define the config flow to handle options."""
return SimpliSafeOptionsFlowHandler(config_entry) return SimpliSafeOptionsFlowHandler(config_entry)
async def async_step_reauth(self, config: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, config: Mapping[str, Any]) -> FlowResult:
"""Handle configuration by re-auth.""" """Handle configuration by re-auth."""
self._reauth = True self._reauth = True

View File

@ -1,6 +1,7 @@
"""Config flow to configure SleepIQ component.""" """Config flow to configure SleepIQ component."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
import logging import logging
from typing import Any from typing import Any
@ -79,12 +80,12 @@ class SleepIQFlowHandler(ConfigFlow, domain=DOMAIN):
last_step=True, last_step=True,
) )
async def async_step_reauth(self, user_input: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
"""Perform reauth upon an API authentication error.""" """Perform reauth upon an API authentication error."""
self._reauth_entry = self.hass.config_entries.async_get_entry( self._reauth_entry = self.hass.config_entries.async_get_entry(
self.context["entry_id"] self.context["entry_id"]
) )
return await self.async_step_reauth_confirm(user_input) return await self.async_step_reauth_confirm(dict(data))
async def async_step_reauth_confirm( async def async_step_reauth_confirm(
self, user_input: dict[str, Any] | None = None self, user_input: dict[str, Any] | None = None

View File

@ -1,6 +1,7 @@
"""Config flow for Steam integration.""" """Config flow for Steam integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from typing import Any from typing import Any
import steam import steam
@ -125,7 +126,7 @@ class SteamFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
import_config[CONF_ACCOUNT] = import_config[CONF_ACCOUNTS][0] import_config[CONF_ACCOUNT] = import_config[CONF_ACCOUNTS][0]
return await self.async_step_user(import_config) return await self.async_step_user(import_config)
async def async_step_reauth(self, user_input: dict[str, str]) -> FlowResult: async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
"""Handle a reauthorization flow request.""" """Handle a reauthorization flow request."""
self.entry = self.hass.config_entries.async_get_entry(self.context["entry_id"]) self.entry = self.hass.config_entries.async_get_entry(self.context["entry_id"])

View File

@ -1,6 +1,7 @@
"""Config flow for Sure Petcare integration.""" """Config flow for Sure Petcare integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
import logging import logging
from typing import Any from typing import Any
@ -86,7 +87,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
step_id="user", data_schema=USER_DATA_SCHEMA, errors=errors step_id="user", data_schema=USER_DATA_SCHEMA, errors=errors
) )
async def async_step_reauth(self, config: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, config: Mapping[str, Any]) -> FlowResult:
"""Handle configuration by re-auth.""" """Handle configuration by re-auth."""
self._username = config[CONF_USERNAME] self._username = config[CONF_USERNAME]
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()

View File

@ -1,6 +1,7 @@
"""Config flow to configure the Synology DSM integration.""" """Config flow to configure the Synology DSM integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Mapping
from ipaddress import ip_address from ipaddress import ip_address
import logging import logging
from typing import Any from typing import Any
@ -120,7 +121,7 @@ class SynologyDSMFlowHandler(ConfigFlow, domain=DOMAIN):
"""Initialize the synology_dsm config flow.""" """Initialize the synology_dsm config flow."""
self.saved_user_input: dict[str, Any] = {} self.saved_user_input: dict[str, Any] = {}
self.discovered_conf: dict[str, Any] = {} self.discovered_conf: dict[str, Any] = {}
self.reauth_conf: dict[str, Any] = {} self.reauth_conf: Mapping[str, Any] = {}
self.reauth_reason: str | None = None self.reauth_reason: str | None = None
def _show_form( def _show_form(
@ -299,9 +300,9 @@ class SynologyDSMFlowHandler(ConfigFlow, domain=DOMAIN):
user_input = {**self.discovered_conf, **user_input} user_input = {**self.discovered_conf, **user_input}
return await self.async_validate_input_create_entry(user_input, step_id=step) return await self.async_validate_input_create_entry(user_input, step_id=step)
async def async_step_reauth(self, data: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, data: Mapping[str, Any]) -> FlowResult:
"""Perform reauth upon an API authentication error.""" """Perform reauth upon an API authentication error."""
self.reauth_conf = data.copy() self.reauth_conf = data
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()
async def async_step_reauth_confirm( async def async_step_reauth_confirm(

View File

@ -2,6 +2,7 @@
from __future__ import annotations from __future__ import annotations
import asyncio import asyncio
from collections.abc import Mapping
import logging import logging
from typing import Any from typing import Any
@ -202,7 +203,7 @@ class ConfigFlow(
return await self.async_step_authenticate() return await self.async_step_authenticate()
async def async_step_reauth(self, entry_data: dict[str, Any]) -> FlowResult: async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult:
"""Perform reauth upon an API authentication error.""" """Perform reauth upon an API authentication error."""
self._name = entry_data[CONF_HOST] self._name = entry_data[CONF_HOST]
self._input = { self._input = {