From 799112205214d340e3377da93c91e088ad0660e1 Mon Sep 17 00:00:00 2001 From: epenet Date: Tue, 30 Nov 2021 09:32:07 +0000 Subject: [PATCH] Use dataclass properties in gogogate2 discovery --- homeassistant/components/gogogate2/config_flow.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/gogogate2/config_flow.py b/homeassistant/components/gogogate2/config_flow.py index e71046f797470d..e97b62102c4f79 100644 --- a/homeassistant/components/gogogate2/config_flow.py +++ b/homeassistant/components/gogogate2/config_flow.py @@ -40,16 +40,16 @@ async def async_step_homekit( ) -> data_entry_flow.FlowResult: """Handle homekit discovery.""" await self.async_set_unique_id( - discovery_info[zeroconf.ATTR_PROPERTIES][zeroconf.ATTR_PROPERTIES_ID] + discovery_info.properties[zeroconf.ATTR_PROPERTIES_ID] ) - return await self._async_discovery_handler(discovery_info[zeroconf.ATTR_HOST]) + return await self._async_discovery_handler(discovery_info.host) async def async_step_dhcp( self, discovery_info: dhcp.DhcpServiceInfo ) -> data_entry_flow.FlowResult: """Handle dhcp discovery.""" - await self.async_set_unique_id(discovery_info[dhcp.MAC_ADDRESS]) - return await self._async_discovery_handler(discovery_info[dhcp.IP_ADDRESS]) + await self.async_set_unique_id(discovery_info.macaddress) + return await self._async_discovery_handler(discovery_info.ip) async def _async_discovery_handler(self, ip_address): """Start the user flow from any discovery."""