From 9374ce87ba12e1b2ae08508805786dd6df631e00 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 30 Nov 2021 08:42:30 +0100 Subject: [PATCH] Use dataclass properties in wled discovery (#60573) Co-authored-by: epenet --- homeassistant/components/wled/config_flow.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/wled/config_flow.py b/homeassistant/components/wled/config_flow.py index b2112e8962b..485afef4f6c 100644 --- a/homeassistant/components/wled/config_flow.py +++ b/homeassistant/components/wled/config_flow.py @@ -44,14 +44,14 @@ class WLEDFlowHandler(ConfigFlow, domain=DOMAIN): """Handle zeroconf discovery.""" # Hostname is format: wled-livingroom.local. - host = discovery_info[zeroconf.ATTR_HOSTNAME].rstrip(".") + host = discovery_info.hostname.rstrip(".") name, _ = host.rsplit(".") self.context.update( { - CONF_HOST: discovery_info[zeroconf.ATTR_HOST], + CONF_HOST: discovery_info.host, CONF_NAME: name, - CONF_MAC: discovery_info[zeroconf.ATTR_PROPERTIES].get(CONF_MAC), + CONF_MAC: discovery_info.properties.get(CONF_MAC), "title_placeholders": {"name": name}, } )