jaclang 0.5.11__py3-none-any.whl → 0.5.15__py3-none-any.whl

jaclang/cli/cli.py

@@ -1,5 +1,6 @@
 """Command line interface tool for the Jac language."""
 
+import ast as ast3
 import importlib
 import marshal
 import os
@@ -8,10 +9,12 @@ import shutil
 import types
 from typing import Optional
 
+import jaclang.compiler.absyntree as ast
 from jaclang import jac_import
 from jaclang.cli.cmdreg import CommandShell, cmd_registry
 from jaclang.compiler.compile import jac_file_to_pass
 from jaclang.compiler.constant import Constants
+from jaclang.compiler.passes.main.pyast_load_pass import PyastBuildPass
 from jaclang.compiler.passes.main.schedules import py_code_gen_typed
 from jaclang.compiler.passes.tool.schedules import format_pass
 from jaclang.plugin.builtin import dotgen
@@ -290,6 +293,23 @@ def graph(
         print("Not a .jac file.")
 
 
+@cmd_registry.register
+def py_to_jac(filename: str, tree: bool = False) -> None:
+    """Convert a Python file to Jac.
+
+    :param filename: The path to the .py file.
+    :param tree: Flag to show the AST tree.(Default-False).
+    """
+    if filename.endswith(".py"):
+        with open(filename, "r") as f:
+            mod = PyastBuildPass(
+                input_ir=ast.PythonModuleAst(ast3.parse(f.read()), mod_path=filename),
+            ).ir.unparse()
+        print(mod)
+    else:
+        print("Not a .py file.")
+
+
 def start_cli() -> None:
     """
     Start the command line interface.

jaclang/compiler/__init__.py

@@ -3,33 +3,48 @@
 import contextlib
 import logging
 import os
+import shutil
 import sys
 
-cur_dir = os.path.dirname(__file__)
-if not os.path.exists(os.path.join(cur_dir, "__jac_gen__", "jac_parser.py")):
-    from jaclang.vendor.lark.tools import standalone
+
+def generate_static_parser(force: bool = False) -> None:
+    """Generate static parser."""
     from jaclang.utils.helpers import auto_generate_refs
+    from jaclang.vendor.lark.tools import standalone
 
-    os.makedirs(os.path.join(cur_dir, "__jac_gen__"), exist_ok=True)
-    with open(os.path.join(cur_dir, "__jac_gen__", "__init__.py"), "w"):
-        pass
-    save_argv = sys.argv
-    sys.argv = [
-        "lark",
-        os.path.join(cur_dir, "jac.lark"),
-        "-o",
-        os.path.join(cur_dir, "__jac_gen__", "jac_parser.py"),
-        "-c",
-    ]
-    standalone.main()  # type: ignore
-    sys.argv = save_argv
-    auto_generate_refs()
+    cur_dir = os.path.dirname(__file__)
+    if force or not os.path.exists(os.path.join(cur_dir, "generated", "jac_parser.py")):
+        if os.path.exists(os.path.join(cur_dir, "generated")):
+            shutil.rmtree(os.path.join(cur_dir, "generated"))
+        os.makedirs(os.path.join(cur_dir, "generated"), exist_ok=True)
+        with open(os.path.join(cur_dir, "generated", "__init__.py"), "w"):
+            pass
+        save_argv = sys.argv
+        sys.argv = [
+            "lark",
+            os.path.join(cur_dir, "jac.lark"),
+            "-o",
+            os.path.join(cur_dir, "generated", "jac_parser.py"),
+            "-c",
+        ]
+        standalone.main()  # type: ignore
+        sys.argv = save_argv
+        try:
+            auto_generate_refs()
+        except Exception as e:
+            print(f"Error generating reference files: {e}")
 
 
-from .__jac_gen__ import jac_parser as jac_lark  # noqa: E402
+generate_static_parser()
+try:
+    from jaclang.compiler.generated import jac_parser as jac_lark  # noqa: E402
 
+    jac_lark.logger.setLevel(logging.DEBUG)
+except AttributeError:
+    generate_static_parser(force=True)
+    from jaclang.compiler.generated import jac_parser as jac_lark  # noqa: E402
 
-jac_lark.logger.setLevel(logging.DEBUG)
+    jac_lark.logger.setLevel(logging.DEBUG)
 contextlib.suppress(AttributeError)
 TOKEN_MAP = {
     x.name: x.pattern.value
@@ -81,6 +96,7 @@ TOKEN_MAP.update(
         "STAR_MUL": "*",
         "FLOOR_DIV": "//",
         "DIV": "/",
+        "PYNLINE": "::py::",
     }
 )