compile.py

import argparse
import platform
import subprocess
from pathlib import Path

SOURCE_DIR = Path("./skyportal")
DEST_DIR = Path("./dist/lib/skyportal")

COMPILERS = {
    "Windows": Path("./mpy-cross/mpy-cross-windows-9.2.2.static.exe"),
    "Linux": Path("./mpy-cross/mpy-cross-linux-amd64-9.2.2.static"),
    "Darwin": Path("./mpy-cross/mpy-cross-macos-9.2.2-arm64"),
}


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("target", type=str, choices=("pyportal", "feather"))
    args = parser.parse_args()

    DEST_DIR.mkdir(parents=True, exist_ok=True)

    # Remove unnecessary compatibility layer
    to_compile = set(SOURCE_DIR.glob("*.py"))
    if args.target == "pyportal":
        to_compile.remove(SOURCE_DIR / "feather_compat.py")
    elif args.target == "feather":
        to_compile.remove(SOURCE_DIR / "pyportal_compat.py")

    print(f"Found {len(to_compile)} *.py files to compile")
    for filepath in to_compile:
        subprocess.run([COMPILERS[platform.system()], filepath])

    compiled = list(SOURCE_DIR.glob("*.mpy"))
    if len(compiled) != len(to_compile):
        raise RuntimeError(
            f"Number of compiled files less than original source ({len(compiled)} < {len(to_compile)}), may need a longer wait"  # noqa: E501
        )

    print(f"Compiled {len(compiled)} files ... moving to {DEST_DIR}")
    for filepath in compiled:
        dest = DEST_DIR / filepath.name
        dest.unlink(missing_ok=True)

        filepath.rename(DEST_DIR / filepath.name)