default.nix

@@ -22,6 +22,7 @@ in stdenv.mkDerivation rec {
     ninja
     py
     py.pkgs.pillow
+    py.pkgs.pyserial
   ];
   src = ./.;
   buildCommand = ''

tools/run.py

+#!/usr/bin/env python3
+
+# or using nix-shell
+#!/usr/bin/env nix-shell
+#! nix-shell -i python3 -p "python3.withPackages (ps: [ ps.pyserial ])"
+
+import sys
+import os.path
+import base64
+import argparse
+import tarfile
+import serial
+
+parser = argparse.ArgumentParser(description="Executes python scripts on card10")
+parser.add_argument(
+    "-r", "--run",
+    required=False,
+    type=argparse.FileType("r"),
+    help="MicroPython file to execute without writing to storage"
+)
+parser.add_argument(
+    "-u", "--upload",
+    required=False,
+    type=argparse.FileType("r"), 
+    help="Tarball to unpack and upload into apps folder",
+)
+parser.add_argument(
+    "-d",
+    "--device",
+    default="/dev/ttyACM0",
+    help="Serial device of the pycardium console",
+)
+
+def applyBase64(s, code, fun="exec", extraSetup="", extraNewline=False):
+    n = 80
+    codechunks = [code[i:i+n] for i in range(0, len(code), n)]
+
+    s.write(b"\x03") # ctrl-c
+    s.readline()
+    s.write(b"import ubinascii; \r\n")
+    s.readline()
+    s.write(b"code=[\r\n")
+    s.readline()
+    for chunk in codechunks:
+        s.write(b"'"+chunk+b"',\r\n")
+        s.readline() #dont echo this line
+    s.write(b"'']\r\n")
+    s.readline()
+    #print(s.readline())
+
+    for cmd in extraSetup:
+        if len(cmd) > 0:
+            s.write(bytes(cmd, 'utf-8'))
+            s.write(b"\r\n")
+            s.readline()
+        else:
+            s.write(b"\r\n")
+            s.readline()
+
+    s.write(bytes(fun, 'utf-8'))
+    s.write(b"(ubinascii.a2b_base64(''.join(code)))\r\n")
+    if extraNewline:
+        s.write(b"\r\n")
+        s.readline()
+        s.readline()
+    s.readline()
+
+
+args = parser.parse_args()
+
+if args.run is None and args.upload is None:
+    print("Either -e or -u has to be specified")
+
+with serial.Serial(port=args.device, baudrate=115200) as s:
+
+    if args.upload is not None:
+        tar = tarfile.TarFile.gzopen(args.upload.name)
+        for tarinfo in tar:
+            print(tarinfo.name)
+            payload = base64.b64encode(tar.extractfile(tarinfo.name).read())
+            applyBase64(s, payload,
+                        fun="with open('/apps/{}',\"w\") as f: f.write".format(tarinfo.name),
+                        extraSetup=[
+                            "try: os.mkdir(\"/apps/{}\")".format(os.path.dirname(tarinfo.name)),
+                            "except: pass",
+                            ""
+                        ],
+                        extraNewline=True)
+
+    elif args.run is not None:
+        payload = base64.b64encode(bytes(args.run.read(), "utf-8"))
+        applyBase64(s, payload)
+
+        try:
+            buf = b""
+            while True:
+                buf += s.read()
+                
+                try:
+                    char = buf.decode("utf-8")
+                    print(char, end="")
+                    buf = b""
+
+                except UnicodeDecodeError as e:
+                    if len(buf) > 3:
+                        print(e)
+                        buf = b""
+
+        except KeyboardInterrupt:
+            print('CTRL-C detected. Exiting gracefully.')
+        s.write(b"\x03")