Correctly type the SSDP callback function (home-assistant#60964)

steve1011011 · Dec 4, 2021 · b8071c6 · b8071c6
1 parent 7fbe1db
commit b8071c6
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/homeassistant/components/yeelight/scanner.py b/homeassistant/components/yeelight/scanner.py
@@ -7,7 +7,7 @@
 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 @@ def _async_start_flow(*_):
         # 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)