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

Use dataclass properties in plugwise discovery (#60631)

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

16
homeassistant/components/plugwise/config_flow.py
View File

@ -109,14 +109,12 @@ class PlugwiseConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
) -> FlowResult: ) -> FlowResult:
"""Prepare configuration for a discovered Plugwise Smile.""" """Prepare configuration for a discovered Plugwise Smile."""
self.discovery_info = discovery_info self.discovery_info = discovery_info
_properties = discovery_info[zeroconf.ATTR_PROPERTIES] _properties = discovery_info.properties
# unique_id is needed here, to be able to determine whether the discovered device is known, or not. # unique_id is needed here, to be able to determine whether the discovered device is known, or not.
unique_id = discovery_info[zeroconf.ATTR_HOSTNAME].split(".")[0] unique_id = discovery_info.hostname.split(".")[0]
await self.async_set_unique_id(unique_id) await self.async_set_unique_id(unique_id)
self._abort_if_unique_id_configured( self._abort_if_unique_id_configured({CONF_HOST: discovery_info.host})
{CONF_HOST: discovery_info[zeroconf.ATTR_HOST]}
)
if DEFAULT_USERNAME not in unique_id: if DEFAULT_USERNAME not in unique_id:
self._username = STRETCH_USERNAME self._username = STRETCH_USERNAME
@ -125,9 +123,9 @@ class PlugwiseConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
_name = f"{ZEROCONF_MAP.get(_product, _product)} v{_version}" _name = f"{ZEROCONF_MAP.get(_product, _product)} v{_version}"
self.context["title_placeholders"] = { self.context["title_placeholders"] = {
CONF_HOST: discovery_info[zeroconf.ATTR_HOST], CONF_HOST: discovery_info.host,
CONF_NAME: _name, CONF_NAME: _name,
CONF_PORT: discovery_info[zeroconf.ATTR_PORT], CONF_PORT: discovery_info.port,
CONF_USERNAME: self._username, CONF_USERNAME: self._username,
} }
return await self.async_step_user_gateway() return await self.async_step_user_gateway()
@ -143,8 +141,8 @@ class PlugwiseConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
user_input.pop(FLOW_TYPE, None) user_input.pop(FLOW_TYPE, None)
if self.discovery_info: if self.discovery_info:
user_input[CONF_HOST] = self.discovery_info[zeroconf.ATTR_HOST] user_input[CONF_HOST] = self.discovery_info.host
user_input[CONF_PORT] = self.discovery_info[zeroconf.ATTR_PORT] user_input[CONF_PORT] = self.discovery_info.port
user_input[CONF_USERNAME] = self._username user_input[CONF_USERNAME] = self._username
try: try: