returnvisitor.java

Python Development Environment (Python IDE plugin for Eclipse). Features editor, code completion, re

package org.python.pydev.parser.visitors.scope;

import java.util.ArrayList;
import java.util.List;

import org.python.pydev.parser.jython.SimpleNode;
import org.python.pydev.parser.jython.ast.ClassDef;
import org.python.pydev.parser.jython.ast.FunctionDef;
import org.python.pydev.parser.jython.ast.Return;
import org.python.pydev.parser.jython.ast.VisitorBase;
import org.python.pydev.parser.jython.ast.stmtType;

public class ReturnVisitor extends VisitorBase{
    

    public static List<Return> findReturns(FunctionDef functionDef) {
        ReturnVisitor visitor = new ReturnVisitor();
        try {
            for(stmtType b:functionDef.body){
                b.accept(visitor);
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return visitor.ret;
    }
    
    private ArrayList<Return> ret = new ArrayList<Return>();

    @Override
    public Object visitReturn(Return node) throws Exception {
        ret.add(node);
        return null;
    }
    
    @Override
    public void traverse(SimpleNode node) throws Exception {
        node.traverse(this);
    }
    
    @Override
    public Object visitClassDef(ClassDef node) throws Exception {
        return null;
    }
    
    @Override
    public Object visitFunctionDef(FunctionDef node) throws Exception {
        return null;
    }

    @Override
    protected Object unhandled_node(SimpleNode node) throws Exception {
        return null;
    }

}