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:
"""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},
}