diff --git a/homeassistant/util/network.py b/homeassistant/util/network.py index b35a8063f..48840f339 100644 --- a/homeassistant/util/network.py +++ b/homeassistant/util/network.py @@ -1,6 +1,6 @@ """Network utilities.""" -from ipaddress import ip_address, ip_network - +from ipaddress import IPv4Address, IPv6Address, ip_address, ip_network +from typing import Union # IP addresses of loopback interfaces LOCAL_IPS = ( @@ -16,7 +16,7 @@ LOCAL_NETWORKS = ( ) -def is_local(address: ip_address) -> bool: +def is_local(address: Union[IPv4Address, IPv6Address]) -> bool: """Check if an address is local.""" return address in LOCAL_IPS or \ any(address in network for network in LOCAL_NETWORKS)