mirror of https://github.com/labapart/gattlib
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
#
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
#
|
|
# Copyright (c) 2016-2024, Olivier Martin <olivier@labapart.org>
|
|
#
|
|
|
|
"""Module to manipulate Gattlib UUID in Python environment."""
|
|
|
|
import re
|
|
from uuid import UUID
|
|
|
|
from gattlib import * #pylint: disable=wildcard-import,unused-wildcard-import
|
|
|
|
SDP_UUID16 = 0x19
|
|
SDP_UUID32 = 0x1A
|
|
SDP_UUID128 = 0x1C
|
|
|
|
GATT_STANDARD_UUID_FORMAT = re.compile(r"(\S+)-0000-1000-8000-00805f9b34fb", flags=re.IGNORECASE)
|
|
|
|
|
|
def gattlib_uuid_to_uuid(gattlib_uuid) -> UUID:
|
|
"""Convert Gattlib UUID to Python UUID"""
|
|
if gattlib_uuid.type == SDP_UUID16:
|
|
return UUID(fields=(gattlib_uuid.value.uuid16, 0x0000, 0x1000, 0x80, 0x00, 0x00805f9b34fb))
|
|
elif gattlib_uuid.type == SDP_UUID32:
|
|
return UUID(fields=(gattlib_uuid.value.uuid32, 0x0000, 0x1000, 0x80, 0x00, 0x00805f9b34fb))
|
|
elif gattlib_uuid.type == SDP_UUID128:
|
|
data = bytes(gattlib_uuid.value.uuid128.data)
|
|
return UUID(bytes=data)
|
|
else:
|
|
return ValueError(f"Gattlib UUID not recognized (type:0x{gattlib_uuid.type:02x})")
|
|
|
|
|
|
def gattlib_uuid_to_int(gattlib_uuid) -> int:
|
|
"""Convert Gattlib UUID to integer."""
|
|
if gattlib_uuid.type == SDP_UUID16:
|
|
return gattlib_uuid.value.uuid16
|
|
elif gattlib_uuid.type == SDP_UUID32:
|
|
return gattlib_uuid.value.uuid32
|
|
elif gattlib_uuid.type == SDP_UUID128:
|
|
data = bytes(gattlib_uuid.value.uuid128.data)
|
|
return int.from_bytes(data, byteorder='big')
|
|
else:
|
|
return ValueError(f"Gattlib UUID not recognized (type:0x{gattlib_uuid.type:02x})")
|
|
|
|
|
|
def gattlib_uuid_str_to_int(uuid_str: str) -> int:
|
|
"""Convert uuid string to integer"""
|
|
# Check if the string could already encode a UUID16 or UUID32
|
|
if len(uuid_str) <= 8:
|
|
return int(uuid_str, 16)
|
|
|
|
# Check if it is a standard UUID or not
|
|
match = GATT_STANDARD_UUID_FORMAT.search(uuid_str)
|
|
if match:
|
|
return int(match.group(1), 16)
|
|
else:
|
|
try:
|
|
return UUID(uuid_str).int
|
|
except ValueError:
|
|
logger.error("Could not convert %s to a UUID", uuid_str)
|
|
raise
|