Use dataclass properties in smappee discovery (#60602)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-11-30 11:01:28 +01:00 committed by GitHub
parent ba1cc00c24
commit f0df3e4646
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -42,15 +42,11 @@ class SmappeeFlowHandler(
) -> FlowResult: ) -> FlowResult:
"""Handle zeroconf discovery.""" """Handle zeroconf discovery."""
if not discovery_info[zeroconf.ATTR_HOSTNAME].startswith( if not discovery_info.hostname.startswith(SUPPORTED_LOCAL_DEVICES):
SUPPORTED_LOCAL_DEVICES
):
return self.async_abort(reason="invalid_mdns") return self.async_abort(reason="invalid_mdns")
serial_number = ( serial_number = discovery_info.hostname.replace(".local.", "").replace(
discovery_info[zeroconf.ATTR_HOSTNAME] "Smappee", ""
.replace(".local.", "")
.replace("Smappee", "")
) )
# Check if already configured (local) # Check if already configured (local)
@ -63,7 +59,7 @@ class SmappeeFlowHandler(
self.context.update( self.context.update(
{ {
CONF_IP_ADDRESS: discovery_info[zeroconf.ATTR_HOST], CONF_IP_ADDRESS: discovery_info.host,
CONF_SERIALNUMBER: serial_number, CONF_SERIALNUMBER: serial_number,
"title_placeholders": {"name": serial_number}, "title_placeholders": {"name": serial_number},
} }