evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor dhcp to move all mac formatting into the client processor (#110509)

Browse source
This commit is contained in:
J. Nick Koston 2024-02-13 19:12:38 -06:00 • committed by GitHub
parent 9a8c5af0a6
commit b3452dc3f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
homeassistant/components/dhcp/__init__.py
View file

@ -186,7 +186,7 @@ class WatcherBase(ABC):
@callback
def async_process_client(
self, ip_address: str, hostname: str, mac_address: str
self, ip_address: str, hostname: str, unformatted_mac_address: str
) -> None:
"""Process a client."""
if (made_ip_address := cached_ip_addresses(ip_address)) is None:
@ -202,6 +202,12 @@ class WatcherBase(ABC):
# Ignore self assigned addresses, loopback, invalid
return
formatted_mac = format_mac(unformatted_mac_address)
# Historically, the MAC address was formatted without colons
# and since all consumers of this data are expecting it to be
# formatted without colons we will continue to do so
mac_address = formatted_mac.replace(":", "")
data = self._address_data.get(ip_address)
if (
data
@ -231,7 +237,7 @@ class WatcherBase(ABC):
dev_reg: DeviceRegistry = async_get(self.hass)
if device := dev_reg.async_get_device(
connections={(CONNECTION_NETWORK_MAC, uppercase_mac)}
connections={(CONNECTION_NETWORK_MAC, formatted_mac)}
):
for entry_id in device.config_entries:
if (
@ -319,7 +325,7 @@ class NetworkWatcher(WatcherBase):
self.async_process_client(
host[DISCOVERY_IP_ADDRESS],
host[DISCOVERY_HOSTNAME],
_format_mac(host[DISCOVERY_MAC_ADDRESS]),
host[DISCOVERY_MAC_ADDRESS],
)
@ -360,7 +366,7 @@ class DeviceTrackerWatcher(WatcherBase):
if ip_address is None or mac_address is None:
return
self.async_process_client(ip_address, hostname, _format_mac(mac_address))
self.async_process_client(ip_address, hostname, mac_address)
class DeviceTrackerRegisteredWatcher(WatcherBase):
@ -383,7 +389,7 @@ class DeviceTrackerRegisteredWatcher(WatcherBase):
if ip_address is None or mac_address is None:
return
self.async_process_client(ip_address, hostname, _format_mac(mac_address))
self.async_process_client(ip_address, hostname, mac_address)
class DHCPWatcher(WatcherBase):
@ -393,7 +399,7 @@ class DHCPWatcher(WatcherBase):
def _async_process_dhcp_request(self, response: aiodhcpwatcher.DHCPRequest) -> None:
"""Process a dhcp request."""
self.async_process_client(
response.ip_address, response.hostname, _format_mac(response.mac_address)
response.ip_address, response.hostname, response.mac_address
)
@callback
@ -402,11 +408,6 @@ class DHCPWatcher(WatcherBase):
self._unsub = aiodhcpwatcher.start(self._async_process_dhcp_request)
def _format_mac(mac_address: str) -> str:
"""Format a mac address for matching."""
return format_mac(mac_address).replace(":", "")
@lru_cache(maxsize=4096, typed=True)
def _compile_fnmatch(pattern: str) -> re.Pattern:
"""Compile a fnmatch pattern."""