Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix async_scanner_devices_by_address unexpectedly combining Bluetooth scanners #94990

Merged
merged 3 commits into from
Jun 22, 2023
Merged

Fix async_scanner_devices_by_address unexpectedly combining Bluetooth scanners #94990

Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
3ad4ad6
Fix async_scanner_devices_by_address unexpectedly combining scanners
bdraco Jun 21, 2023
930e5c0
Update homeassistant/components/bluetooth/manager.py
bdraco Jun 21, 2023
8a4a936
use chain instead
bdraco Jun 21, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
68 changes: 29 additions & 39 deletions homeassistant/components/bluetooth/manager.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -246,9 +246,9 @@ def async_scanner_devices_by_address(
self, address: str, connectable: bool
) -> list[BluetoothScannerDevice]:
"""Get BluetoothScannerDevice by address."""
scanners = self._get_scanners_by_type(True)
scanners = [*self._connectable_scanners]
bdraco marked this conversation as resolved.
Show resolved Hide resolved
if not connectable:
scanners.extend(self._get_scanners_by_type(False))
scanners.extend(self._non_connectable_scanners)
return [
BluetoothScannerDevice(scanner, *device_adv)
for scanner in scanners
Expand All @@ -267,21 +267,19 @@ def _async_all_discovered_addresses(self, connectable: bool) -> Iterable[str]:
"""
yield from itertools.chain.from_iterable(
scanner.discovered_devices_and_advertisement_data
for scanner in self._get_scanners_by_type(True)
for scanner in self._connectable_scanners
)
if not connectable:
yield from itertools.chain.from_iterable(
scanner.discovered_devices_and_advertisement_data
for scanner in self._get_scanners_by_type(False)
for scanner in self._non_connectable_scanners
)

@hass_callback
def async_discovered_devices(self, connectable: bool) -> list[BLEDevice]:
"""Return all of combined best path to discovered from all the scanners."""
return [
history.device
for history in self._get_history_by_type(connectable).values()
]
histories = self._connectable_history if connectable else self._all_history
return [history.device for history in histories.values()]

@hass_callback
def async_setup_unavailable_tracking(self) -> None:
Expand All @@ -303,7 +301,10 @@ def _async_check_unavailable(self, now: datetime) -> None:
intervals = tracker.intervals

for connectable in (True, False):
unavailable_callbacks = self._get_unavailable_callbacks_by_type(connectable)
if connectable:
unavailable_callbacks = self._connectable_unavailable_callbacks
else:
unavailable_callbacks = self._unavailable_callbacks
history = connectable_history if connectable else all_history
disappeared = set(history).difference(
self._async_all_discovered_addresses(connectable)
Expand Down Expand Up @@ -583,7 +584,10 @@ def async_track_unavailable(
connectable: bool,
) -> Callable[[], None]:
"""Register a callback."""
unavailable_callbacks = self._get_unavailable_callbacks_by_type(connectable)
if connectable:
unavailable_callbacks = self._connectable_unavailable_callbacks
else:
unavailable_callbacks = self._unavailable_callbacks
unavailable_callbacks.setdefault(address, []).append(callback)

@hass_callback
Expand Down Expand Up @@ -620,13 +624,13 @@ def _async_remove_callback() -> None:
# If we have history for the subscriber, we can trigger the callback
# immediately with the last packet so the subscriber can see the
# device.
all_history = self._get_history_by_type(connectable)
history = self._connectable_history if connectable else self._all_history
service_infos: Iterable[BluetoothServiceInfoBleak] = []
if address := callback_matcher.get(ADDRESS):
if service_info := all_history.get(address):
if service_info := history.get(address):
service_infos = [service_info]
else:
service_infos = all_history.values()
service_infos = history.values()

for service_info in service_infos:
if ble_device_matches(callback_matcher, service_info):
Expand All @@ -642,29 +646,32 @@ def async_ble_device_from_address(
self, address: str, connectable: bool
) -> BLEDevice | None:
"""Return the BLEDevice if present."""
all_history = self._get_history_by_type(connectable)
if history := all_history.get(address):
histories = self._connectable_history if connectable else self._all_history
if history := histories.get(address):
return history.device
return None

@hass_callback
def async_address_present(self, address: str, connectable: bool) -> bool:
"""Return if the address is present."""
return address in self._get_history_by_type(connectable)
histories = self._connectable_history if connectable else self._all_history
return address in histories

@hass_callback
def async_discovered_service_info(
self, connectable: bool
) -> Iterable[BluetoothServiceInfoBleak]:
"""Return all the discovered services info."""
return self._get_history_by_type(connectable).values()
histories = self._connectable_history if connectable else self._all_history
return histories.values()

@hass_callback
def async_last_service_info(
self, address: str, connectable: bool
) -> BluetoothServiceInfoBleak | None:
"""Return the last service info for an address."""
return self._get_history_by_type(connectable).get(address)
histories = self._connectable_history if connectable else self._all_history
return histories.get(address)

def _async_trigger_matching_discovery(
self, service_info: BluetoothServiceInfoBleak
Expand All @@ -688,26 +695,6 @@ def async_rediscover_address(self, address: str) -> None:
if service_info := self._all_history.get(address):
self._async_trigger_matching_discovery(service_info)

def _get_scanners_by_type(self, connectable: bool) -> list[BaseHaScanner]:
"""Return the scanners by type."""
if connectable:
return self._connectable_scanners
return self._non_connectable_scanners

def _get_unavailable_callbacks_by_type(
self, connectable: bool
) -> dict[str, list[Callable[[BluetoothServiceInfoBleak], None]]]:
"""Return the unavailable callbacks by type."""
if connectable:
return self._connectable_unavailable_callbacks
return self._unavailable_callbacks

def _get_history_by_type(
self, connectable: bool
) -> dict[str, BluetoothServiceInfoBleak]:
"""Return the history by type."""
return self._connectable_history if connectable else self._all_history

def async_register_scanner(
self,
scanner: BaseHaScanner,
Expand All @@ -716,7 +703,10 @@ def async_register_scanner(
) -> CALLBACK_TYPE:
"""Register a new scanner."""
_LOGGER.debug("Registering scanner %s", scanner.name)
scanners = self._get_scanners_by_type(connectable)
if connectable:
scanners = self._connectable_scanners
else:
scanners = self._non_connectable_scanners

def _unregister_scanner() -> None:
_LOGGER.debug("Unregistering scanner %s", scanner.name)
Expand Down