translator.py

#!/usr/bin/env python3
import sys
import os
import ast

from core import Transpiler, TranspileError


def main():
    if len(sys.argv) < 2:
        print("Usage: python translator.py file1.py [file2.py ...] [output.cpp]")
        sys.exit(1)

    args = sys.argv[1:]

    # ultimo argomento .cpp = output esplicito
    if args[-1].endswith(".cpp"):
        output_path = args[-1]
        input_paths = args[:-1]
    else:
        input_paths = args
        first = input_paths[0]
        for ext in [".py", ".mpy"]:
            if first.endswith(ext):
                output_path = first[:-len(ext)] + ".cpp"
                break
        else:
            output_path = first + ".cpp"

    if not input_paths:
        print("No input files provided.")
        sys.exit(1)

    modules = []
    module_names = []

    for p in input_paths:
        with open(p, "r", encoding="utf-8") as f:
            src = f.read()
        try:
            tree = ast.parse(src, filename=p)
        except SyntaxError as e:
            print(f"Syntax error in {p}: {e}")
            sys.exit(1)
        modules.append(tree)

        base = os.path.basename(p)
        name, _ = os.path.splitext(base)
        module_names.append(name)

    try:
        transpiler = Transpiler(modules, module_names)
        code = transpiler.transpile()
    except TranspileError as e:
        print("Transpile error:", e)
        sys.exit(1)

    with open(output_path, "w", encoding="utf-8") as f:
        f.write(code)

    print("Generated:", output_path)


if __name__ == "__main__":
    main()