From f0df3e4646c2c730100d133fea9b674deba0a2e2 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 30 Nov 2021 11:01:28 +0100 Subject: [PATCH] Use dataclass properties in smappee discovery (#60602) Co-authored-by: epenet --- homeassistant/components/smappee/config_flow.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/smappee/config_flow.py b/homeassistant/components/smappee/config_flow.py index 2fb5abaa05e..e57071b4938 100644 --- a/homeassistant/components/smappee/config_flow.py +++ b/homeassistant/components/smappee/config_flow.py @@ -42,15 +42,11 @@ class SmappeeFlowHandler( ) -> FlowResult: """Handle zeroconf discovery.""" - if not discovery_info[zeroconf.ATTR_HOSTNAME].startswith( - SUPPORTED_LOCAL_DEVICES - ): + if not discovery_info.hostname.startswith(SUPPORTED_LOCAL_DEVICES): return self.async_abort(reason="invalid_mdns") - serial_number = ( - discovery_info[zeroconf.ATTR_HOSTNAME] - .replace(".local.", "") - .replace("Smappee", "") + serial_number = discovery_info.hostname.replace(".local.", "").replace( + "Smappee", "" ) # Check if already configured (local) @@ -63,7 +59,7 @@ class SmappeeFlowHandler( self.context.update( { - CONF_IP_ADDRESS: discovery_info[zeroconf.ATTR_HOST], + CONF_IP_ADDRESS: discovery_info.host, CONF_SERIALNUMBER: serial_number, "title_placeholders": {"name": serial_number}, }