Improve sonos ConfigFlow registration (#67110)

This commit is contained in:
Marc Mueller 2022-02-23 15:21:46 +01:00 committed by GitHub
parent f82d1a88e8
commit 0a6e30e4b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,7 +2,6 @@
from collections.abc import Awaitable from collections.abc import Awaitable
import dataclasses import dataclasses
from homeassistant import config_entries
from homeassistant.components import ssdp, zeroconf from homeassistant.components import ssdp, zeroconf
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResult from homeassistant.data_entry_flow import FlowResult
@ -17,7 +16,7 @@ async def _async_has_devices(hass: HomeAssistant) -> bool:
return bool(await ssdp.async_get_discovery_info_by_st(hass, UPNP_ST)) return bool(await ssdp.async_get_discovery_info_by_st(hass, UPNP_ST))
class SonosDiscoveryFlowHandler(DiscoveryFlowHandler[Awaitable[bool]]): class SonosDiscoveryFlowHandler(DiscoveryFlowHandler[Awaitable[bool]], domain=DOMAIN):
"""Sonos discovery flow that callsback zeroconf updates.""" """Sonos discovery flow that callsback zeroconf updates."""
def __init__(self) -> None: def __init__(self) -> None:
@ -43,6 +42,3 @@ class SonosDiscoveryFlowHandler(DiscoveryFlowHandler[Awaitable[bool]]):
"Zeroconf", properties, host, uid, boot_seqnum, model, mdns_name "Zeroconf", properties, host, uid, boot_seqnum, model, mdns_name
) )
return await self.async_step_discovery(dataclasses.asdict(discovery_info)) return await self.async_step_discovery(dataclasses.asdict(discovery_info))
config_entries.HANDLERS.register(DOMAIN)(SonosDiscoveryFlowHandler)