From ee4e785b1ee366249cc2e2217dc54dc7af38ee73 Mon Sep 17 00:00:00 2001 From: Jussi Pakkanen Date: Wed, 19 Aug 2015 23:34:49 +0300 Subject: Add support for subscripting array objects with []. --- interpreter.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'interpreter.py') diff --git a/interpreter.py b/interpreter.py index 91ec6aa..76964a6 100644 --- a/interpreter.py +++ b/interpreter.py @@ -1056,6 +1056,8 @@ class Interpreter(): return self.evaluate_foreach(cur) elif isinstance(cur, mparser.PlusAssignmentNode): return self.evaluate_plusassign(cur) + elif isinstance(cur, mparser.IndexNode): + return self.evaluate_indexing(cur) elif self.is_elementary_type(cur): return cur else: @@ -1903,6 +1905,18 @@ class Interpreter(): new_value = old_variable + [addition] self.set_variable(varname, new_value) + def evaluate_indexing(self, node): + assert(isinstance(node, mparser.IndexNode)) + iobject = self.evaluate_statement(node.iobject) + if not isinstance(iobject, list): + raise InterpreterException('Tried to index a non-array object.') + index = self.evaluate_statement(node.index) + if not isinstance(index, int): + raise InterpreterException('Index value is not an integer.') + if index < -len(iobject) or index >= len(iobject): + raise InterpreterException('Index %d out of bounds of array of size %d.' % (index, len(iobject))) + return iobject[index] + def is_elementary_type(self, v): if isinstance(v, (int, float, str, bool, list)): return True -- cgit v1.1