Function generateSplootStatement

python/convert_ast.py:285–328

(statement)

  }, {'id': None})

def generateSplootStatement(statement):
  global lineno
  statement_lineno = lineno
  lineno += 1
  if type(statement) == ast.Expr:
    expr = generateExpression(statement.value)
    return SplootNode("PYTHON_STATEMENT", {"statement": [expr]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.Assign:
    assign = generateAssignment(statement)
    return SplootNode("PYTHON_STATEMENT", {"statement": [assign]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.For:
    forNode = generateFor(statement)
    return SplootNode("PYTHON_STATEMENT", {"statement": [forNode]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.While:
    whileNode = generateWhile(statement)
    return SplootNode("PYTHON_STATEMENT", {"statement": [whileNode]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.If:
    return SplootNode("PYTHON_STATEMENT", {"statement": [generateIf(statement)]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.FunctionDef:
    func = generateFunction(statement)
    return SplootNode("PYTHON_STATEMENT", {"statement": [func]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.Break:
    return SplootNode("PYTHON_STATEMENT", {"statement": [SplootNode('PY_BREAK')]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.Continue:
    return SplootNode("PYTHON_STATEMENT", {"statement": [SplootNode('PY_CONTINUE')]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.Return:
    value = SplootNode("PYTHON_EXPRESSION", {"tokens": []})

    if statement.value:
      value = generateExpression(statement.value)

    return SplootNode("PYTHON_STATEMENT", {"statement": [SplootNode("PYTHON_RETURN", {"value": [value]})]}, metadata={'lineno': statement_lineno})
  elif type(statement) == ast.Import:
    return SplootNode("PYTHON_STATEMENT", {"statement": [generateImport(statement)]})
  elif type(statement) == ast.ImportFrom:
    return SplootNode("PYTHON_STATEMENT", {"statement": [generateImportFrom(statement)]})
  elif type(statement) == ast_comments.Comment:
    if statement.value == "##SPLOOTCODEEMPTYLINE":
      return SplootNode("PYTHON_STATEMENT", {"statement": []})
    return SplootNode("PYTHON_STATEMENT", {"statement": [SplootNode("PY_COMMENT", {}, {"value": statement.value.lstrip().lstrip('#').lstrip()})]})
  elif type(statement) == ast.Pass:
    return None
  else:
    raise Exception(f'Unrecognised statement type: {type(statement)}')
  

def splootFromPython(codeString):