123 lines
3.5 KiB
Python
123 lines
3.5 KiB
Python
from pyboard import Pyboard, PyboardError
|
|
import glob, sys, pyboard
|
|
|
|
_pyb = None
|
|
|
|
def get_pyb(args):
|
|
global _pyb
|
|
if not _pyb:
|
|
print("Connected to badge:", end="", flush=True)
|
|
if not args.device:
|
|
args.device = find_tty()
|
|
|
|
# open the connection to the pyboard
|
|
try:
|
|
_pyb = Pyboard(args.device, args.baudrate, None, None, args.wait)
|
|
except PyboardError as er:
|
|
print(" FAIL")
|
|
print(er)
|
|
sys.exit(1)
|
|
print(" DONE")
|
|
return _pyb
|
|
|
|
def close_pyb():
|
|
global _pyb
|
|
if _pyb:
|
|
_pyb.close()
|
|
|
|
def stop_badge(args, verbose):
|
|
pyb = get_pyb(args)
|
|
print("Stopping running app:", end="", flush=True)
|
|
write_command(pyb, b'\r\x03\x03') # ctrl-C twice: interrupt any running program
|
|
print(" DONE")
|
|
|
|
def write_command(pyb, command):
|
|
flush_input(pyb)
|
|
pyb.serial.write(command)
|
|
flush_input(pyb)
|
|
|
|
def flush_input(pyb):
|
|
n = pyb.serial.inWaiting()
|
|
while n > 0:
|
|
pyb.serial.read(n)
|
|
n = pyb.serial.inWaiting()
|
|
|
|
def soft_reset(args, verbose = True):
|
|
pyb = get_pyb(args)
|
|
if verbose:
|
|
print("Soft reboot:", end="", flush=True)
|
|
write_command(pyb, b'\x04') # ctrl-D: soft reset
|
|
data = pyb.read_until(1, b'soft reboot\r\n')
|
|
if data.endswith(b'soft reboot\r\n'):
|
|
if verbose:
|
|
print(" DONE")
|
|
else:
|
|
if verbose:
|
|
print(" FAIL")
|
|
raise PyboardError('could not soft reboot')
|
|
|
|
def find_tty():
|
|
# Todo: find solution for windows, test in linux
|
|
for pattern in ['/dev/ttyACM*', '/dev/tty.usbmodem*']:
|
|
for path in glob.glob(pattern):
|
|
return path
|
|
print("Couldn't find badge tty - Please make it's plugged in and reset it if necessary")
|
|
sys.exit(1)
|
|
|
|
def check_run(paths):
|
|
for filename in paths:
|
|
with open(filename, 'r') as f:
|
|
pyfile = f.read()
|
|
compile(pyfile + '\n', filename, 'exec')
|
|
|
|
def run(args, paths, verbose=True):
|
|
pyb = get_pyb(args)
|
|
|
|
if verbose:
|
|
print("Preparing execution:", end=" ", flush=True)
|
|
# run any command or file(s) - this is mostly a copy from pyboard.py
|
|
if len(paths):
|
|
# we must enter raw-REPL mode to execute commands
|
|
# this will do a soft-reset of the board
|
|
try:
|
|
pyb.enter_raw_repl()
|
|
except PyboardError as er:
|
|
if verbose:
|
|
print(" FAIL")
|
|
print(er)
|
|
pyb.close()
|
|
sys.exit(1)
|
|
if verbose:
|
|
print(" DONE")
|
|
|
|
def execbuffer(buf):
|
|
try:
|
|
ret, ret_err = pyb.exec_raw(buf, timeout=None, data_consumer=pyboard.stdout_write_bytes)
|
|
except PyboardError as er:
|
|
print(er)
|
|
pyb.close()
|
|
sys.exit(1)
|
|
except KeyboardInterrupt:
|
|
sys.exit(1)
|
|
if ret_err:
|
|
pyb.exit_raw_repl()
|
|
pyb.close()
|
|
pyboard.stdout_write_bytes(ret_err)
|
|
sys.exit(1)
|
|
|
|
try:
|
|
# run any files
|
|
for filename in paths:
|
|
with open(filename, 'rb') as f:
|
|
print("-------- %s --------" % filename)
|
|
pyfile = f.read()
|
|
execbuffer(pyfile)
|
|
|
|
# exiting raw-REPL just drops to friendly-REPL mode
|
|
pyb.exit_raw_repl()
|
|
except OSError as e:
|
|
if "Device not configured" in str(e):
|
|
print("Connection to badge lost") # This can happen on a hard rest
|
|
else:
|
|
raise e
|