From ffb9b4cd2d3138438227832ebbd3544162f59215 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 30 Nov 2021 16:16:30 +0100 Subject: [PATCH] Use dataclass properties in hunterdouglas discovery (#60605) --- .../hunterdouglas_powerview/config_flow.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/hunterdouglas_powerview/config_flow.py b/homeassistant/components/hunterdouglas_powerview/config_flow.py index d2c78227fe..e0ebef7abb 100644 --- a/homeassistant/components/hunterdouglas_powerview/config_flow.py +++ b/homeassistant/components/hunterdouglas_powerview/config_flow.py @@ -88,16 +88,16 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): async def async_step_dhcp(self, discovery_info: dhcp.DhcpServiceInfo) -> FlowResult: """Handle DHCP discovery.""" - self.discovered_ip = discovery_info[dhcp.IP_ADDRESS] - self.discovered_name = discovery_info[dhcp.HOSTNAME] + self.discovered_ip = discovery_info.ip + self.discovered_name = discovery_info.hostname return await self.async_step_discovery_confirm() async def async_step_zeroconf( self, discovery_info: zeroconf.ZeroconfServiceInfo ) -> FlowResult: """Handle zeroconf discovery.""" - self.discovered_ip = discovery_info[zeroconf.ATTR_HOST] - name = discovery_info[zeroconf.ATTR_NAME] + self.discovered_ip = discovery_info.host + name = discovery_info.name if name.endswith(POWERVIEW_SUFFIX): name = name[: -len(POWERVIEW_SUFFIX)] self.discovered_name = name @@ -107,8 +107,8 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): self, discovery_info: zeroconf.ZeroconfServiceInfo ) -> FlowResult: """Handle HomeKit discovery.""" - self.discovered_ip = discovery_info[zeroconf.ATTR_HOST] - name = discovery_info[zeroconf.ATTR_NAME] + self.discovered_ip = discovery_info.host + name = discovery_info.name if name.endswith(HAP_SUFFIX): name = name[: -len(HAP_SUFFIX)] self.discovered_name = name