52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
"""The yalexs_ble integration models."""
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.components.bluetooth import (
|
|
BluetoothScanningMode,
|
|
BluetoothServiceInfoBleak,
|
|
async_discovered_service_info,
|
|
async_process_advertisements,
|
|
)
|
|
from homeassistant.components.bluetooth.match import ADDRESS, BluetoothCallbackMatcher
|
|
from homeassistant.core import HomeAssistant, callback
|
|
|
|
from .const import DEVICE_TIMEOUT
|
|
|
|
|
|
@callback
|
|
def async_find_existing_service_info(
|
|
hass: HomeAssistant, local_name: str, address: str
|
|
) -> BluetoothServiceInfoBleak | None:
|
|
"""Return the service info for the given local_name and address."""
|
|
for service_info in async_discovered_service_info(hass):
|
|
device = service_info.device
|
|
if device.address == address:
|
|
return service_info
|
|
return None
|
|
|
|
|
|
async def async_get_service_info(
|
|
hass: HomeAssistant, local_name: str, address: str
|
|
) -> BluetoothServiceInfoBleak:
|
|
"""Wait for the service info for the given local_name and address."""
|
|
if service_info := async_find_existing_service_info(hass, local_name, address):
|
|
return service_info
|
|
return await async_process_advertisements(
|
|
hass,
|
|
lambda service_info: True,
|
|
BluetoothCallbackMatcher({ADDRESS: address}),
|
|
BluetoothScanningMode.ACTIVE,
|
|
DEVICE_TIMEOUT,
|
|
)
|
|
|
|
|
|
def short_address(address: str) -> str:
|
|
"""Convert a Bluetooth address to a short address."""
|
|
split_address = address.replace("-", ":").split(":")
|
|
return f"{split_address[-2].upper()}{split_address[-1].upper()}"[-4:]
|
|
|
|
|
|
def human_readable_name(name: str | None, local_name: str, address: str) -> str:
|
|
"""Return a human readable name for the given name, local_name, and address."""
|
|
return f"{name or local_name} ({short_address(address)})"
|