File: parser.py

package info (click to toggle)
python-aioshelly 11.2.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 396 kB
  • sloc: python: 2,583; makefile: 8; sh: 3
file content (63 lines) | stat: -rw-r--r-- 1,815 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
"""Shelly Gen2 BLE support."""

from __future__ import annotations

import logging
from binascii import a2b_base64
from typing import Any

from bluetooth_data_tools import parse_advertisement_data_tuple

BLEGAPAdvertisementTupleType = tuple[
    str | None, list[str], dict[str, bytes], dict[int, bytes], int | None
]

LOGGER = logging.getLogger(__name__)


def parse_ble_scan_result_event(
    data: list[Any],
) -> list[tuple[str, int, BLEGAPAdvertisementTupleType]]:
    """Parse BLE scan result event."""
    version: int = data[0]
    if version == 1:
        return _parse_v1(data)
    if version == 2:  # noqa: PLR2004
        return _parse_v2(data[1])
    raise ValueError(f"Unsupported BLE scan result version: {version}")


def _parse_v1(adv: list[Any]) -> list[tuple[str, int, BLEGAPAdvertisementTupleType]]:
    """Convert v1 format to a list of ble tuples."""
    _, address, rssi, advertisement_data_b64, scan_response_b64 = adv
    return [
        (
            address.upper(),
            rssi,
            parse_advertisement_data_tuple(
                (
                    a2b_base64(advertisement_data_b64.encode("ascii")),
                    a2b_base64(scan_response_b64.encode("ascii")),
                )
            ),
        )
    ]


def _parse_v2(
    advs: list[list[Any]],
) -> list[tuple[str, int, BLEGAPAdvertisementTupleType]]:
    """Convert v2 format to a list of ble tuples."""
    return [
        (
            address.upper(),
            rssi,
            parse_advertisement_data_tuple(
                (
                    a2b_base64(advertisement_data_b64.encode("ascii")),
                    a2b_base64(scan_response_b64.encode("ascii")),
                )
            ),
        )
        for address, rssi, advertisement_data_b64, scan_response_b64 in advs
    ]