jaclang 0.5.6__py3-none-any.whl → 0.5.8__py3-none-any.whl

jaclang/compiler/codeloc.py

@@ -22,6 +22,13 @@ class CodeGenTarget:
     mypy_ast: list[MypyNode] = field(default_factory=lambda: [])
     py_bytecode: Optional[bytes] = None
 
+    def clean(self) -> None:
+        """Clean code generation target."""
+        self.py = ""
+        self.jac = ""
+        self.py_ast = []
+        self.mypy_ast = []
+
 
 class CodeLocInfo:
     """Code location info."""

jaclang/compiler/compile.py

@@ -8,20 +8,19 @@ from jaclang.compiler.passes import Pass
 from jaclang.compiler.passes.main import PyOutPass, pass_schedule
 from jaclang.compiler.passes.tool import JacFormatPass
 from jaclang.compiler.passes.tool.schedules import format_pass
-from jaclang.compiler.passes.transform import Alert
 
 
-def compile_jac(file_path: str) -> list[Alert]:
+def compile_jac(file_path: str, cache_result: bool = False) -> Pass:
     """Start Compile for Jac file and return python code as string."""
     code = jac_file_to_pass(
         file_path=file_path,
         schedule=pass_schedule,
     )
-    if isinstance(code.ir, ast.Module) and not code.errors_had:
+    if cache_result and isinstance(code.ir, ast.Module):
         print_pass = PyOutPass(input_ir=code.ir, prior=code)
+        return print_pass
     else:
-        return code.errors_had
-    return print_pass.errors_had
+        return code
 
 
 def jac_file_to_pass(

jaclang/compiler/constant.py

@@ -201,12 +201,12 @@ class Tokens(str, Enum):
     CARROW_R_P2 = "CARROW_R_P2"
     GLOBAL_OP = "GLOBAL_OP"
     NONLOCAL_OP = "NONLOCAL_OP"
-    HERE_OP = "HERE_OP"
-    SELF_OP = "SELF_OP"
-    INIT_OP = "INIT_OP"
-    POST_INIT_OP = "POST_INIT_OP"
-    SUPER_OP = "SUPER_OP"
-    ROOT_OP = "ROOT_OP"
+    KW_HERE = "KW_HERE"
+    KW_SELF = "KW_SELF"
+    KW_INIT = "KW_INIT"
+    KW_SUPER = "KW_SUPER"
+    KW_ROOT = "KW_ROOT"
+    KW_POST_INIT = "KW_POST_INIT"
     WALKER_OP = "WALKER_OP"
     NODE_OP = "NODE_OP"
     EDGE_OP = "EDGE_OP"
@@ -234,3 +234,49 @@ class Tokens(str, Enum):
     def __str__(self) -> str:
         """Return the string representation of the token."""
         return self.value
+
+
+DELIM_MAP = {
+    Tokens.COMMA: ",",
+    Tokens.EQ: "=",
+    Tokens.DECOR_OP: "@",
+    Tokens.WS: "\n",
+    Tokens.SEMI: ";",
+    Tokens.COLON: ":",
+    Tokens.LBRACE: "{",
+    Tokens.RBRACE: "}",
+    Tokens.LSQUARE: "[",
+    Tokens.RSQUARE: "]",
+    Tokens.LPAREN: "(",
+    Tokens.RPAREN: ")",
+    Tokens.RETURN_HINT: "->",
+    Tokens.DOT: ".",
+}
+
+colors = [
+    "#FFE9E9",
+    "#F0FFF0",
+    "#F5E5FF",
+    "#FFFFE0",
+    "#D2FEFF ",
+    "#E8FFD7",
+    "#FFDEAD",
+    "#FFF0F5",
+    "#F5FFFA",
+    "#FFC0CB",
+    "#7FFFD4",
+    "#C0C0C0",
+    "#ADD8E6",
+    "#FFFAF0",
+    "#f4f3f7",
+    "#f5efff",
+    "#b5d7fd",
+    "#ffc0cb",
+    "#FFC0CB",
+    "#e1d4c0",
+    "#FCDFFF",
+    "#F0FFFF",
+    "#F0F8FF",
+    "#F8F8FF",
+    "#F0FFFF",
+]