Release 0.0.58

DOCS.md

@@ -264,6 +264,8 @@ Types:
 
 `.` (`msg.prm`) - Allows to access the value of `prm` from `msg`.
 
+`LOGS level: {debug, info, warning, error, critical}, name0: Any, name1: Any, ...` - Logs a message with specified `level` and `name0`, `name1`, ... as arguments.
+
 ## Message Scope <a name = "message-scope"></a>
 
 ### Parameters <a name = "message-scope-parameters"></a>

aasm/__init__.py

@@ -1,7 +1,9 @@
 """Agents Assembly translator"""
 
-__version__ = "0.0.57"
+__version__ = "0.0.58"
 
 from aasm.generating.code import Code
+from aasm.modules.module import Module
 from aasm.generating.python_spade import get_spade_code
+from aasm.generating.python_module import get_modules_for_target
 from aasm.utils.exception import PanicException

aasm/generating/code.py

@@ -4,15 +4,27 @@
 
 
 class Code:
-    def __init__(self, agent_code_lines: List[str], graph_code_lines: List[str]):
+    def __init__(
+        self,
+        agent_code_lines: List[str],
+        graph_code_lines: List[str],
+        module_code_lines: List[str],
+    ):
         self.agent_code_lines: List[str] = agent_code_lines
         self.graph_code_lines: List[str] = graph_code_lines
+        self.module_code_lines: List[str] = module_code_lines
         self._iter_code_lines: List[str] | None = None
         self._iter_idx: int | None = None
         self._iter_num_code_lines: int | None = None
 
     def __iter__(self) -> Code:
-        self._iter_code_lines = self.agent_code_lines + ["\n"] + self.graph_code_lines
+        self._iter_code_lines = (
+            self.agent_code_lines
+            + ["\n"]
+            + self.graph_code_lines
+            + ["\n"]
+            + self.module_code_lines
+        )
         self._iter_num_code_lines = len(self._iter_code_lines)
         self._iter_idx = -1
         return self

aasm/generating/python_code.py

@@ -1,22 +1,30 @@
 from __future__ import annotations
 
-from typing import List
+from typing import List, Set
 
 
 class PythonCode:
     def __init__(self, indent_size: int):
         self.indent_size = indent_size
         self.indent: int = 0
         self.code_lines: List[str] = []
+        self.required_imports: Set[str] = set()
 
     def indent_left(self) -> None:
         self.indent -= self.indent_size
 
     def indent_right(self) -> None:
         self.indent += self.indent_size
 
-    def add_line(self, line: str) -> None:
-        self.code_lines.append(self.indent * " " + line + "\n")
+    def add_line(
+        self, line: str, required_imports: Set[str] | None = None, add_newline=True
+    ) -> None:
+        if required_imports is not None:
+            self.required_imports.update(required_imports)
+        if add_newline:
+            self.code_lines.append(self.indent * " " + line + "\n")
+        else:
+            self.code_lines.append(self.indent * " " + line)
 
     def add_newline(self) -> None:
         self.add_line("")
@@ -25,18 +33,9 @@ def add_newlines(self, count: int) -> None:
         for _ in range(count):
             self.add_newline()
 
-    def add_template(self, template: str, **kwargs: Any) -> None:
-        lines = template.render(kwargs).splitlines()
-        current_indent = 0
-        for line in lines:
-            space_count = len(line) - len(
-                line.lstrip(" ")
-            )  # templates should have 4 space indents per pep8
-            indent_count = space_count // 4
-            if indent_count < current_indent:
-                for _ in range(current_indent - indent_count):
-                    self.indent_left()
-            elif indent_count > current_indent:
-                for _ in range(indent_count - current_indent):
-                    self.indent_right()
-            self.add_line(line.strip())
+    def add_required_imports(self) -> None:
+        lines: List[str] = []
+        for required_import in self.required_imports:
+            lines.append(f"import {required_import}\n")
+        lines.sort()
+        self.code_lines = lines + self.code_lines