Déclaration et affichage

spf.py

@@ -6,9 +6,68 @@
 import argparse
 import lark
 import sys
+from enum import Enum
+
+variables = {}
+
+class VariableType(Enum):
+    boolean     = 1
+    integer     = 2
+    string      = 3
+    list        = 4
+
+
+class Variable():
+    def __init__(self, type, value):
+        self.type = type
+        self.value = value
+
+    def __str__(self):
+        if self.type == "booléen":
+            return "Vrai" if self.value else "Faux"
+        return f"{self.value}"
+
+    def __repr__(self):
+        if self.type == "texte":
+            return f"\"{self.value}\""
+        return f"{self.value}"
+
+class SPFInterpreter(lark.visitors.Interpreter):
+    def afficher(self, el):
+        for toprint in el.children[1:]:
+            print(self.visit_children(toprint)[0])
+        return
+
+    def declaration(self, el):
+        type = el.children[0].value
+        name = el.children[1].value
+        assert el.children[2].value == "=", "Unexpected"
+        assert el.children[3].data == "expression", "Unexpected"
+        value = self.visit_children(el.children[3])[0]
+        variables[name] = Variable(type, value)
+
+    def assignation(self, el):
+        name = el.children[0].value
+        assert name in variables, f"Unknown variable : {el.children[0].value} not in {variables}"
+        assert el.children[1].value == "=", "Unexpected"
+        assert el.children[2].data == "expression", "Unexpected"
+        value = self.visit_children(el.children[2])[0]
+        variables[name] = value #TODO: vérifier type
+
+    def expressionleft(self, el):
+        return self.visit_children(el)[0]
+
+
+    def variable(self, el):
+        assert el.children[0].value in variables, f"Unknown variable : {el.children[0].value} not in {variables}"
+        return variables[el.children[0].value]
+
+    # Literals
+    string = lambda self, el: el.children[0][1:-1]
+    entier = lambda self, el: int(el.children[0])
+    true = lambda self, _: True
+    false = lambda self, _: False
 
-class SPFTransformer(lark.Transformer):
-    pass
 
 
 def main():
@@ -22,9 +81,6 @@ def main():
                         action="store_true")
     args = arg_parser.parse_args()
 
-    if args.dump:
-        print("Dump activated (TODO)", file=sys.stderr)
-
     if args.trace:
         print("Trace activated (TODO)", file=sys.stderr)
 
@@ -35,7 +91,11 @@ def main():
         program = spf_input.read()
         parsed = spf_parser.parse(program)
 
-    print(parsed.pretty())
+    interpreter = SPFInterpreter()
+    interpreted = interpreter.visit(parsed)
+
+    if args.dump:
+        print(variables, file=sys.stderr)
 
 
 if __name__ == "__main__":