refactor!: rewrite python package

1 files changed, 75 insertions, 0 deletions

diff --git a/nivis-python/interpreter.py b/nivis-python/interpreter.py
new file mode 100644
index 0000000..8aa8fad
--- /dev/null
+++ b/nivis-python/interpreter.py
@@ -0,0 +1,75 @@
+from psi.lexer import Token
+
+
+__all__ = ['Interpreter']
+
+class Interpreter:
+    """
+    A class representing an interpreter for Psi code.
+
+    Args:
+        ast: The abstract syntax tree (AST) of the code to be interpreted.
+
+    Returns:
+        None
+
+    Example:
+        ```python
+        interpreter = Interpreter(ast)
+        interpreter.interpret()
+        ```
+    """
+
+    def __init__(self, ast):
+        """
+        Initializes an Interpreter object.
+
+        Args:
+            ast: The abstract syntax tree (AST) of the code to be interpreted.
+
+        Returns:
+            None
+        """
+        self.ast = ast
+
+    def interpret(self):
+        """
+        Interprets the code represented by the AST.
+
+        Returns:
+            The result of the interpretation.
+        """
+        return self.interpret_expr(self.ast)
+
+    def interpret_expr(self, node):
+        """
+        Interprets an expression node in the AST.
+
+        Args:
+            node: The expression node to be interpreted.
+
+        Returns:
+            The result of the interpretation.
+        """
+        if isinstance(node, Token):
+            return node.value
+        elif isinstance(node, list):
+            for expr in node:
+                result = self.interpret_expr(expr)
+                if result is not None:
+                    return result
+
+    def interpret_condition(self, node):
+        """
+        Interprets a condition node in the AST.
+
+        Args:
+            node: The condition node to be interpreted.
+
+        Returns:
+            The result of the interpretation.
+        """
+        variable = self.interpret_expr(node[0])
+        value = self.interpret_expr(node[2])
+
+        return variable == value