plugins/org.eclipse.dltk.python.core/src/org/eclipse/dltk/python/parser/ast/statements/TryStatement.java

/*******************************************************************************
 * Copyright (c) 2005, 2016 IBM Corporation and others.
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *
 *******************************************************************************/
package org.eclipse.dltk.python.parser.ast.statements;

import java.util.Iterator;
import java.util.List;

import org.eclipse.dltk.ast.ASTVisitor;
import org.eclipse.dltk.ast.DLTKToken;
import org.eclipse.dltk.ast.statements.Block;
import org.eclipse.dltk.ast.statements.Statement;
import org.eclipse.dltk.utils.CorePrinter;

/**
 * Try statement.
 */
public class TryStatement extends Statement {
	private Block body;

	private List catchFinallyStatements;

	protected TryStatement(int start, int end) {
		super(start, end);
	}

	public TryStatement(DLTKToken tryToken, Block body, List catchFin) {

		super(tryToken);
		this.body = body;
		this.catchFinallyStatements = catchFin;
	}

	@Override
	public void traverse(ASTVisitor pVisitor) throws Exception {
		if (pVisitor.visit(this)) {
			if (this.body != null) {
				this.body.traverse(pVisitor);
			}
			if (this.catchFinallyStatements != null) {
				Iterator i = this.catchFinallyStatements.iterator();
				while (i.hasNext()) {
					Statement st = (Statement) i.next();
					st.traverse(pVisitor);
				}
			}
			pVisitor.endvisit(this);
		}
	}

	@Override
	public int getKind() {
		return S_TRY;
	}

	public Block getBody() {
		return body;
	}

	public List getCatchFinallyStatements() {
		return catchFinallyStatements;
	}

	@Override
	public void printNode(CorePrinter output) {
		output.formatPrint("try:");
		if (this.body != null) {
			this.body.printNode(output);
		}
		if (this.catchFinallyStatements != null) {
			Iterator i = this.catchFinallyStatements.iterator();
			while (i.hasNext()) {
				Statement st = (Statement) i.next();
				st.printNode(output);
				output.println("");
			}
		}
	}
}