micropython/shared/memzip/make-memzip.py

#!/usr/bin/env python
#
# Takes a directory of files and zips them up (as uncompressed files).
# This then gets converted into a C data structure which can be read
# like a filesystem at runtime.
#
# This is somewhat like frozen modules in python, but allows arbitrary files
# to be used.

from __future__ import print_function

import argparse
import os
import subprocess
import sys
import types


def create_zip(zip_filename, zip_dir):
    abs_zip_filename = os.path.abspath(zip_filename)
    save_cwd = os.getcwd()
    os.chdir(zip_dir)
    if os.path.exists(abs_zip_filename):
        os.remove(abs_zip_filename)
    subprocess.check_call(["zip", "-0", "-r", "-D", abs_zip_filename, "."])
    os.chdir(save_cwd)


def create_c_from_file(c_filename, zip_filename):
    with open(zip_filename, "rb") as zip_file:
        with open(c_filename, "wb") as c_file:
            print("#include <stdint.h>", file=c_file)
            print("", file=c_file)
            print("const uint8_t memzip_data[] = {", file=c_file)
            while True:
                buf = zip_file.read(16)
                if not buf:
                    break
                print("   ", end="", file=c_file)
                for byte in buf:
                    if isinstance(byte, types.StringType):
                        print(" 0x{:02x},".format(ord(byte)), end="", file=c_file)
                    else:
                        print(" 0x{:02x},".format(byte), end="", file=c_file)
                print("", file=c_file)
            print("};", file=c_file)


def main():
    parser = argparse.ArgumentParser(
        prog="make-memzip.py",
        usage="%(prog)s [options] [command]",
        description="Generates a C source memzip file.",
    )
    parser.add_argument(
        "-z",
        "--zip-file",
        dest="zip_filename",
        help="Specifies the name of the created zip file.",
        default="memzip_files.zip",
    )
    parser.add_argument(
        "-c",
        "--c-file",
        dest="c_filename",
        help="Specifies the name of the created C source file.",
        default="memzip_files.c",
    )
    parser.add_argument(dest="source_dir", default="memzip_files")
    args = parser.parse_args(sys.argv[1:])

    print("args.zip_filename =", args.zip_filename)
    print("args.c_filename =", args.c_filename)
    print("args.source_dir =", args.source_dir)

    create_zip(args.zip_filename, args.source_dir)
    create_c_from_file(args.c_filename, args.zip_filename)


if __name__ == "__main__":
    main()