jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 21:57:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve config flow type hints in vesync (#124351)

Browse Source
This commit is contained in:
epenet 2024-08-28 12:56:22 +02:00 committed by GitHub
parent d1681fac72
commit 0a94242337
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
homeassistant/components/vesync/config_flow.py
View File

@ -1,40 +1,42 @@
"""Config flow utilities.""" """Config flow utilities."""
from collections import OrderedDict from typing import Any
from pyvesync import VeSync from pyvesync import VeSync
import voluptuous as vol import voluptuous as vol
from homeassistant.config_entries import ConfigFlow from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import callback from homeassistant.core import callback
import homeassistant.helpers.config_validation as cv
from .const import DOMAIN from .const import DOMAIN
DATA_SCHEMA = vol.Schema(
{
vol.Required(CONF_USERNAME): cv.string,
vol.Required(CONF_PASSWORD): cv.string,
}
)
class VeSyncFlowHandler(ConfigFlow, domain=DOMAIN): class VeSyncFlowHandler(ConfigFlow, domain=DOMAIN):
"""Handle a config flow.""" """Handle a config flow."""
VERSION = 1 VERSION = 1
def __init__(self) -> None:
"""Instantiate config flow."""
self._username = None
self._password = None
self.data_schema = OrderedDict()
self.data_schema[vol.Required(CONF_USERNAME)] = str
self.data_schema[vol.Required(CONF_PASSWORD)] = str
@callback @callback
def _show_form(self, errors=None): def _show_form(self, errors: dict[str, str] | None = None) -> ConfigFlowResult:
"""Show form to the user.""" """Show form to the user."""
return self.async_show_form( return self.async_show_form(
step_id="user", step_id="user",
data_schema=vol.Schema(self.data_schema), data_schema=DATA_SCHEMA,
errors=errors if errors else {}, errors=errors if errors else {},
) )
async def async_step_user(self, user_input=None): async def async_step_user(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Handle a flow start.""" """Handle a flow start."""
if self._async_current_entries(): if self._async_current_entries():
return self.async_abort(reason="single_instance_allowed") return self.async_abort(reason="single_instance_allowed")
@ -42,15 +44,15 @@ class VeSyncFlowHandler(ConfigFlow, domain=DOMAIN):
if not user_input: if not user_input:
return self._show_form() return self._show_form()
self._username = user_input[CONF_USERNAME] username = user_input[CONF_USERNAME]
self._password = user_input[CONF_PASSWORD] password = user_input[CONF_PASSWORD]
manager = VeSync(self._username, self._password) manager = VeSync(username, password)
login = await self.hass.async_add_executor_job(manager.login) login = await self.hass.async_add_executor_job(manager.login)
if not login: if not login:
return self._show_form(errors={"base": "invalid_auth"}) return self._show_form(errors={"base": "invalid_auth"})
return self.async_create_entry( return self.async_create_entry(
title=self._username, title=username,
data={CONF_USERNAME: self._username, CONF_PASSWORD: self._password}, data={CONF_USERNAME: username, CONF_PASSWORD: password},
) )