23 lines
781 B
Python
23 lines
781 B
Python
"""bluetooth usage utility to handle multiple instances."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import bleak
|
|
|
|
from .models import HaBleakClientWrapper, HaBleakScannerWrapper
|
|
|
|
ORIGINAL_BLEAK_SCANNER = bleak.BleakScanner
|
|
ORIGINAL_BLEAK_CLIENT = bleak.BleakClient
|
|
|
|
|
|
def install_multiple_bleak_catcher() -> None:
|
|
"""Wrap the bleak classes to return the shared instance if multiple instances are detected."""
|
|
bleak.BleakScanner = HaBleakScannerWrapper # type: ignore[misc, assignment]
|
|
bleak.BleakClient = HaBleakClientWrapper # type: ignore[misc]
|
|
|
|
|
|
def uninstall_multiple_bleak_catcher() -> None:
|
|
"""Unwrap the bleak classes."""
|
|
bleak.BleakScanner = ORIGINAL_BLEAK_SCANNER # type: ignore[misc]
|
|
bleak.BleakClient = ORIGINAL_BLEAK_CLIENT # type: ignore[misc]
|