From 601ad8f71aa98330692b8085bcd09079eedccf87 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 30 Nov 2021 16:28:49 +0100 Subject: [PATCH] Use dataclass properties in xiaomi_aqara discovery (#60636) Co-authored-by: epenet --- homeassistant/components/xiaomi_aqara/config_flow.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/xiaomi_aqara/config_flow.py b/homeassistant/components/xiaomi_aqara/config_flow.py index 2b486cb87ad..f9f8a761321 100644 --- a/homeassistant/components/xiaomi_aqara/config_flow.py +++ b/homeassistant/components/xiaomi_aqara/config_flow.py @@ -150,9 +150,9 @@ class XiaomiAqaraFlowHandler(config_entries.ConfigFlow, domain=DOMAIN): self, discovery_info: zeroconf.ZeroconfServiceInfo ) -> FlowResult: """Handle zeroconf discovery.""" - name = discovery_info[zeroconf.ATTR_NAME] - self.host = discovery_info[zeroconf.ATTR_HOST] - mac_address = discovery_info[zeroconf.ATTR_PROPERTIES].get("mac") + name = discovery_info.name + self.host = discovery_info.host + mac_address = discovery_info.properties.get("mac") if not name or not self.host or not mac_address: return self.async_abort(reason="not_xiaomi_aqara")