jaclang 0.7.23__py3-none-any.whl → 0.7.26__py3-none-any.whl

jaclang/plugin/tests/test_jaseci.py

@@ -269,6 +269,75 @@ class TestJaseciPlugin(TestCase):
         )
         self._del_session(session)
 
+    def test_walker_purger(self) -> None:
+        """Test simple persistent object."""
+        session = self.fixture_abs_path("test_walker_purger.session")
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="populate",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="traverse",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="check",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="purge",
+            args=[],
+        )
+        output = self.capturedOutput.getvalue().strip()
+        self.assertEqual(
+            output,
+            (
+                "Root()\n"
+                "A(id=0)\nA(id=1)\n"
+                "B(id=0)\nB(id=1)\nB(id=0)\nB(id=1)\n"
+                "C(id=0)\nC(id=1)\nC(id=0)\nC(id=1)\nC(id=0)\nC(id=1)\nC(id=0)\nC(id=1)\n"
+                "D(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\n"
+                "D(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "125\n124"
+            ),
+        )
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="traverse",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="check",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="purge",
+            args=[],
+        )
+        output = self.capturedOutput.getvalue().strip()
+        self.assertEqual(output, "Root()\n1\n0")
+
+        self._del_session(session)
+
     def trigger_access_validation_test(
         self, give_access_to_full_graph: bool, via_all: bool = False
     ) -> None:
@@ -340,6 +409,55 @@ class TestJaseciPlugin(TestCase):
         self.assertEqual(archs[0], "A(val=2)")
         self.assertEqual(archs[1], "A(val=1)")
 
+        ##############################################
+        #        WITH READ ACCESS BUT ELEVATED       #
+        ##############################################
+
+        self._output2buffer()
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node_forced",
+            args=[20],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node_forced",
+            args=[10],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+        archs = self.capturedOutput.getvalue().strip().split("\n")
+        self.assertTrue(len(archs) == 2)
+
+        # ---------- UPDATE SHOULD HAPPEN ---------- #
+
+        self.assertEqual(archs[0], "A(val=20)")
+        self.assertEqual(archs[1], "A(val=10)")
+
+        # ---------- DISALLOW READ ACCESS ---------- #
+
         self._output2buffer()
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
@@ -400,7 +518,7 @@ class TestJaseciPlugin(TestCase):
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
             entrypoint="update_node",
-            args=[20],
+            args=[200],
             root=self.roots[0],
             node=self.nodes[1],
             session=session,
@@ -408,7 +526,7 @@ class TestJaseciPlugin(TestCase):
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
             entrypoint="update_node",
-            args=[10],
+            args=[100],
             session=session,
             root=self.roots[1],
             node=self.nodes[0],
@@ -433,10 +551,12 @@ class TestJaseciPlugin(TestCase):
         archs = self.capturedOutput.getvalue().strip().split("\n")
         self.assertTrue(len(archs) == 2)
 
-        # --------- UPDATE SHOULD HAPPEN -------- #
+        # ---------- UPDATE SHOULD HAPPEN ---------- #
 
-        self.assertEqual(archs[0], "A(val=20)")
-        self.assertEqual(archs[1], "A(val=10)")
+        self.assertEqual(archs[0], "A(val=200)")
+        self.assertEqual(archs[1], "A(val=100)")
+
+        # ---------- DISALLOW WRITE ACCESS --------- #
 
         self._output2buffer()
         cli.enter(
@@ -474,7 +594,7 @@ class TestJaseciPlugin(TestCase):
         )
         self.assertFalse(self.capturedOutput.getvalue().strip())
 
-        # --------- ROOTS RESET OWN NODE -------- #
+        # ---------- ROOTS RESET OWN NODE ---------- #
 
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),

jaclang/runtimelib/architype.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import inspect
 from dataclasses import asdict, dataclass, field, fields, is_dataclass
 from enum import IntEnum
 from logging import getLogger
@@ -291,9 +292,19 @@ class DSFunc:
     """Data Spatial Function."""
 
     name: str
-    trigger: type | UnionType | tuple[type | UnionType, ...] | None
     func: Callable[[Any, Any], Any] | None = None
 
     def resolve(self, cls: type) -> None:
         """Resolve the function."""
         self.func = getattr(cls, self.name)
+
+    def get_funcparam_annotations(
+        self, func: Callable[[Any, Any], Any] | None
+    ) -> type | UnionType | tuple[type | UnionType, ...] | None:
+        """Get function parameter annotations."""
+        if not func:
+            return None
+        annotation = (
+            inspect.signature(func, eval_str=True).parameters["_jac_here_"].annotation
+        )
+        return annotation if annotation != inspect._empty else None

jaclang/runtimelib/context.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import unittest
 from contextvars import ContextVar
+from dataclasses import MISSING
 from typing import Any, Callable, Optional, cast
 from uuid import UUID
 
@@ -26,6 +27,7 @@ class ExecutionContext:
 
     mem: Memory
     reports: list[Any]
+    custom: Any = MISSING
     system_root: NodeAnchor
     root: NodeAnchor
     entry_node: NodeAnchor

jaclang/runtimelib/importer.py

@@ -344,14 +344,19 @@ class JacImporter(Importer):
                     cachable=spec.cachable,
                     reload=reload if reload else False,
                 )
+
+                # Since this is a compile time error, we can safely raise an exception here.
+                if not codeobj:
+                    raise ImportError(f"No bytecode found for {spec.full_target}")
+
                 try:
-                    if not codeobj:
-                        raise ImportError(f"No bytecode found for {spec.full_target}")
                     with sys_path_context(spec.caller_dir):
                         exec(codeobj, module.__dict__)
                 except Exception as e:
+                    logger.error(e)
                     logger.error(dump_traceback(e))
                     raise e
+
         import_return = ImportReturn(module, unique_loaded_items, self)
         if spec.items:
             import_return.process_items(

jaclang/runtimelib/machine.py

@@ -1,5 +1,7 @@
 """Jac Machine module."""
 
+from __future__ import annotations
+
 import inspect
 import marshal
 import os
@@ -12,6 +14,7 @@ from typing import Optional, Union
 from jaclang.compiler.absyntree import Module
 from jaclang.compiler.compile import compile_jac
 from jaclang.compiler.constant import Constants as Con
+from jaclang.compiler.semtable import SemRegistry
 from jaclang.runtimelib.architype import (
     Architype,
     EdgeArchitype,
@@ -70,6 +73,14 @@ class JacMachine:
             )
         return None
 
+    def get_sem_ir(self, mod_sem_ir: SemRegistry | None) -> None:
+        """Update semtable on the attached JacProgram."""
+        if self.jac_program and mod_sem_ir:
+            if self.jac_program.sem_ir:
+                self.jac_program.sem_ir.registry.update(mod_sem_ir.registry)
+            else:
+                self.jac_program.sem_ir = mod_sem_ir
+
     def load_module(self, module_name: str, module: types.ModuleType) -> None:
         """Load a module into the machine."""
         self.loaded_modules[module_name] = module
@@ -263,14 +274,18 @@ class JacMachine:
 
 
 class JacProgram:
-    """Class to hold the mod_bundle and bytecode for Jac modules."""
+    """Class to hold the mod_bundle bytecode and sem_ir for Jac modules."""
 
     def __init__(
-        self, mod_bundle: Optional[Module], bytecode: Optional[dict[str, bytes]]
+        self,
+        mod_bundle: Optional[Module],
+        bytecode: Optional[dict[str, bytes]],
+        sem_ir: Optional[SemRegistry],
     ) -> None:
         """Initialize the JacProgram object."""
         self.mod_bundle = mod_bundle
         self.bytecode = bytecode or {}
+        self.sem_ir = sem_ir if sem_ir else SemRegistry()
 
     def get_bytecode(
         self,
@@ -292,10 +307,10 @@ class JacProgram:
 
         result = compile_jac(full_target, cache_result=cachable)
         if result.errors_had or not result.ir.gen.py_bytecode:
-            logger.error(
-                f"While importing {len(result.errors_had)} errors"
-                f" found in {full_target}"
-            )
+            for alrt in result.errors_had:
+                # We're not logging here, it already gets logged as the errors were added to the errors_had list.
+                # Regardless of the logging, this needs to be sent to the end user, so we'll printing it to stderr.
+                logger.error(alrt.pretty_print())
             return None
         if result.ir.gen.py_bytecode is not None:
             return marshal.loads(result.ir.gen.py_bytecode)

jaclang/runtimelib/memory.py

@@ -25,6 +25,10 @@ class Memory(Generic[ID, TANCH]):
         self.__mem__.clear()
         self.__gc__.clear()
 
+    def is_cached(self, id: ID) -> bool:
+        """Check if id if already cached."""
+        return id in self.__mem__
+
     def find(
         self,
         ids: ID | Iterable[ID],
@@ -96,7 +100,7 @@ class ShelfStorage(Memory[UUID, Anchor]):
                             and p_d.edges != d.edges
                             and Jac.check_connect_access(d)
                         ):
-                            if not d.edges:
+                            if not d.edges and not isinstance(d.architype, Root):
                                 self.__shelf__.pop(_id, None)
                                 continue
                             p_d.edges = d.edges

jaclang/settings.py

@@ -58,6 +58,9 @@ class Settings:
         """Override settings from environment variables if available."""
         for key in [f.name for f in fields(self)]:
             env_value = os.getenv("JACLANG_" + key.upper())
+            env_value = (
+                env_value if env_value is not None else os.getenv("JAC_" + key.upper())
+            )
             if env_value is not None:
                 setattr(self, key, self.convert_type(env_value))
 

jaclang/tests/fixtures/architype_def_bug.jac

@@ -0,0 +1,17 @@
+# fix test type 2
+walker MyWalker {
+    can travel with `root | MyNode entry {
+        print("MyWalker");
+    }
+}
+
+node MyNode {
+    can work with MyWalker entry {
+        print("MyNode");
+    }
+}
+
+with entry {
+    Node_1 = MyNode();
+    Node_1 spawn MyWalker();
+}

jaclang/tests/fixtures/builtin_dotgen.jac

@@ -27,11 +27,11 @@ with entry{
     d3=dotgen(b[2],edge_limit=5,depth=5);l3=d3|>len; #generate dot for all connected with b[1] node
     d4=dotgen(b[1],bfs=True,edge_type= ["Edge1"],node_limit=100,edge_limit=900,depth=300);l4=d4|>len; #generate dot from nodes with depth 3 connected with b[1] node
     d5=dotgen(b[1],node_limit=10,edge_limit=90);l5:=d5|>len; #generate dot from nodes with depth 3 connected with b[1] node
-    print(d1.count('a(val')==12,d1.count('#FFFFE0')==3,'Root' in d1,d1.count('GenericEdge')==30);
-    print(d2.count('a(val')==19,d2.count('#F5E5FF')==2 ,'Edge1' not in d2,d2.count('GenericEdge')==42);
-    print(d3.count('a(val')==6,d3.count("GenericEdge")==5,d3.count('#F5E5FF')==1);
-    print(d4.count("a(val")==25,d4.count("GenericEdge")==66,d4.count('#FFF0F')==3);
-    print(d5.count("Edge1(val=6)")==2, d5.count("GenericEdge()")==24);
+    print(d1.count('a(val')==12,d1.count('#FFFFE0')==3,'Root' in d1,d1.count('label=""')==30);
+    print(d2.count('a(val')==19,d2.count('#F5E5FF')==2 ,'Edge1' not in d2,d2.count('label=""')==42);
+    print(d3.count('a(val')==6,d3.count('label=""')==5,d3.count('#F5E5FF')==1);
+    print(d4.count("a(val")==25,d4.count('label=""')==66,d4.count('#FFF0F')==3);
+    print(d5.count("Edge1(val=6)")==2, d5.count('label=""')==24);
     # print(l3<l2);
     # print(d1);
     # print(d2);
@@ -39,4 +39,4 @@ with entry{
     # print(d4);
     # print(dotgen(node=b[2],bfs=True,depth=3.96,edge_limit=12,node_limit=12.96));
 
-}
+}

jaclang/tests/fixtures/decl_defn_param_name.jac

@@ -0,0 +1,19 @@
+
+obj SomeClass {
+    can method1(some_long_paramenter_name: int) -> None;
+    can method2(param1:int, param2:str) -> None;
+}
+
+:o:SomeClass:c:method1(short_name:int) -> None {
+  print("short_name =", short_name);
+}
+
+:o:SomeClass:c:method2(p1:int, p2: str) -> None {
+  print("p1 =", p1, ", p2 =", p2);
+}
+
+with entry {
+  sc = SomeClass();
+  sc.method1(42);
+  sc.method2(64, "foobar");
+}

jaclang/tests/fixtures/enum_inside_archtype.jac

@@ -1,20 +1,25 @@
-obj outer{
-    has o1:int=9;
+obj outer {
+    has o1: int = 9;
 
-    obj inner{
-        has i1:int=8;
+    obj inner {
+        has i1: int = 8;
     }
-    can foo(){
+    can foo() {
         return 'foo';
     }
-    enum color{
+    enum  color {
         red,
         green,
-        blue
+        blue,
+        with entry {
+            print('Initializing role system..');
+        },
+        can foo -> str {
+            return 'Accessing privileged Data';
         }
-
+    }
 }
 
-with entry{
-    print(outer.color.green.value);
-}
+with entry {
+    print(outer.color.green.value, outer.color.foo());
+}

jaclang/tests/fixtures/expr_type.jac

@@ -0,0 +1,54 @@
+with entry {
+    a = 4;
+    b = 7;
+    c = a + b; # OpExpr.
+    d = a + b + c;
+
+    h = float(a); # CallExpr.
+
+    (1 < 2);  # ComparisonExpr.
+    ('a' + 'b').upper(); # CallExpr.
+    [1,2][0]; # IndexExpr.
+    not False; # UnaryExpr.
+    "a" if True else "b"; # ConditionalExpr.
+    [i for i in range(10)]; # ListComprehension.
+}
+
+# Remaining expressions to test:
+# AssertTypeExpr,
+# AssignmentExpr,
+# AwaitExpr,
+# BytesExpr,
+# CastExpr,
+# ComplexExpr,
+# DictionaryComprehension,
+# DictExpr,
+# EllipsisExpr,
+# EnumCallExpr,
+# Expression,
+# FloatExpr,
+# GeneratorExpr,
+# IntExpr,
+# LambdaExpr,
+# ListExpr,
+# MemberExpr,
+# NamedTupleExpr,
+# NameExpr,
+# NewTypeExpr,
+# ParamSpecExpr,
+# PromoteExpr,
+# RefExpr,
+# RevealExpr,
+# SetComprehension,
+# SetExpr,
+# SliceExpr,
+# StarExpr,
+# StrExpr,
+# SuperExpr,
+# TupleExpr,
+# TypeAliasExpr,
+# TypedDictExpr,
+# TypeVarExpr,
+# TypeVarTupleExpr,
+# YieldExpr,
+# YieldFromExpr,

jaclang/tests/fixtures/glob_multivar_statement.jac

@@ -0,0 +1,15 @@
+glob a: str, b: str;
+glob c: str;
+
+can do() {
+    :g: a, b, c ;
+
+    (a, b, c) = ("Welcome", "to", "Jaseci!");
+}
+
+with entry {
+    (a, b, c) = ("Hello", "World", "!");
+    print(a, b, c);
+    do();
+    print(a, b, c);
+}

jaclang/tests/fixtures/multi_dim_array_split.jac

@@ -0,0 +1,19 @@
+import:py numpy as np;
+
+with entry {
+    arr = np.array(
+        [[1,   2,  3,  4,  5],
+         [6,   7,  8,  9, 10],
+         [11, 12, 13, 14, 15],
+         [16, 17, 18, 19, 20],
+         [21, 22, 23, 24, 25]]
+    );
+
+    print('Original Array:');
+    print(arr);
+
+    print();
+
+    print('Sliced Array:');
+    print(arr[1:3, 1::2]);
+}

jaclang/tests/fixtures/registry.jac

@@ -13,16 +13,16 @@ Person {
 
 enum 'Personality of the Person'
 Personality {
-   INTROVERT: 'Person who is shy and reticent' = 9,
-   EXTROVERT: 'Person who is outgoing and socially confident'
+    INTROVERT: 'Person who is shy and reticent' = 9,
+    EXTROVERT: 'Person who is outgoing and socially confident'
 }
 
-glob personality_examples: 'Personality Information of Famous People': dict[str, Personality|None] = {
+glob personality_examples: 'Personality Information of Famous People': dict[str, Personality | None] = {
     'Albert Einstein': Personality.INTROVERT,
     'Barack Obama': Personality.EXTROVERT
 };
 
-glob person_value : list[tuple[dict[str, Personality], int]] =(-90);
+glob person_value: list[tuple[dict[str, Personality], int]] = (-90);
 
 can 'GenAI ability'
 genai_ability(x: 'Something': str) -> 'Something Else': str by llm();
@@ -34,13 +34,25 @@ normal_ability(x: 'Something': str) -> 'Something Else': str {
 }
 
 can foo() {
-   person_value=22;
-   can bar() {
-       person_value=33;
-   }
+    person_value = 22;
+    can bar() {
+        person_value = 33;
+    }
 }
 with entry {
     einstein_age: int = 75;
     einstein_age += 1;
     einstein = Person('Albert Einstein', einstein_age);
 }
+
+import from jaclang.runtimelib.machine { JacMachine }
+
+with entry {
+    registry = JacMachine.get().jac_program.sem_ir;
+
+    print(len(registry.registry));
+    print(len(list(registry.registry.items())[0][1]));
+    print(list(registry.registry.items())[3][0].scope);
+    (_, sem_info) = registry.lookup(name="normal_ability");
+    print(len(sem_info.get_children(registry)));
+}

jaclang/tests/main.jac

@@ -0,0 +1,2 @@
+import foo;
+