From 3db3f264c29937603f13a43725da43c7861911d2 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 1 Dec 2021 18:31:51 +0100 Subject: [PATCH] Use dataclass properties in nuki discovery (#60731) Co-authored-by: epenet --- homeassistant/components/nuki/config_flow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/nuki/config_flow.py b/homeassistant/components/nuki/config_flow.py index c3e0ed93372..bd2c5a0d750 100644 --- a/homeassistant/components/nuki/config_flow.py +++ b/homeassistant/components/nuki/config_flow.py @@ -69,13 +69,13 @@ class NukiConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): async def async_step_dhcp(self, discovery_info: dhcp.DhcpServiceInfo) -> FlowResult: """Prepare configuration for a DHCP discovered Nuki bridge.""" - await self.async_set_unique_id(int(discovery_info[dhcp.HOSTNAME][12:], 16)) + await self.async_set_unique_id(int(discovery_info.hostname[12:], 16)) self._abort_if_unique_id_configured() self.discovery_schema = vol.Schema( { - vol.Required(CONF_HOST, default=discovery_info[dhcp.IP_ADDRESS]): str, + vol.Required(CONF_HOST, default=discovery_info.ip): str, vol.Required(CONF_PORT, default=DEFAULT_PORT): int, vol.Required(CONF_TOKEN): str, }