1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
|
#!/usr/bin/env python
"""
This module allows files to loaded onto the Micro:Bit without
flashing the device.
Example usage:
$ ./tools/upload.py /dev/ttyACM0 ./examples/simple_slalom.py --name main.py
Will upload the simple_slalom game into main.py
"""
import sys
import pyboard
PY2 = sys.version_info < (3,)
def main():
import argparse
cmd_parser = argparse.ArgumentParser(description='Load files onto the microbit.')
cmd_parser.add_argument('-b', '--baudrate', default=115200, help='the baud rate of the serial device')
cmd_parser.add_argument('-u', '--user', default='micro', help='the telnet login username')
cmd_parser.add_argument('-p', '--password', default='python', help='the telnet login password')
cmd_parser.add_argument('-w', '--wait', default=0, type=int, help='seconds to wait for USB connected board to become available')
cmd_parser.add_argument('--name', nargs=1, help='name to save the file as, if different from the file name')
cmd_parser.add_argument('device', nargs=1, help='the serial device or the IP address of the microbit')
cmd_parser.add_argument('file', nargs=1, help='file to load')
args = cmd_parser.parse_args()
def exec_commands(cmds):
try:
pyb = pyboard.Pyboard(args.device[0], args.baudrate, args.user, args.password, args.wait)
pyb.enter_raw_repl()
for cmd in cmds:
ret, ret_err = pyb.exec_raw(cmd, timeout=None, data_consumer=pyboard.stdout_write_bytes)
if ret_err:
break
pyb.exit_raw_repl()
pyb.close()
except Exception as er:
print(er)
sys.exit(1)
except KeyboardInterrupt:
sys.exit(1)
if ret_err:
pyboard.stdout_write_bytes(ret_err)
sys.exit(1)
with open(args.file[0], 'rb') as f:
pyfile = f.read()
if args.name:
name = args.name[0]
else:
name = args.file[0]
script = make_save_script(pyfile, name)
exec_commands(script)
def make_save_script(pyfile, filename):
'''Convert a file into a script that saves that file
into the persistent script area on the microbit'''
output = [
'''
fd = open("%s", "wb")
f = fd.write''' % filename
]
while pyfile:
line = pyfile[:64]
if PY2:
output.append('f(b' + repr(line) + ')')
else:
output.append('f(' + repr(line) + ')')
pyfile = pyfile[64:]
output.append('fd.close()')
return output
if __name__ == "__main__":
main()
|