64 lines
2.6 KiB
Python
64 lines
2.6 KiB
Python
from smnp.ast.node.identifier import Identifier, FunctionCall
|
|
from smnp.error.runtime import RuntimeException
|
|
from smnp.runtime.evaluator import Evaluator
|
|
from smnp.runtime.evaluators.expression import expressionEvaluator
|
|
from smnp.runtime.evaluators.iterable import abstractIterableEvaluator
|
|
from smnp.runtime.tools.error import updatePos
|
|
|
|
|
|
class AccessEvaluator(Evaluator):
|
|
|
|
@classmethod
|
|
def evaluator(cls, node, environment):
|
|
left = expressionEvaluator(doAssert=True)(node.left, environment).value #TODO check if it isn't necessary to verify 'result' attr of EvaluatioNResult
|
|
right = node.right
|
|
|
|
if type(right) == Identifier:
|
|
try:
|
|
return left.properties[right.value]
|
|
except KeyError:
|
|
raise RuntimeException(f"Unknown property '{right.value}' of type '{left.type.name.lower()}'", right.pos)
|
|
|
|
if type(right) == FunctionCall:
|
|
try:
|
|
arguments = abstractIterableEvaluator(expressionEvaluator(doAssert=True))(right.arguments, environment)
|
|
return environment.invokeMethod(left, right.name.value, arguments)
|
|
except RuntimeException as e:
|
|
raise updatePos(e, right)
|
|
|
|
#
|
|
# def evaluateAccess(access, environment):
|
|
# element = evaluate(access.element, environment)
|
|
# if type(access.property) == FunctionCallNode:
|
|
# return evaluateMethodCall(element, access.property, environment)
|
|
# if type(access.property) == AccessNode:
|
|
# return evaluateAccess(access.property, environment)
|
|
#
|
|
# raise RuntimeException("Not implemented yet", access.property.pos)
|
|
# # TODO only methods can be handled so far
|
|
#
|
|
#
|
|
# def evaluateMethodCall(element, methodCall, environment):
|
|
# try:
|
|
# methodName = methodCall.identifier.identifier
|
|
# arguments = evaluateList(methodCall.arguments, environment)
|
|
#
|
|
# return environment.invokeMethod(methodName, element, arguments)
|
|
#
|
|
# except SmnpException as e:
|
|
# e.pos = methodCall.pos
|
|
# raise e
|
|
# for name, library in environment.customFunctions.items():
|
|
# if funcName == name:
|
|
# if len(library['params']) != len(arguments):
|
|
# raise RuntimeException(functionCall.pos, f"Calling '{funcName}' requires {len(library['params'])} and {len(arguments)} was passed")
|
|
# environment.scopes.append({ library['params'][i].identifier: v for i, v in enumerate(arguments) })
|
|
# returnValue = None
|
|
# for node in library['body']:
|
|
# if not isinstance(node, ReturnNode):
|
|
# evaluate(node, environment)
|
|
# else:
|
|
# returnValue = evaluateReturn(node, environment)
|
|
# environment.scopes.pop(-1)
|
|
# return returnValue
|