Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use dataclass properties in roku discovery () 

Co-authored-by: epenet <epenet@users.noreply.github.com>
pull/60596/head
epenet 2021-11-30 08:46:40 +01:00 committed by GitHub
parent 9374ce87ba
commit d74145ed7b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 11 deletions
homeassistant/components/roku

18
homeassistant/components/roku/config_flow.py
View File

@ -8,11 +8,7 @@ from rokuecp import Roku, RokuError
import voluptuous as vol
from homeassistant.components import ssdp, zeroconf
from homeassistant.components.ssdp import (
ATTR_SSDP_LOCATION,
ATTR_UPNP_FRIENDLY_NAME,
ATTR_UPNP_SERIAL,
)
from homeassistant.components.ssdp import ATTR_UPNP_FRIENDLY_NAME, ATTR_UPNP_SERIAL
from homeassistant.config_entries import ConfigFlow
from homeassistant.const import CONF_HOST, CONF_NAME
from homeassistant.core import HomeAssistant, callback
@ -91,9 +87,9 @@ class RokuConfigFlow(ConfigFlow, domain=DOMAIN):
# If we already have the host configured do
# not open connections to it if we can avoid it.
self._async_abort_entries_match({CONF_HOST: discovery_info[zeroconf.ATTR_HOST]})
self._async_abort_entries_match({CONF_HOST: discovery_info.host})
self.discovery_info.update({CONF_HOST: discovery_info[zeroconf.ATTR_HOST]})
self.discovery_info.update({CONF_HOST: discovery_info.host})
try:
info = await validate_input(self.hass, self.discovery_info)
@ -106,7 +102,7 @@ class RokuConfigFlow(ConfigFlow, domain=DOMAIN):
await self.async_set_unique_id(info["serial_number"])
self._abort_if_unique_id_configured(
updates={CONF_HOST: discovery_info[zeroconf.ATTR_HOST]},
updates={CONF_HOST: discovery_info.host},
)
self.context.update({"title_placeholders": {"name": info["title"]}})
@ -116,9 +112,9 @@ class RokuConfigFlow(ConfigFlow, domain=DOMAIN):
async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> FlowResult:
"""Handle a flow initialized by discovery."""
host = urlparse(discovery_info[ATTR_SSDP_LOCATION]).hostname
name = discovery_info[ATTR_UPNP_FRIENDLY_NAME]
serial_number = discovery_info[ATTR_UPNP_SERIAL]
host = urlparse(discovery_info.ssdp_location).hostname
name = discovery_info.upnp[ATTR_UPNP_FRIENDLY_NAME]
serial_number = discovery_info.upnp[ATTR_UPNP_SERIAL]
await self.async_set_unique_id(serial_number)
self._abort_if_unique_id_configured(updates={CONF_HOST: host})