mirror of https://github.com/labapart/gattlib
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
from gattlib import *
|
|
from .device import Device
|
|
from .exception import handle_return
|
|
|
|
GATTLIB_DISCOVER_FILTER_USE_UUID = (1 << 0)
|
|
GATTLIB_DISCOVER_FILTER_USE_RSSI = (1 << 1)
|
|
|
|
|
|
class Adapter:
|
|
|
|
def __init__(self, name=c_char_p(None)):
|
|
self._name = name
|
|
self._adapter = c_void_p(None)
|
|
|
|
@property
|
|
def name(self):
|
|
return self._name
|
|
|
|
@staticmethod
|
|
def list():
|
|
# TODO: Add support
|
|
return []
|
|
|
|
def open(self):
|
|
return gattlib_adapter_open(self._name, byref(self._adapter))
|
|
|
|
def close(self):
|
|
return gattlib.gattlib_adapter_close(self._adapter)
|
|
|
|
def on_discovered_device(self, adapter, addr, name, user_data):
|
|
device = Device(self, addr, name)
|
|
self.on_discovered_device_callback(device, user_data)
|
|
|
|
def scan_enable(self, on_discovered_device_callback, timeout, uuids=None, rssi_threshold=None, user_data=None):
|
|
assert on_discovered_device_callback != None
|
|
self.on_discovered_device_callback = on_discovered_device_callback
|
|
|
|
enabled_filters = 0
|
|
uuid_list = None
|
|
rssi = 0
|
|
|
|
if uuids is not None:
|
|
enabled_filters |= GATTLIB_DISCOVER_FILTER_USE_UUID
|
|
|
|
# We add 1 to make sure the array finishes with a NULL pointer
|
|
uuid_list = (POINTER(GattlibUuid) * (len(uuids) + 1))()
|
|
index = 0
|
|
for uuid in uuids:
|
|
gattlib_uuid = GattlibUuid()
|
|
|
|
uuid_ascii = uuid.encode("utf-8")
|
|
ret = gattlib.gattlib_string_to_uuid(uuid_ascii, len(uuid_ascii), byref(gattlib_uuid))
|
|
handle_return(ret)
|
|
|
|
uuid_list[index] = cast(byref(gattlib_uuid), POINTER(GattlibUuid))
|
|
index += 1
|
|
|
|
if rssi_threshold is not None:
|
|
enabled_filters |= GATTLIB_DISCOVER_FILTER_USE_RSSI
|
|
rssi = int(rssi_threshold)
|
|
|
|
ret = gattlib_adapter_scan_enable_with_filter(self._adapter,
|
|
uuid_list, rssi, enabled_filters,
|
|
gattlib_discovered_device_type(self.on_discovered_device),
|
|
timeout, user_data)
|
|
handle_return(ret)
|
|
|
|
def scan_disable(self):
|
|
ret = gattlib.gattlib_adapter_scan_disable(self._adapter)
|
|
handle_return(ret)
|
|
|
|
def get_rssi_from_mac(self, mac_address):
|
|
if isinstance(mac_address, str):
|
|
mac_address = mac_address.encode("utf-8")
|
|
|
|
rssi = c_int16(0)
|
|
gattlib_get_rssi_from_mac(self._adapter, mac_address, byref(rssi))
|
|
return rssi.value
|