coordinator.py

Go to the documentation of this file.

"""Integration to integrate Keymitt BLE devices with Home Assistant."""
 
from __future__ import annotations
 
import logging
from typing import TYPE_CHECKING, Any
 
from microbot import MicroBotApiClient, parse_advertisement_data
 
from homeassistant.components import bluetooth
from homeassistant.components.bluetooth.passive_update_coordinator import (
    PassiveBluetoothDataUpdateCoordinator,
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
 
if TYPE_CHECKING:
    from bleak.backends.device import BLEDevice
 
_LOGGER: logging.Logger = logging.getLogger(__package__)
PLATFORMS: list[str] = [Platform.SWITCH]
 
 
class MicroBotDataUpdateCoordinator(PassiveBluetoothDataUpdateCoordinator):
    """Class to manage fetching data from the MicroBot."""
 
    def __init__(
        self,
        hass: HomeAssistant,
        client: MicroBotApiClient,
        ble_device: BLEDevice,
    ) -> None:
        """Initialize."""
        self.api: MicroBotApiClient = client
        self.datadata: dict[str, Any] = {}
        self.ble_deviceble_device = ble_device
        super().__init__(
            hass,
            _LOGGER,
            ble_device.address,
            bluetooth.BluetoothScanningMode.ACTIVE,
        )
 
    @callback
    def _async_handle_bluetooth_event(
        self,
        service_info: bluetooth.BluetoothServiceInfoBleak,
        change: bluetooth.BluetoothChange,
    ) -> None:
        """Handle a Bluetooth event."""
        if adv := parse_advertisement_data(
            service_info.device, service_info.advertisement
        ):
            self.datadata = adv.data
            _LOGGER.debug("%s: MicroBot data: %s", self.ble_deviceble_device.address, self.datadata)
            self.api.update_from_advertisement(adv)
        super()._async_handle_bluetooth_event(service_info, change)