From b8071c688b679fda7f7a6ec8e40df403c351050f Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Fri, 3 Dec 2021 23:24:22 -0800 Subject: [PATCH] Correctly type the SSDP callback function (#60964) --- homeassistant/components/yeelight/scanner.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/yeelight/scanner.py b/homeassistant/components/yeelight/scanner.py index 648168ff84a..1756fbe865c 100644 --- a/homeassistant/components/yeelight/scanner.py +++ b/homeassistant/components/yeelight/scanner.py @@ -7,7 +7,7 @@ from ipaddress import IPv4Address, IPv6Address import logging from urllib.parse import urlparse -from async_upnp_client.search import SsdpSearchListener +from async_upnp_client.search import SsdpHeaders, SsdpSearchListener from homeassistant import config_entries from homeassistant.components import network, ssdp @@ -174,10 +174,9 @@ class YeelightScanner: # of another discovery async_call_later(self._hass, 1, _async_start_flow) - async def _async_process_entry(self, response: ssdp.SsdpServiceInfo): + async def _async_process_entry(self, headers: SsdpHeaders): """Process a discovery.""" - _LOGGER.debug("Discovered via SSDP: %s", response) - headers = response.ssdp_headers + _LOGGER.debug("Discovered via SSDP: %s", headers) unique_id = headers["id"] host = urlparse(headers["location"]).hostname current_entry = self._unique_id_capabilities.get(unique_id)