From 63ed0af644af7338a5ff7512d27b4255a6010935 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 1 Dec 2021 14:00:46 +0100 Subject: [PATCH] Use dataclass properties in hyperion discovery (#60714) Co-authored-by: epenet --- homeassistant/components/hyperion/config_flow.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/hyperion/config_flow.py b/homeassistant/components/hyperion/config_flow.py index f5beb185830..3b5785dd533 100644 --- a/homeassistant/components/hyperion/config_flow.py +++ b/homeassistant/components/hyperion/config_flow.py @@ -11,7 +11,6 @@ from hyperion import client, const import voluptuous as vol from homeassistant.components import ssdp -from homeassistant.components.ssdp import ATTR_SSDP_LOCATION, ATTR_UPNP_SERIAL from homeassistant.config_entries import ( SOURCE_REAUTH, ConfigEntry, @@ -189,9 +188,11 @@ class HyperionConfigFlow(ConfigFlow, domain=DOMAIN): # SSDP requires user confirmation. self._require_confirm = True - self._data[CONF_HOST] = urlparse(discovery_info[ATTR_SSDP_LOCATION]).hostname + self._data[CONF_HOST] = urlparse(discovery_info.ssdp_location).hostname try: - self._port_ui = urlparse(discovery_info[ATTR_SSDP_LOCATION]).port + self._port_ui = ( + urlparse(discovery_info.ssdp_location).port or const.DEFAULT_PORT_UI + ) except ValueError: self._port_ui = const.DEFAULT_PORT_UI @@ -204,7 +205,7 @@ class HyperionConfigFlow(ConfigFlow, domain=DOMAIN): except ValueError: self._data[CONF_PORT] = const.DEFAULT_PORT_JSON - if not (hyperion_id := discovery_info.get(ATTR_UPNP_SERIAL)): + if not (hyperion_id := discovery_info.get(ssdp.ATTR_UPNP_SERIAL)): return self.async_abort(reason="no_id") # For discovery mechanisms, we set the unique_id as early as possible to