From ae9320b61691a3d21830f954744d831cb57148fb Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 30 Nov 2021 11:05:42 +0100 Subject: [PATCH] Use dataclass properties in tradfri discovery (#60592) Co-authored-by: epenet --- homeassistant/components/tradfri/config_flow.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/tradfri/config_flow.py b/homeassistant/components/tradfri/config_flow.py index af3ed00e974..a0b63f94f4f 100644 --- a/homeassistant/components/tradfri/config_flow.py +++ b/homeassistant/components/tradfri/config_flow.py @@ -97,13 +97,11 @@ class FlowHandler(config_entries.ConfigFlow, domain=DOMAIN): ) -> FlowResult: """Handle homekit discovery.""" await self.async_set_unique_id( - discovery_info[zeroconf.ATTR_PROPERTIES][zeroconf.ATTR_PROPERTIES_ID] - ) - self._abort_if_unique_id_configured( - {CONF_HOST: discovery_info[zeroconf.ATTR_HOST]} + discovery_info.properties[zeroconf.ATTR_PROPERTIES_ID] ) + self._abort_if_unique_id_configured({CONF_HOST: discovery_info.host}) - host = discovery_info[zeroconf.ATTR_HOST] + host = discovery_info.host for entry in self._async_current_entries(): if entry.data.get(CONF_HOST) != host: @@ -113,9 +111,7 @@ class FlowHandler(config_entries.ConfigFlow, domain=DOMAIN): if not entry.unique_id: self.hass.config_entries.async_update_entry( entry, - unique_id=discovery_info[zeroconf.ATTR_PROPERTIES][ - zeroconf.ATTR_PROPERTIES_ID - ], + unique_id=discovery_info.properties[zeroconf.ATTR_PROPERTIES_ID], ) return self.async_abort(reason="already_configured")