Use dataclass properties in apple_tv discovery (#60557)

This commit is contained in:
epenet 2021-11-30 08:31:25 +01:00 committed by GitHub
parent 0e3a229a1f
commit bb92dd2467
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -147,14 +147,14 @@ class AppleTVConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
self, discovery_info: zeroconf.ZeroconfServiceInfo
) -> FlowResult:
"""Handle device found via zeroconf."""
service_type = discovery_info[zeroconf.ATTR_TYPE]
properties = discovery_info[zeroconf.ATTR_PROPERTIES]
service_type = discovery_info.type
properties = discovery_info.properties
if service_type == "_mediaremotetv._tcp.local.":
identifier = properties["UniqueIdentifier"]
name = properties["Name"]
elif service_type == "_touch-able._tcp.local.":
identifier = discovery_info[zeroconf.ATTR_NAME].split(".")[0]
identifier = discovery_info.name.split(".")[0]
name = properties["CtlN"]
else:
return self.async_abort(reason="unknown")