jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-14 08:47:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjust config-flow type hints in sharkiq (#72688)

Browse Source
This commit is contained in:
epenet 2022-05-30 16:35:50 +02:00 committed by GitHub
parent 5273e3ea9d
commit f2fde5c1f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
homeassistant/components/sharkiq/config_flow.py
View File

@ -2,6 +2,8 @@
from __future__ import annotations from __future__ import annotations
import asyncio import asyncio
from collections.abc import Mapping
from typing import Any
import aiohttp import aiohttp
import async_timeout import async_timeout
@ -10,6 +12,7 @@ import voluptuous as vol
from homeassistant import config_entries, core, exceptions from homeassistant import config_entries, core, exceptions
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.data_entry_flow import FlowResult
from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import DOMAIN, LOGGER from .const import DOMAIN, LOGGER
@ -19,7 +22,9 @@ SHARKIQ_SCHEMA = vol.Schema(
) )
async def validate_input(hass: core.HomeAssistant, data): async def _validate_input(
hass: core.HomeAssistant, data: Mapping[str, Any]
) -> dict[str, str]:
"""Validate the user input allows us to connect.""" """Validate the user input allows us to connect."""
ayla_api = get_ayla_api( ayla_api = get_ayla_api(
username=data[CONF_USERNAME], username=data[CONF_USERNAME],
@ -45,14 +50,16 @@ class SharkIqConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
VERSION = 1 VERSION = 1
async def _async_validate_input(self, user_input): async def _async_validate_input(
self, user_input: Mapping[str, Any]
) -> tuple[dict[str, str] | None, dict[str, str]]:
"""Validate form input.""" """Validate form input."""
errors = {} errors = {}
info = None info = None
# noinspection PyBroadException # noinspection PyBroadException
try: try:
info = await validate_input(self.hass, user_input) info = await _validate_input(self.hass, user_input)
except CannotConnect: except CannotConnect:
errors["base"] = "cannot_connect" errors["base"] = "cannot_connect"
except InvalidAuth: except InvalidAuth:
@ -62,9 +69,11 @@ class SharkIqConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
errors["base"] = "unknown" errors["base"] = "unknown"
return info, errors return info, errors
async def async_step_user(self, user_input: dict | None = None): async def async_step_user(
self, user_input: dict[str, str] | None = None
) -> FlowResult:
"""Handle the initial step.""" """Handle the initial step."""
errors = {} errors: dict[str, str] = {}
if user_input is not None: if user_input is not None:
info, errors = await self._async_validate_input(user_input) info, errors = await self._async_validate_input(user_input)
if info: if info:
@ -76,9 +85,9 @@ class SharkIqConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
step_id="user", data_schema=SHARKIQ_SCHEMA, errors=errors step_id="user", data_schema=SHARKIQ_SCHEMA, errors=errors
) )
async def async_step_reauth(self, user_input: dict | None = None): async def async_step_reauth(self, user_input: Mapping[str, Any]) -> FlowResult:
"""Handle re-auth if login is invalid.""" """Handle re-auth if login is invalid."""
errors = {} errors: dict[str, str] = {}
if user_input is not None: if user_input is not None:
_, errors = await self._async_validate_input(user_input) _, errors = await self._async_validate_input(user_input)