evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve type hints in smlight config flow

Browse source
This commit is contained in:
epenet 2024-11-12 13:49:37 +00:00
parent 7045b776b6
commit 9893347dc3
1 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
homeassistant/components/smlight/config_flow.py
View file

@ -34,10 +34,11 @@ STEP_AUTH_DATA_SCHEMA = vol.Schema(
class SmlightConfigFlow(ConfigFlow, domain=DOMAIN): class SmlightConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow for SMLIGHT Zigbee.""" """Handle a config flow for SMLIGHT Zigbee."""
host: str
def __init__(self) -> None: def __init__(self) -> None:
"""Initialize the config flow.""" """Initialize the config flow."""
self.client: Api2 self.client: Api2
self.host: str | None = None
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
@ -46,9 +47,8 @@ class SmlightConfigFlow(ConfigFlow, domain=DOMAIN):
errors: dict[str, str] = {} errors: dict[str, str] = {}
if user_input is not None: if user_input is not None:
host = user_input[CONF_HOST] self.host = user_input[CONF_HOST]
self.client = Api2(host, session=async_get_clientsession(self.hass)) self.client = Api2(self.host, session=async_get_clientsession(self.hass))
self.host = host
try: try:
if not await self._async_check_auth_required(user_input): if not await self._async_check_auth_required(user_input):
@ -138,9 +138,8 @@ class SmlightConfigFlow(ConfigFlow, domain=DOMAIN):
) -> ConfigFlowResult: ) -> ConfigFlowResult:
"""Handle reauth when API Authentication failed.""" """Handle reauth when API Authentication failed."""
host = entry_data[CONF_HOST] self.host = entry_data[CONF_HOST]
self.client = Api2(host, session=async_get_clientsession(self.hass)) self.client = Api2(self.host, session=async_get_clientsession(self.hass))
self.host = host
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()