Function generateExpressionTokens

python/convert_ast.py:168–200

(expr, tokens=None)

  tokens.append(SplootNode('PY_IDENTIFIER', {}, {'identifier': name.id}))

def generateExpressionTokens(expr, tokens=None):
  if tokens is None:
    tokens = []

  if type(expr) == ast.Call:
    appendCallToken(expr, tokens)
  elif type(expr) == ast.Constant:
    appendConstantToken(expr, tokens)
  elif type(expr) == ast.Name:
    appendIdentifier(expr, tokens)
  elif type(expr) == ast.Attribute:
    appendAttribute(expr, tokens)
  elif type(expr) == ast.BinOp:
    appendBinaryOperatorExpression(expr, tokens)
  elif type(expr) == ast.BoolOp:
    appendBooleanOperatorExpression(expr, tokens)
  elif type(expr) == ast.UnaryOp:
    appendUnaryOperatorExpression(expr, tokens)
  elif type(expr) == ast.Compare:
    appendCompareOperatorExpression(expr, tokens)
  elif type(expr) == ast.List:
    appendListToken(expr, tokens)
  elif type(expr) == ast.Set:
    appendSetToken(expr, tokens)
  elif type(expr) == ast.Tuple:
    appendTupleToken(expr, tokens)
  elif type(expr) == ast.Dict:
    appendDictToken(expr, tokens)
  elif type(expr) == ast.Subscript:
    appendSubscriptExpression(expr, tokens)
  else:
    raise Exception(f'Unrecognised expression type: {ast.dump(expr)}')
  return tokens

def generateExpression(expr, tokens=None):
  tokens = generateExpressionTokens(expr, tokens)