mirror of
https://github.com/kawasaki/pyscrlink.git
synced 2025-09-05 17:20:20 +02:00
Compare commits
34 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
ed4df282b1 | ||
|
00154f2b51 | ||
|
dc2ee5a22f | ||
|
caa344ecbb | ||
|
0f9ccd3b63 | ||
|
d1f7f58ca2 | ||
|
7050016ee6 | ||
|
d845d69bb8 | ||
|
751190935a | ||
|
3956c81869 | ||
|
caaec40935 | ||
|
845b37707e | ||
|
a5d19fa2ba | ||
|
865a4d6f09 | ||
|
bf154e0a14 | ||
|
7cc9ccac2e | ||
|
4558ea43df | ||
|
641b84a86e | ||
|
e6d63a5c97 | ||
|
6f7da4f720 | ||
|
32da72492f | ||
|
1af4baa7d6 | ||
|
39587b497b | ||
|
f506bbcaab | ||
|
9265086b12 | ||
|
58a60c94db | ||
|
ea1109cee2 | ||
|
f4af270fbc | ||
|
1b92af0f0a | ||
|
c206f7a5c5 | ||
|
e34ec61f3b | ||
|
e552bd21bd | ||
|
dc46869760 | ||
|
461377f7ea |
6
.gitignore
vendored
6
.gitignore
vendored
@@ -1 +1,5 @@
|
||||
/__pycache__
|
||||
__pycache__
|
||||
build
|
||||
dist
|
||||
pyscrlink/__pycache__
|
||||
pyscrlink.egg-info
|
||||
|
145
README.md
145
README.md
@@ -9,11 +9,16 @@ micro:bit.
|
||||
|
||||
Pyscrlink allows you to connect Scratch and bluetooth devices with the Linux
|
||||
OSes. It uses the Linux Bluetooth protocol stack [Bluez](http://www.bluez.org/)
|
||||
and its python interfaces [pybluez](https://github.com/pybluez/pybluez) to
|
||||
handle Bluetooth, and [bluepy](https://github.com/IanHarvey/bluepy) to handle
|
||||
Bluetooth Low Energy (BLE) connections. It has been reported that pyscrlink
|
||||
connects Scratch 3.0 with micro:bit, LEGO Mindstorms EV3, LEGO WeDo, LEGO
|
||||
Boost and Intelino Smart Train.
|
||||
and [bluepy](https://github.com/IanHarvey/bluepy) to handle Bluetooth Low Energy
|
||||
(BLE) connections. It has been reported that pyscrlink connects Scratch 3.0 with
|
||||
micro:bit, LEGO WeDo, LEGO Boost and toio.
|
||||
|
||||
Until version v0.2.5, pyscrlink supported Bluetooth Classic protocol using
|
||||
[pybluez](https://github.com/pybluez/pybluez). Unfortunately, pybluez is not
|
||||
well maintained and caused technical troubles. Then Bluetooth Classic protocol
|
||||
support is dropped from pyscrlink. This means that LEGO Mindstorm EV3 can not
|
||||
be connected with pyscrlink. Bluetooth Classic support is the improvement
|
||||
opportunity of pyscrlink.
|
||||
|
||||
To use websockets, pyscrlink requires python version 3.6 or later. If your
|
||||
system has python older than version 3.6, install newer version. If your
|
||||
@@ -39,25 +44,24 @@ Devices:
|
||||
|
||||
Linux distros:
|
||||
* Arch Linux
|
||||
* elementary OS 5.1 Hera
|
||||
|
||||
Browsers:
|
||||
* FireFox
|
||||
* Firefox
|
||||
* Chromium
|
||||
|
||||
It was reported that pyscrlink (former bluepy-scratch-link) working with
|
||||
following devices and Linux distros.
|
||||
|
||||
Devices:
|
||||
* LEGO Mindstorm EV3 by @chrisglencross
|
||||
* LEGO WeDo by @zhaowe, @KingBBQ
|
||||
* LEGO Boost by @laurentchar, @miguev, @jacquesdt
|
||||
* Intelino Smart Train by @ErrorJan
|
||||
* LEGO Boost and compatible devices by @laurentchar, @miguev, @jacquesdt, @n3storm
|
||||
* toio by @shimodash
|
||||
|
||||
Linux distros:
|
||||
* Raspbian by @chrisglencross
|
||||
* Ubuntu 16.04 @jacquesdt
|
||||
* Ubuntu Studio 20.04 @miguev
|
||||
* Debian 11 @n3storm
|
||||
|
||||
Installation
|
||||
------------
|
||||
@@ -73,7 +77,7 @@ Installation
|
||||
|
||||
```sh
|
||||
Ubuntu
|
||||
$ sudo apt install bluez libbluetooth-dev libnss3-tools libcap2-bin
|
||||
$ sudo apt install bluez libbluetooth-dev libnss3-tools libcap2-bin libglib2.0-dev
|
||||
Arch
|
||||
$ sudo pacman -S bluez bluez-utils nss libcap
|
||||
```
|
||||
@@ -86,7 +90,7 @@ Installation
|
||||
$ pip3 install pyscrlink
|
||||
```
|
||||
|
||||
4. For Bluetooth Low Energy (BLE) devices, set bluepy-helper capability.
|
||||
4. Set bluepy-helper capability.
|
||||
|
||||
```
|
||||
$ bluepy_helper_cap
|
||||
@@ -98,72 +102,25 @@ Installation
|
||||
|
||||
5. For micro:bit, install Scratch-link hex on your device.
|
||||
|
||||
* Download and unzip the [micro:bit Scratch Hex file](https://downloads.scratch.mit.edu/microbit/scratch-microbit-1.1.0.hex.zip).
|
||||
* Flash the micro:bit over USB with the Scratch .Hex File, you will see the
|
||||
* Download and unzip the [micro:bit Scratch Hex file](https://downloads.scratch.mit.edu/microbit/scratch-microbit.hex.zip).
|
||||
* Flash the micro:bit over USB with the Scratch Hex File, you will see the
|
||||
five character name of the micro:bit scroll across the screen such as
|
||||
'zo9ev'.
|
||||
|
||||
Usage
|
||||
-----
|
||||
1. For LEGO Mindstorms EV3, pair your Linux PC to the EV3 brick.
|
||||
|
||||
First, turn on the EV3 and ensure Bluetooth is enabled.
|
||||
|
||||
Then, pair using your Linux desktop's the Bluetooth settings.
|
||||
|
||||
If using Gnome:
|
||||
* Settings -> Bluetooth
|
||||
* Click on the EV3 device name
|
||||
* Accept the connection on EV3 brick
|
||||
* Enter a matching PIN on EV3 brick and Linux PC. '1234' is the value Scratch suggests.
|
||||
* Confirm EV3 status is "Disconnected" in Bluetooth settings
|
||||
|
||||
With a Raspberry Pi default Raspbian desktop, click the Bluetooth logo in the top right of the screen and
|
||||
Add Device. Then follow the Gnome instructions. You will be warned that the Raspberry Pi
|
||||
does not know how to talk to this device; that is not a problem.
|
||||
|
||||
Alternatively you can perform pairing from the command-line:
|
||||
```shell script
|
||||
$ bluetoothctl
|
||||
|
||||
[bluetooth]# power on
|
||||
Changing power on succeeded
|
||||
|
||||
[bluetooth]# pairable on
|
||||
Changing pairable on succeeded
|
||||
|
||||
[bluetooth]# agent KeyboardOnly
|
||||
Agent registered
|
||||
|
||||
[bluetooth]# devices
|
||||
...
|
||||
Device 00:16:53:53:D3:19 EV3
|
||||
...
|
||||
|
||||
[bluetooth]# pair 00:16:53:53:D3:19
|
||||
Attempting to pair with 00:16:53:53:D3:19
|
||||
|
||||
# Confirm pairing on the EV3 display, set PIN to 1234
|
||||
|
||||
Request PIN code
|
||||
[agent] Enter PIN code: 1234
|
||||
[CHG] Device 00:16:53:53:D3:19 Connected: yes
|
||||
[CHG] Device 00:16:53:53:D3:19 Paired: yes
|
||||
Pairing successful
|
||||
|
||||
[bluetooth]# quit
|
||||
```
|
||||
|
||||
2. Start scratch-link python script.
|
||||
1. Start scratch-link python script.
|
||||
```sh
|
||||
$ scratch_link
|
||||
```
|
||||
If your device is toio, add "-s 1" option to the scratch_link command. It
|
||||
allows the toio Do Visual Programming to connect to toio automatically.
|
||||
|
||||
3. Connect scratch to micro:bit or LEGO Mindstorms:
|
||||
2. Connect scratch to the target device such as micro:bit:
|
||||
* Open FireFox or Chrome. (Make sure to run as the same user for scratch-link python script.)
|
||||
* Access [Scratch 3.0](https://scratch.mit.edu/) and create your project.
|
||||
* Select the "Add Extension" button.
|
||||
* Select the extension for your device (e.g., micro:bit or Lego Mindstorms EV3 extension) and follow the prompts to connect.
|
||||
* Select the extension for your device (e.g., micro:bit) and follow the prompts to connect.
|
||||
* Build your project with the extension blocks.
|
||||
|
||||
In Case You Fail to Connect
|
||||
@@ -184,9 +141,27 @@ In Case You Fail to Connect
|
||||
* If the service is not working, refer guide of your distro to set it up.
|
||||
* If the service is working, also check that /etc/bluetooth/main.conf sets AutoEnable=true.
|
||||
|
||||
3. If scratch_link.py says "failed to connect to BT device: [Errno 13] Permission denied",
|
||||
3. If device scan still fails, use -r option to retry device scan.
|
||||
The command line below does device scan twice. Each scan takes 10 seconds.
|
||||
```
|
||||
$ scratch_link -r 2
|
||||
```
|
||||
It would be good to use -s option together to reduce each scan duration.
|
||||
The command line below does 3 seconds device scan twice.
|
||||
|
||||
```
|
||||
$ scratch_link -r 2 -s 3
|
||||
```
|
||||
|
||||
4. If scratch_link.py says "failed to connect to BT device: [Errno 13] Permission denied",
|
||||
make sure to pair the bluetooth device to your PC before connecting to Scratch.
|
||||
|
||||
5. To connect to multiple devices at the same time, make all the target devices
|
||||
ready for scan at the first device scan. This is important for toio. The toio
|
||||
allows a single project to connect to two toio devices.
|
||||
* When the second device was prepared after the first device was connected, device scan can not find the second device.
|
||||
* To scan and find the second device, disconnect connections for the first device beforehand.
|
||||
|
||||
Issus Reporting
|
||||
---------------
|
||||
|
||||
@@ -195,6 +170,42 @@ Please file issues to [GitHub issue tracker](https://github.com/kawasaki/pyscrli
|
||||
Releases
|
||||
--------
|
||||
|
||||
Release 0.2.8
|
||||
|
||||
* Supported Microbit More v2
|
||||
|
||||
Release 0.2.7
|
||||
|
||||
* Supported Snap Firefox and Chromium
|
||||
* Added -r option to retry BLE scan
|
||||
|
||||
Release 0.2.6
|
||||
|
||||
* Removed Bluetooth Classic and LEGO Mindstorm EV3 support
|
||||
|
||||
Release 0.2.5
|
||||
|
||||
* Fixed handling of multiple UUIDs for LEGO Boost
|
||||
|
||||
Release 0.2.4
|
||||
|
||||
* Added -s option to specify BLE scan duration
|
||||
* Improved README.md
|
||||
|
||||
Release 0.2.3
|
||||
|
||||
* Fixed eternal loop caused by hostname resolve failure
|
||||
|
||||
Release 0.2.2
|
||||
|
||||
* Supported multiple device connections for toio
|
||||
* Improved session closure handling
|
||||
|
||||
Release 0.2.1
|
||||
|
||||
* Added libglib to required package list in README.md
|
||||
* Improved setcap and getcap tool finding
|
||||
|
||||
Release 0.2.0
|
||||
|
||||
* Latency issue fix for BLE devices' write characteristics
|
||||
|
@@ -22,10 +22,26 @@ logger.propagate = False
|
||||
# Check dependent tools
|
||||
DEPENDENT_TOOLS = {
|
||||
"setcap": "libcap2-bin (Ubuntu) or libcap (Arch)",
|
||||
"getcap": "libcap2-bin (Ubuntu) or libcap (Arch)",
|
||||
}
|
||||
|
||||
tools = {}
|
||||
|
||||
for cmd in DEPENDENT_TOOLS:
|
||||
if not shutil.which(cmd):
|
||||
# find the tools in PATH
|
||||
path = shutil.which(cmd)
|
||||
if path:
|
||||
tools[cmd] = path
|
||||
logger.debug(f"{cmd} found: {path}")
|
||||
continue
|
||||
# find the tools out of PATH but in major directories
|
||||
for d in ["/usr/bin", "/bin", "/usr/sbin", "/sbin"]:
|
||||
path = d + '/' + cmd
|
||||
if os.path.isfile(path) and os.access(path, os.X_OK):
|
||||
tools[cmd] = path
|
||||
logger.debug(f"{cmd} found: {path}")
|
||||
break
|
||||
if not cmd in tools:
|
||||
print(f"'{cmd}' not found. Install package {DEPENDENT_TOOLS[cmd]}.")
|
||||
sys.exit(1)
|
||||
|
||||
@@ -42,7 +58,7 @@ def helper_path():
|
||||
|
||||
def is_set():
|
||||
path = helper_path()
|
||||
p = subprocess.run(["getcap", path], stdout=subprocess.PIPE)
|
||||
p = subprocess.run([tools["getcap"], path], stdout=subprocess.PIPE)
|
||||
if p.returncode != 0:
|
||||
logger.error(f"Failed to get capability of {path}")
|
||||
return False
|
||||
@@ -53,8 +69,8 @@ def setcap():
|
||||
path = helper_path()
|
||||
if is_set():
|
||||
return True
|
||||
p = subprocess.run(["sudo", "setcap", "cap_net_raw,cap_net_admin+eip",
|
||||
path])
|
||||
p = subprocess.run(["sudo", tools["setcap"],
|
||||
"cap_net_raw,cap_net_admin+eip", path])
|
||||
if p.returncode !=0:
|
||||
logger.error(f"Failed to set capability to {path}")
|
||||
return False
|
||||
|
@@ -152,36 +152,37 @@ def prep_nss_cert(dir, cert, nickname):
|
||||
remove_cert(dir, nickname)
|
||||
add_cert(dir, cert, nickname)
|
||||
|
||||
def prep_cert():
|
||||
# Generate certification and key
|
||||
gen_cert(cert_file_path, key_file_path)
|
||||
|
||||
# Add certificate to FireFox
|
||||
def prep_cert_for_app(cert, app, search_path):
|
||||
"""
|
||||
Find a NSS DB in the search_path for the app and prepare the cert in the DB.
|
||||
"""
|
||||
nssdb = None
|
||||
firefox_nss_path = os.path.join(homedir, ".mozilla/firefox/")
|
||||
for root, dirs, files in os.walk(firefox_nss_path):
|
||||
for root, dirs, files in os.walk(os.path.join(homedir, search_path)):
|
||||
for name in files:
|
||||
if not re.match("key.*\.db", name):
|
||||
continue
|
||||
nssdb = root
|
||||
if prep_nss_cert(nssdb, cert_file_path, SCRATCH_CERT_NICKNAME):
|
||||
logger.error(f"Failed to add certificate to FireFox NSS DB: {nssdb}")
|
||||
if prep_nss_cert(nssdb, cert, SCRATCH_CERT_NICKNAME):
|
||||
logger.error(f"Failed to add certificate to {app}: {nssdb}")
|
||||
sys.exit(3)
|
||||
else:
|
||||
logger.info(f"Certificate is ready in FireFox NSS DB: {nssdb}")
|
||||
logger.info(f"Certificate is ready in {app} NSS DB: {nssdb}")
|
||||
if not nssdb:
|
||||
logger.info("FireFox NSS DB not found. Do not add certificate.")
|
||||
logger.debug(f"NSS DB for {app} not found. Do not add certificate.")
|
||||
|
||||
# Add certificate to Chrome
|
||||
nssdb = os.path.join(homedir, ".pki/nssdb")
|
||||
if os.path.isdir(nssdb):
|
||||
if prep_nss_cert(nssdb, cert_file_path, SCRATCH_CERT_NICKNAME):
|
||||
logger.error(f"Failed to add certificate to Chrome")
|
||||
sys.exit(4)
|
||||
else:
|
||||
logger.info("Certificate is ready for Chrome")
|
||||
else:
|
||||
logger.info("Chrome NSS DB not found. Do not add certificate.")
|
||||
|
||||
def prep_cert():
|
||||
# Generate certification and key
|
||||
gen_cert(cert_file_path, key_file_path)
|
||||
|
||||
nss_dbs = {
|
||||
"FireFox": ".mozilla/firefox/",
|
||||
"FireFox(Snap)": "snap/firefox/common/.mozilla/firefox/",
|
||||
"Chrome": ".pki",
|
||||
"Chromium(Snap)": "snap/chromium",
|
||||
}
|
||||
|
||||
[ prep_cert_for_app(cert_file_path, k, nss_dbs[k]) for k in nss_dbs ]
|
||||
|
||||
if __name__ == "__main__":
|
||||
prep_cert()
|
||||
|
@@ -8,18 +8,17 @@ import asyncio
|
||||
import pathlib
|
||||
import ssl
|
||||
import websockets
|
||||
import socket
|
||||
import json
|
||||
import base64
|
||||
import logging
|
||||
import sys
|
||||
import signal
|
||||
import traceback
|
||||
|
||||
# for Bluetooth (e.g. Lego EV3)
|
||||
import bluetooth
|
||||
import argparse
|
||||
|
||||
# for BLESession (e.g. BBC micro:bit)
|
||||
from bluepy.btle import Scanner, UUID, Peripheral, DefaultDelegate
|
||||
from bluepy.btle import Scanner, UUID, Peripheral, DefaultDelegate, ScanEntry
|
||||
from bluepy.btle import BTLEDisconnectError, BTLEManagementError
|
||||
from pyscrlink import bluepy_helper_cap
|
||||
|
||||
@@ -42,6 +41,9 @@ logger.setLevel(logLevel)
|
||||
logger.addHandler(handler)
|
||||
logger.propagate = False
|
||||
|
||||
HOSTNAME="device-manager.scratch.mit.edu"
|
||||
scan_seconds=10.0
|
||||
|
||||
class Session():
|
||||
"""Base class for BTSession and BLESession"""
|
||||
def __init__(self, websocket, loop):
|
||||
@@ -133,195 +135,6 @@ class Session():
|
||||
class BTSession(Session):
|
||||
"""Manage a session for Bluetooth device"""
|
||||
|
||||
INITIAL = 1
|
||||
DISCOVERY = 2
|
||||
DISCOVERY_COMPLETE = 3
|
||||
CONNECTED = 4
|
||||
DONE = 5
|
||||
|
||||
# Split this into discovery thread and communication thread
|
||||
# discovery thread should auto-terminate
|
||||
|
||||
class BTThread(threading.Thread):
|
||||
"""
|
||||
Separated thread to control notifications to Scratch.
|
||||
It handles device discovery notification in DISCOVERY status
|
||||
and notifications from bluetooth devices in CONNECTED status.
|
||||
"""
|
||||
|
||||
class BTDiscoverer(bluetooth.DeviceDiscoverer):
|
||||
|
||||
def __init__(self, major_class, minor_class):
|
||||
super().__init__()
|
||||
self.major_class = major_class
|
||||
self.minor_class = minor_class
|
||||
self.found_devices = {}
|
||||
self.done = False
|
||||
|
||||
def pre_inquiry(self):
|
||||
self.done = False
|
||||
|
||||
def device_discovered(self, address, device_class, rssi, name):
|
||||
logger.debug(f"Found device {name} addr={address} class={device_class} rssi={rssi}")
|
||||
major_class = (device_class & 0x1F00) >> 8
|
||||
minor_class = (device_class & 0xFF) >> 2
|
||||
if major_class == self.major_class and minor_class == self.minor_class:
|
||||
self.found_devices[address] = (name, device_class, rssi)
|
||||
|
||||
def inquiry_complete(self):
|
||||
self.done = True
|
||||
|
||||
def __init__(self, session, major_device_class, minor_device_class):
|
||||
threading.Thread.__init__(self)
|
||||
self.session = session
|
||||
self.major_device_class = major_device_class
|
||||
self.minor_device_class = minor_device_class
|
||||
self.cancel_discovery = False
|
||||
self.ping_time = None
|
||||
|
||||
def discover(self):
|
||||
discoverer = self.BTDiscoverer(self.major_device_class, self.minor_device_class)
|
||||
discoverer.find_devices(lookup_names=True)
|
||||
while self.session.status == self.session.DISCOVERY and not discoverer.done and not self.cancel_discovery:
|
||||
readable = select.select([discoverer], [], [], 0.5)[0]
|
||||
if discoverer in readable:
|
||||
discoverer.process_event()
|
||||
for addr, (device_name, device_class, rssi) in discoverer.found_devices.items():
|
||||
logger.debug(f"notifying discovered {addr}: {device_name}")
|
||||
params = {"rssi": rssi, 'peripheralId': addr, 'name': device_name.decode("utf-8")}
|
||||
self.session.notify('didDiscoverPeripheral', params)
|
||||
discoverer.found_devices.clear()
|
||||
|
||||
if not discoverer.done:
|
||||
discoverer.cancel_inquiry()
|
||||
|
||||
def run(self):
|
||||
while self.session.status != self.session.DONE:
|
||||
|
||||
logger.debug("loop in BT thread")
|
||||
current_time = int(round(time.time()))
|
||||
|
||||
if self.session.status == self.session.DISCOVERY and not self.cancel_discovery:
|
||||
logger.debug("in discovery status:")
|
||||
try:
|
||||
self.discover()
|
||||
self.ping_time = current_time + 5
|
||||
finally:
|
||||
self.session.status = self.session.DISCOVERY_COMPLETE
|
||||
|
||||
elif self.session.status == self.session.CONNECTED:
|
||||
logger.debug("in connected status:")
|
||||
sock = self.session.sock
|
||||
try:
|
||||
ready = select.select([sock], [], [], 1)
|
||||
if ready[0]:
|
||||
header = sock.recv(2)
|
||||
[msg_len] = struct.unpack("<H", header)
|
||||
msg_data = sock.recv(msg_len)
|
||||
data = header + msg_data
|
||||
params = {'message': base64.standard_b64encode(data).decode('utf-8'), "encoding": "base64"}
|
||||
self.session.notify('didReceiveMessage', params)
|
||||
self.ping_time = current_time + 5
|
||||
|
||||
except Exception as e:
|
||||
logger.error(e)
|
||||
self.session.close()
|
||||
break
|
||||
|
||||
# To avoid repeated lock by this single thread,
|
||||
# yield CPU to other lock waiting threads.
|
||||
time.sleep(0)
|
||||
else:
|
||||
# Nothing to do:
|
||||
time.sleep(1)
|
||||
|
||||
# Terminate if we have lost websocket connection to Scratch (e.g. browser closed)
|
||||
if self.ping_time is None or self.ping_time <= current_time:
|
||||
try:
|
||||
self.session.notify('ping', {})
|
||||
self.ping_time = current_time + 5
|
||||
except Exception as e:
|
||||
logger.error(e)
|
||||
self.session.close()
|
||||
break
|
||||
|
||||
def __init__(self, websocket, loop):
|
||||
super().__init__(websocket, loop)
|
||||
self.status = self.INITIAL
|
||||
self.sock = None
|
||||
self.bt_thread = None
|
||||
|
||||
def close(self):
|
||||
self.status = self.DONE
|
||||
if self.sock:
|
||||
logger.info(f"disconnect to BT socket: {self.sock}")
|
||||
self.sock.close()
|
||||
self.sock = None
|
||||
|
||||
def __del__(self):
|
||||
self.close()
|
||||
|
||||
def handle_request(self, method, params):
|
||||
"""Handle requests from Scratch"""
|
||||
logger.debug("handle request to BT device")
|
||||
logger.debug(method)
|
||||
if len(params) > 0:
|
||||
logger.debug(params)
|
||||
|
||||
res = { "jsonrpc": "2.0" }
|
||||
|
||||
if self.status == self.INITIAL and method == 'discover':
|
||||
logger.debug("Starting async discovery")
|
||||
self.status = self.DISCOVERY
|
||||
self.bt_thread = self.BTThread(self, params["majorDeviceClass"], params["minorDeviceClass"])
|
||||
self.bt_thread.start()
|
||||
res["result"] = None
|
||||
|
||||
elif self.status in [self.DISCOVERY, self.DISCOVERY_COMPLETE] and method == 'connect':
|
||||
|
||||
# Cancel discovery
|
||||
while self.status == self.DISCOVERY:
|
||||
logger.debug("Cancelling discovery")
|
||||
self.bt_thread.cancel_discovery = True
|
||||
time.sleep(1)
|
||||
|
||||
addr = params['peripheralId']
|
||||
logger.debug(f"connecting to the BT device {addr}")
|
||||
try:
|
||||
self.sock = bluetooth.BluetoothSocket(bluetooth.RFCOMM)
|
||||
self.sock.connect((addr, 1))
|
||||
logger.info(f"connected to BT device: {addr}")
|
||||
except bluetooth.BluetoothError as e:
|
||||
logger.error(f"failed to connect to BT device: {e}", exc_info=e)
|
||||
self.status = self.DONE
|
||||
self.sock = None
|
||||
|
||||
if self.sock:
|
||||
res["result"] = None
|
||||
self.status = self.CONNECTED
|
||||
else:
|
||||
err_msg = f"BT connect failed: {addr}"
|
||||
res["error"] = { "message": err_msg }
|
||||
self.status = self.DONE
|
||||
|
||||
elif self.status == self.CONNECTED and method == 'send':
|
||||
logger.debug("handle send request")
|
||||
if params['encoding'] != 'base64':
|
||||
logger.error("encoding other than base 64 is not "
|
||||
"yet supported: ", params['encoding'])
|
||||
msg_bstr = params['message'].encode('ascii')
|
||||
data = base64.standard_b64decode(msg_bstr)
|
||||
self.sock.send(data)
|
||||
res['result'] = len(data)
|
||||
|
||||
logger.debug(res)
|
||||
return res
|
||||
|
||||
def end_request(self):
|
||||
logger.debug(f"end_request of BTSession {self}")
|
||||
return self.status == self.DONE
|
||||
|
||||
|
||||
class BLESession(Session):
|
||||
"""
|
||||
Manage a session for Bluetooth Low Energy device such as micro:bit
|
||||
@@ -343,6 +156,11 @@ class BLESession(Session):
|
||||
|
||||
MAX_SCANNER_IF = 3
|
||||
|
||||
found_devices = []
|
||||
nr_connected = 0
|
||||
scan_lock = threading.RLock()
|
||||
scan_started = False
|
||||
|
||||
class BLEThread(threading.Thread):
|
||||
"""
|
||||
Separated thread to control notifications to Scratch.
|
||||
@@ -358,7 +176,7 @@ class BLESession(Session):
|
||||
logger.debug("loop in BLE thread")
|
||||
if self.session.status == self.session.DISCOVERY:
|
||||
logger.debug("send out found devices")
|
||||
devices = self.session.found_devices
|
||||
devices = BLESession.found_devices
|
||||
for d in devices:
|
||||
params = { 'rssi': d.rssi }
|
||||
params['peripheralId'] = devices.index(d)
|
||||
@@ -418,7 +236,6 @@ class BLESession(Session):
|
||||
def __init__(self, websocket, loop):
|
||||
super().__init__(websocket, loop)
|
||||
self.status = self.INITIAL
|
||||
self.found_devices = []
|
||||
self.device = None
|
||||
self.deviceName = None
|
||||
self.perip = None
|
||||
@@ -426,6 +243,13 @@ class BLESession(Session):
|
||||
self.characteristics_cache = []
|
||||
|
||||
def close(self):
|
||||
if self.status == self.CONNECTED:
|
||||
BLESession.nr_connected -= 1
|
||||
logger.info(f"BLE session disconnected")
|
||||
logger.debug(f"BLE session connected={BLESession.nr_connected}")
|
||||
if BLESession.nr_connected == 0:
|
||||
logger.info("all BLE sessions disconnected")
|
||||
BLESession.scan_started = False
|
||||
self.status = self.DONE
|
||||
if self.perip:
|
||||
logger.info("disconnect from the BLE peripheral: "
|
||||
@@ -437,12 +261,14 @@ class BLESession(Session):
|
||||
def __del__(self):
|
||||
self.close()
|
||||
|
||||
def _get_dev_uuid(self, dev):
|
||||
def _get_dev_uuids(self, dev):
|
||||
for adtype in self.SERVICE_CLASS_UUID_ADTYPES:
|
||||
service_class_uuid = dev.getValueText(adtype)
|
||||
if service_class_uuid:
|
||||
logger.debug(self.SERVICE_CLASS_UUID_ADTYPES[adtype])
|
||||
return UUID(service_class_uuid)
|
||||
service_class_uuids = dev.getValue(adtype)
|
||||
if service_class_uuids:
|
||||
for u in service_class_uuids:
|
||||
a = self.SERVICE_CLASS_UUID_ADTYPES[adtype]
|
||||
logger.debug(f"service class uuid for {a}/{adtype}: {u}")
|
||||
return service_class_uuids
|
||||
return None
|
||||
|
||||
def matches(self, dev, filters):
|
||||
@@ -455,31 +281,68 @@ class BLESession(Session):
|
||||
for s in f['services']:
|
||||
logger.debug(f"service to check: {s}")
|
||||
given_uuid = s
|
||||
logger.debug(f"given: {given_uuid}")
|
||||
dev_uuid = self._get_dev_uuid(dev)
|
||||
if not dev_uuid:
|
||||
logger.debug(f"given UUID: {given_uuid} hash={UUID(given_uuid).__hash__()}")
|
||||
dev_uuids = self._get_dev_uuids(dev)
|
||||
if not dev_uuids:
|
||||
continue
|
||||
logger.debug(f"dev: {dev_uuid}")
|
||||
logger.debug(given_uuid == dev_uuid)
|
||||
if given_uuid == dev_uuid:
|
||||
logger.debug("match...")
|
||||
return True
|
||||
for u in dev_uuids:
|
||||
logger.debug(f"dev UUID: {u} hash={u.__hash__()}")
|
||||
logger.debug(given_uuid == u)
|
||||
if given_uuid == u:
|
||||
logger.debug("match...")
|
||||
return True
|
||||
if 'namePrefix' in f:
|
||||
# 0x08: Shortened Local Name
|
||||
deviceName = dev.getValueText(0x08)
|
||||
if not deviceName:
|
||||
continue
|
||||
logger.debug(f"Name of \"{deviceName}\" begins with: \"{f['namePrefix']}\"?")
|
||||
if(deviceName.startswith(f['namePrefix'])):
|
||||
logger.debug("Yes")
|
||||
return True
|
||||
logger.debug("No")
|
||||
logger.debug(f"given namePrefix: {f['namePrefix']}")
|
||||
deviceName = dev.getValueText(ScanEntry.SHORT_LOCAL_NAME)
|
||||
if deviceName:
|
||||
logger.debug(f"SHORT_LOCAL_NAME: {deviceName}")
|
||||
if deviceName.startswith(f['namePrefix']):
|
||||
logger.debug(f"match...")
|
||||
return True
|
||||
deviceName = dev.getValueText(ScanEntry.COMPLETE_LOCAL_NAME)
|
||||
if deviceName:
|
||||
logger.debug(f"COMPLETE_LOCAL_NAME: {deviceName}")
|
||||
if deviceName.startswith(f['namePrefix']):
|
||||
logger.debug(f"match...")
|
||||
return True
|
||||
if 'name' in f or 'manufactureData' in f:
|
||||
logger.error("name/manufactureData filters not implemented")
|
||||
# TODO: implement other filters defined:
|
||||
# ref: https://github.com/LLK/scratch-link/blob/develop/Documentation/BluetoothLE.md
|
||||
return False
|
||||
|
||||
def _scan_devices(self, params):
|
||||
global scan_seconds
|
||||
if BLESession.nr_connected > 0:
|
||||
return len(BLESession.found_devices) > 0
|
||||
found = False
|
||||
with BLESession.scan_lock:
|
||||
if not BLESession.scan_started:
|
||||
BLESession.scan_started = True
|
||||
BLESession.found_devices.clear()
|
||||
for i in range(self.MAX_SCANNER_IF):
|
||||
scanner = Scanner(iface=i)
|
||||
for j in range(scan_retry):
|
||||
try:
|
||||
logger.debug(f"start BLE scan: {scan_seconds} seconds")
|
||||
devices = scanner.scan(scan_seconds)
|
||||
for dev in devices:
|
||||
if self.matches(dev, params['filters']):
|
||||
BLESession.found_devices.append(dev)
|
||||
found = True
|
||||
logger.debug(f"BLE device found with iface #{i}");
|
||||
if found:
|
||||
break
|
||||
except BTLEDisconnectError as de:
|
||||
logger.debug(f"BLE iface #{i}: {de}");
|
||||
except BTLEManagementError as me:
|
||||
logger.debug(f"BLE iface #{i}: {me}");
|
||||
if found:
|
||||
break
|
||||
else:
|
||||
found = len(BLESession.found_devices) > 0
|
||||
return found
|
||||
|
||||
def _get_service(self, service_id):
|
||||
with self.lock:
|
||||
service = self.perip.getServiceByUUID(UUID(service_id))
|
||||
@@ -528,29 +391,21 @@ class BLESession(Session):
|
||||
if self.status == self.INITIAL and method == 'discover':
|
||||
if not bluepy_helper_cap.is_set():
|
||||
logger.error("Capability is not set to bluepy helper.")
|
||||
logger.error("Run bluepy_setcap.py with root privilege.")
|
||||
logger.error("Run bluepy_helper_cap(.py).")
|
||||
logger.error("e.g. $ bluepy_helper_cap")
|
||||
logger.error("e.g. $ sudo bluepy_helper_cap.py")
|
||||
sys.exit(1)
|
||||
found_ifaces = 0
|
||||
for i in range(self.MAX_SCANNER_IF):
|
||||
scanner = Scanner(iface=i)
|
||||
try:
|
||||
devices = scanner.scan(10.0)
|
||||
for dev in devices:
|
||||
if self.matches(dev, params['filters']):
|
||||
self.found_devices.append(dev)
|
||||
found_ifaces += 1
|
||||
logger.debug(f"BLE device found with iface #{i}");
|
||||
except BTLEManagementError as e:
|
||||
logger.debug(f"BLE iface #{i}: {e}");
|
||||
|
||||
if found_ifaces == 0:
|
||||
err_msg = "Can not scan BLE devices. Check BLE controller."
|
||||
found = self._scan_devices(params)
|
||||
if not found:
|
||||
if BLESession.nr_connected > 0:
|
||||
err_msg = "Can not scan BLE devices. Disconnect other sessions."
|
||||
elif len(BLESession.found_devices) == 0:
|
||||
err_msg = "Can not scan BLE devices. Check BLE controller."
|
||||
logger.error(err_msg);
|
||||
res["error"] = { "message": err_msg }
|
||||
self.status = self.DONE
|
||||
|
||||
if len(self.found_devices) == 0 and not err_msg:
|
||||
if len(BLESession.found_devices) == 0 and not err_msg:
|
||||
err_msg = (f"No BLE device found: {params['filters']}. "
|
||||
"Check BLE device.")
|
||||
res["error"] = { "message": err_msg }
|
||||
@@ -564,11 +419,12 @@ class BLESession(Session):
|
||||
|
||||
elif self.status == self.DISCOVERY and method == 'connect':
|
||||
logger.debug("connecting to the BLE device")
|
||||
self.device = self.found_devices[params['peripheralId']]
|
||||
self.device = BLESession.found_devices[params['peripheralId']]
|
||||
self.deviceName = self.device.getValueText(0x9) or self.device.getValueText(0x8)
|
||||
try:
|
||||
self.perip = Peripheral(self.device)
|
||||
logger.info(f"connected to the BLE peripheral: {self.deviceName}")
|
||||
BLESession.found_devices.remove(self.device)
|
||||
except BTLEDisconnectError as e:
|
||||
logger.error(f"failed to connect to the BLE device \"{self.deviceName}\": {e}")
|
||||
self.status = self.DONE
|
||||
@@ -576,6 +432,8 @@ class BLESession(Session):
|
||||
if self.perip:
|
||||
res["result"] = None
|
||||
self.status = self.CONNECTED
|
||||
BLESession.nr_connected += 1
|
||||
logger.debug(f"BLE session connected={BLESession.nr_connected}")
|
||||
self.delegate = self.BLEDelegate(self)
|
||||
self.perip.withDelegate(self.delegate)
|
||||
self._cache_characteristics()
|
||||
@@ -668,7 +526,8 @@ async def ws_handler(websocket, path):
|
||||
await session.handle()
|
||||
except Exception as e:
|
||||
logger.error(f"Failure in session for web socket path: {path}")
|
||||
logger.error(e)
|
||||
logger.error(f"{type(e).__name__}: {e}")
|
||||
session.close()
|
||||
|
||||
def stack_trace():
|
||||
print("in stack_trace")
|
||||
@@ -685,18 +544,25 @@ def stack_trace():
|
||||
print(line)
|
||||
|
||||
def main():
|
||||
opts = [opt for opt in sys.argv[1:] if opt.startswith("-")]
|
||||
if "-h" in opts:
|
||||
print((f"Usage: {sys.argv[0]} [OPTS]\n"
|
||||
"OPTS:\t-h Show this help.\n"
|
||||
"\t-d Print debug messages."
|
||||
))
|
||||
sys.exit(1)
|
||||
elif "-d" in opts:
|
||||
global scan_seconds
|
||||
global scan_retry
|
||||
parser = argparse.ArgumentParser(description='start Scratch-link')
|
||||
parser.add_argument('-d', '--debug', action='store_true',
|
||||
help='print debug messages')
|
||||
parser.add_argument('-s', '--scan_seconds', type=float, default=10.0,
|
||||
help='specifiy duration to scan BLE devices in seconds')
|
||||
parser.add_argument('-r', '--scan_retry', type=int, default=1,
|
||||
help='specifiy retry times to scan BLE devices')
|
||||
args = parser.parse_args()
|
||||
if args.debug:
|
||||
print("Print debug messages")
|
||||
logLevel = logging.DEBUG
|
||||
handler.setLevel(logLevel)
|
||||
logger.setLevel(logLevel)
|
||||
scan_seconds = args.scan_seconds
|
||||
scan_retry = args.scan_retry
|
||||
logger.debug(f"set scan_seconds: {scan_seconds}")
|
||||
logger.debug(f"set scan_retry: {scan_retry}")
|
||||
|
||||
# Prepare certificate of the WSS server
|
||||
gencert.prep_cert()
|
||||
@@ -708,7 +574,7 @@ def main():
|
||||
ssl_context.load_cert_chain(localhost_cer, localhost_key)
|
||||
|
||||
start_server = websockets.serve(
|
||||
ws_handler, "device-manager.scratch.mit.edu", 20110, ssl=ssl_context
|
||||
ws_handler, HOSTNAME, 20110, ssl=ssl_context
|
||||
)
|
||||
|
||||
while True:
|
||||
@@ -719,7 +585,13 @@ def main():
|
||||
except KeyboardInterrupt as e:
|
||||
stack_trace()
|
||||
break
|
||||
except socket.gaierror as e:
|
||||
logger.error(f"{type(e).__name__}: {e}")
|
||||
logger.info(f"Check internet connection to {HOSTNAME}. If not "
|
||||
f"available, add '127.0.0.1 {HOSTNAME}' to /etc/hosts.")
|
||||
break
|
||||
except Exception as e:
|
||||
logger.error(f"{type(e).__name__}: {e}")
|
||||
logger.info("Restarting scratch-link...")
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
@@ -4,13 +4,17 @@ usage() {
|
||||
echo "Usage: ${0} COMMAND"
|
||||
echo -e "COMMAND:"
|
||||
echo -e "\tbuild"
|
||||
echo -e "\tupload FILES"
|
||||
echo -e "\tupload-testpypi FILES"
|
||||
echo -e "examples:"
|
||||
echo -e "\t${0} build"
|
||||
echo -e "\t${0} upload dist/*0.2.6*"
|
||||
exit 1
|
||||
}
|
||||
|
||||
case ${1} in
|
||||
build)
|
||||
python setup.py sdist bdist_wheel
|
||||
python -m build
|
||||
;;
|
||||
upload)
|
||||
shift
|
||||
|
3
setup.py
3
setup.py
@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
|
||||
|
||||
setuptools.setup(
|
||||
name="pyscrlink",
|
||||
version="0.2.0",
|
||||
version="0.2.8",
|
||||
author="Shin'ichiro Kawasaki",
|
||||
author_email='kawasaki@juno.dti.ne.jp',
|
||||
description='Scratch-link for Linux with Python',
|
||||
@@ -22,7 +22,6 @@ setuptools.setup(
|
||||
install_requires=[
|
||||
'websockets',
|
||||
'bluepy',
|
||||
'pybluez',
|
||||
'pyOpenSSL',
|
||||
],
|
||||
entry_points={
|
||||
|
Reference in New Issue
Block a user