Filter out IPv6 addresses in Govee Light Local (home-assistant#151540)

Author: abmantis

homeassistant/components/govee_light_local/__init__.py

@@ -5,6 +5,7 @@
 import asyncio
 from contextlib import suppress
 from errno import EADDRINUSE
+from ipaddress import IPv4Address
 import logging
 
 from govee_local_api.controller import LISTENING_PORT
@@ -30,7 +31,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: GoveeLocalConfigEntry) -
     _LOGGER.debug("Enabled source IPs: %s", source_ips)
 
     coordinator: GoveeLocalApiCoordinator = GoveeLocalApiCoordinator(
-        hass=hass, config_entry=entry, source_ips=source_ips
+        hass=hass,
+        config_entry=entry,
+        source_ips=[
+            source_ip for source_ip in source_ips if isinstance(source_ip, IPv4Address)
+        ],
     )
 
     async def await_cleanup():

homeassistant/components/govee_light_local/config_flow.py

@@ -4,7 +4,7 @@
 
 import asyncio
 from contextlib import suppress
-from ipaddress import IPv4Address, IPv6Address
+from ipaddress import IPv4Address
 import logging
 
 from govee_local_api import GoveeController
@@ -24,9 +24,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 
-async def _async_discover(
-    hass: HomeAssistant, adapter_ip: IPv4Address | IPv6Address
-) -> bool:
+async def _async_discover(hass: HomeAssistant, adapter_ip: IPv4Address) -> bool:
     controller: GoveeController = GoveeController(
         loop=hass.loop,
         logger=_LOGGER,
@@ -74,7 +72,9 @@ async def _async_has_devices(hass: HomeAssistant) -> bool:
     _LOGGER.debug("Enabled source IPs: %s", source_ips)
 
     # Run discovery on every IPv4 address and gather results
-    results = await asyncio.gather(*[_async_discover(hass, ip) for ip in source_ips])
+    results = await asyncio.gather(
+        *[_async_discover(hass, ip) for ip in source_ips if isinstance(ip, IPv4Address)]
+    )
 
     return any(results)
 

homeassistant/components/govee_light_local/coordinator.py

@@ -2,7 +2,7 @@
 
 import asyncio
 from collections.abc import Callable
-from ipaddress import IPv4Address, IPv6Address
+from ipaddress import IPv4Address
 import logging
 
 from govee_local_api import GoveeController, GoveeDevice
@@ -30,7 +30,7 @@ def __init__(
         self,
         hass: HomeAssistant,
         config_entry: GoveeLocalConfigEntry,
-        source_ips: list[IPv4Address | IPv6Address],
+        source_ips: list[IPv4Address],
     ) -> None:
         """Initialize my coordinator."""
         super().__init__(
@@ -114,5 +114,4 @@ def devices(self) -> list[GoveeDevice]:
     async def _async_update_data(self) -> list[GoveeDevice]:
         for controller in self._controllers:
             controller.send_update_message()
-
         return self.devices