mirror of
https://github.com/undera/pylgbst.git
synced 2020-11-18 19:37:26 -08:00
16b1612cc6
* Cosmetics * Harmonograph demo * Cleanup * Original * Original * Cosmetics * Original file * Fixes * cosmetics * separate classes * Cosmetics * Cosmetics * fix tests * Remove plotter tests * Add bluegiga * Rename it * Progress * Fix tests * Cosmetics * Found a way for pygatt! * Playing with gatt * Fix hung subscribe * rename class * add test * skeleton for autodetect * safer order * Fix tests * Fix test * Add dbus install * another try * 2 * 3 * 34 * 6 * 7 * Isolate some tests * 8 * back to roots * Try more * 9 * Help * rep * site-packs * Fix? * Py3 come on * dbus * busss * dev null! * Fix test * Cleanup * Fix tests * Fix after review * add package * FIx package paths * Cosmetics * Update * More doc
121 lines
4.0 KiB
Python
121 lines
4.0 KiB
Python
import logging
|
|
import re
|
|
import threading
|
|
from time import sleep
|
|
|
|
import gatt
|
|
|
|
from pylgbst.comms import Connection, LEGO_MOVE_HUB
|
|
from pylgbst.constants import MOVE_HUB_HW_UUID_SERV, MOVE_HUB_HW_UUID_CHAR, MOVE_HUB_HARDWARE_HANDLE
|
|
from pylgbst.utilities import str2hex
|
|
|
|
log = logging.getLogger('comms-gatt')
|
|
|
|
|
|
class CustomDevice(gatt.Device, object):
|
|
def __init__(self, mac_address, manager):
|
|
gatt.Device.__init__(self, mac_address=mac_address, manager=manager)
|
|
self._notify_callback = lambda hnd, val: None
|
|
self._handle = None
|
|
|
|
def connect(self):
|
|
gatt.Device.connect(self)
|
|
log.info("Waiting for device connection...")
|
|
while self._handle is None:
|
|
log.debug("Sleeping...")
|
|
sleep(1)
|
|
|
|
if isinstance(self._handle, BaseException):
|
|
exc = self._handle
|
|
self._handle = None
|
|
raise exc
|
|
|
|
def write(self, data):
|
|
log.debug("Writing to handle: %s", str2hex(data))
|
|
return self._handle.write_value(data)
|
|
|
|
def enable_notifications(self):
|
|
log.debug('Enable Notifications...')
|
|
self._handle.enable_notifications()
|
|
|
|
def set_notific_handler(self, func_hnd):
|
|
self._notify_callback = func_hnd
|
|
|
|
def services_resolved(self):
|
|
log.debug('Getting MoveHub services and characteristics...')
|
|
gatt.Device.services_resolved(self)
|
|
log.debug("[%s] Resolved services", self.mac_address)
|
|
for service in self.services:
|
|
log.debug("[%s] Service [%s]", self.mac_address, service.uuid)
|
|
for characteristic in service.characteristics:
|
|
log.debug("[%s] Characteristic [%s]", self.mac_address, characteristic.uuid)
|
|
if service.uuid == MOVE_HUB_HW_UUID_SERV and characteristic.uuid == MOVE_HUB_HW_UUID_CHAR:
|
|
log.debug('MoveHub characteristic found')
|
|
self._handle = characteristic
|
|
|
|
if self._handle is None:
|
|
self.manager.stop()
|
|
self._handle = RuntimeError("Failed to obtain MoveHub handle")
|
|
|
|
def characteristic_value_updated(self, characteristic, value):
|
|
value = self._fix_weird_bug(value)
|
|
log.debug('Notification in GattDevice: %s', str2hex(value))
|
|
self._notify_callback(MOVE_HUB_HARDWARE_HANDLE, value)
|
|
|
|
def _fix_weird_bug(self, value):
|
|
if isinstance(value, str) and "dbus.Array" in value: # weird bug from gatt on my Ubuntu 16.04!
|
|
log.debug("Fixing broken notify string: %s", value)
|
|
return ''.join([chr(int(x.group(1))) for x in re.finditer(r"dbus.Byte\((\d+)\)", value)])
|
|
|
|
return value
|
|
|
|
|
|
class GattConnection(Connection):
|
|
"""
|
|
:type _device: CustomDevice
|
|
"""
|
|
|
|
def __init__(self, bt_iface_name='hci0'):
|
|
super(GattConnection, self).__init__()
|
|
self._device = None
|
|
self._iface = bt_iface_name
|
|
|
|
def connect(self, hub_mac=None):
|
|
dev_manager = gatt.DeviceManager(adapter_name=self._iface)
|
|
dman_thread = threading.Thread(target=dev_manager.run)
|
|
dman_thread.setDaemon(True)
|
|
log.debug('Starting DeviceManager...')
|
|
dman_thread.start()
|
|
dev_manager.start_discovery()
|
|
|
|
while not self._device:
|
|
log.info("Discovering devices...")
|
|
devices = dev_manager.devices()
|
|
log.debug("Devices: %s", devices)
|
|
|
|
for dev in devices:
|
|
address = dev.mac_address
|
|
name = dev.alias()
|
|
if name == LEGO_MOVE_HUB or hub_mac == address:
|
|
logging.info("Found %s at %s", name, address)
|
|
self._device = CustomDevice(address, dev_manager)
|
|
break
|
|
|
|
if not self._device:
|
|
sleep(1)
|
|
|
|
self._device.connect()
|
|
return self
|
|
|
|
def disconnect(self):
|
|
self._device.disconnect()
|
|
|
|
def write(self, handle, data):
|
|
self._device.write(data)
|
|
|
|
def set_notify_handler(self, handler):
|
|
self._device.set_notific_handler(handler)
|
|
|
|
def enable_notifications(self):
|
|
self._device.enable_notifications()
|