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

Use dataclass properties in nut discovery (#60603)

Browse Source 
Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-11-30 10:17:49 +01:00 committed by GitHub
parent 071296865b
commit efebd1b657
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/nut/config_flow.py
View File

@ -1,4 +1,6 @@
"""Config flow for Network UPS Tools (NUT) integration.""" """Config flow for Network UPS Tools (NUT) integration."""
from __future__ import annotations
import logging import logging
import voluptuous as vol import voluptuous as vol
@ -23,7 +25,7 @@ from .const import DEFAULT_HOST, DEFAULT_PORT, DEFAULT_SCAN_INTERVAL, DOMAIN
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
def _base_schema(discovery_info): def _base_schema(discovery_info: zeroconf.ZeroconfServiceInfo | None) -> vol.Schema:
"""Generate base schema.""" """Generate base schema."""
base_schema = {} base_schema = {}
if not discovery_info: if not discovery_info:
@ -83,7 +85,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
def __init__(self): def __init__(self):
"""Initialize the nut config flow.""" """Initialize the nut config flow."""
self.nut_config = {} self.nut_config = {}
self.discovery_info = {} self.discovery_info: zeroconf.ZeroconfServiceInfo | None = None
self.ups_list = None self.ups_list = None
self.title = None self.title = None
@ -94,8 +96,8 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
self.discovery_info = discovery_info self.discovery_info = discovery_info
await self._async_handle_discovery_without_unique_id() await self._async_handle_discovery_without_unique_id()
self.context["title_placeholders"] = { self.context["title_placeholders"] = {
CONF_PORT: discovery_info[zeroconf.ATTR_PORT] or DEFAULT_PORT, CONF_PORT: discovery_info.port or DEFAULT_PORT,
CONF_HOST: discovery_info[zeroconf.ATTR_HOST], CONF_HOST: discovery_info.host,
} }
return await self.async_step_user() return await self.async_step_user()
@ -106,7 +108,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
if self.discovery_info: if self.discovery_info:
user_input.update( user_input.update(
{ {
CONF_HOST: self.discovery_info[CONF_HOST], CONF_HOST: self.discovery_info.host,
CONF_PORT: self.discovery_info.port or DEFAULT_PORT, CONF_PORT: self.discovery_info.port or DEFAULT_PORT,
} }
) )