From 160e6febc3a8b8f171927c8407ccbf632de3d2ae Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 1 Dec 2021 11:08:36 +0100 Subject: [PATCH] Use dataclass properties in directv discovery (#60692) Co-authored-by: epenet --- homeassistant/components/directv/config_flow.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/directv/config_flow.py b/homeassistant/components/directv/config_flow.py index 338af62364d..34a09a04811 100644 --- a/homeassistant/components/directv/config_flow.py +++ b/homeassistant/components/directv/config_flow.py @@ -9,7 +9,6 @@ from directv import DIRECTV, DIRECTVError 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 ConfigFlow from homeassistant.const import CONF_HOST, CONF_NAME from homeassistant.core import HomeAssistant @@ -69,11 +68,13 @@ class DirecTVConfigFlow(ConfigFlow, domain=DOMAIN): async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> FlowResult: """Handle SSDP discovery.""" - host = urlparse(discovery_info[ATTR_SSDP_LOCATION]).hostname + host = urlparse(discovery_info.ssdp_location).hostname receiver_id = None - if discovery_info.get(ATTR_UPNP_SERIAL): - receiver_id = discovery_info[ATTR_UPNP_SERIAL][4:] # strips off RID- + if discovery_info.upnp.get(ssdp.ATTR_UPNP_SERIAL): + receiver_id = discovery_info.upnp[ssdp.ATTR_UPNP_SERIAL][ + 4: + ] # strips off RID- self.context.update({"title_placeholders": {"name": host}})