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 self, discovery_info: zeroconf.ZeroconfServiceInfo
) -> FlowResult: ) -> FlowResult:
"""Handle device found via zeroconf.""" """Handle device found via zeroconf."""
service_type = discovery_info[zeroconf.ATTR_TYPE] service_type = discovery_info.type
properties = discovery_info[zeroconf.ATTR_PROPERTIES] properties = discovery_info.properties
if service_type == "_mediaremotetv._tcp.local.": if service_type == "_mediaremotetv._tcp.local.":
identifier = properties["UniqueIdentifier"] identifier = properties["UniqueIdentifier"]
name = properties["Name"] name = properties["Name"]
elif service_type == "_touch-able._tcp.local.": elif service_type == "_touch-able._tcp.local.":
identifier = discovery_info[zeroconf.ATTR_NAME].split(".")[0] identifier = discovery_info.name.split(".")[0]
name = properties["CtlN"] name = properties["CtlN"]
else: else:
return self.async_abort(reason="unknown") return self.async_abort(reason="unknown")