File: upload.py

package info (click to toggle)
firmware-microbit-micropython 1.0.1-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, sid
  • size: 25,448 kB
  • sloc: ansic: 83,496; cpp: 27,664; python: 2,475; asm: 274; makefile: 245; javascript: 41; sh: 25
file content (79 lines) | stat: -rwxr-xr-x 2,617 bytes parent folder | download | duplicates (3)
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()