From 76d906d17ee90c6bce687e80af20461c664292a1 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 1 Dec 2021 15:53:02 +0100 Subject: [PATCH] Use dataclass properties in roomba discovery (#60734) Co-authored-by: epenet --- homeassistant/components/roomba/config_flow.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/roomba/config_flow.py b/homeassistant/components/roomba/config_flow.py index e1919941068..2dbecdf32f2 100644 --- a/homeassistant/components/roomba/config_flow.py +++ b/homeassistant/components/roomba/config_flow.py @@ -80,13 +80,13 @@ class RoombaConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): async def async_step_dhcp(self, discovery_info: dhcp.DhcpServiceInfo) -> FlowResult: """Handle dhcp discovery.""" - self._async_abort_entries_match({CONF_HOST: discovery_info[dhcp.IP_ADDRESS]}) + self._async_abort_entries_match({CONF_HOST: discovery_info.ip}) - if not discovery_info[dhcp.HOSTNAME].startswith(("irobot-", "roomba-")): + if not discovery_info.hostname.startswith(("irobot-", "roomba-")): return self.async_abort(reason="not_irobot_device") - self.host = discovery_info[dhcp.IP_ADDRESS] - self.blid = _async_blid_from_hostname(discovery_info[dhcp.HOSTNAME]) + self.host = discovery_info.ip + self.blid = _async_blid_from_hostname(discovery_info.hostname) await self.async_set_unique_id(self.blid) self._abort_if_unique_id_configured(updates={CONF_HOST: self.host})