SubmitEditPythonTransformationCommand.java

/*******************************************************************************
 * Copyright 2012 University of Southern California
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * 	http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 * This code was developed by the Information Integration Group as part 
 * of the Karma project at the Information Sciences Institute of the 
 * University of Southern California.  For more information, publications, 
 * and related projects, please see: http://www.isi.edu/integration
 ******************************************************************************/

package edu.isi.karma.controller.command.transformation;

import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import edu.isi.karma.controller.command.CommandException;
import edu.isi.karma.controller.command.CommandType;
import edu.isi.karma.controller.update.ErrorUpdate;
import edu.isi.karma.controller.update.UpdateContainer;
import edu.isi.karma.controller.update.WorksheetUpdateFactory;
import edu.isi.karma.rep.HNode;
import edu.isi.karma.rep.Node;
import edu.isi.karma.rep.Node.NodeStatus;
import edu.isi.karma.rep.RepFactory;
import edu.isi.karma.rep.Worksheet;
import edu.isi.karma.rep.Workspace;
import edu.isi.karma.webserver.ExecutionController;
import edu.isi.karma.webserver.WorkspaceRegistry;

public class SubmitEditPythonTransformationCommand extends SubmitPythonTransformationCommand {

	private final String targetHNodeId;
	private static Logger logger = LoggerFactory
			.getLogger(SubmitEditPythonTransformationCommand.class);

	public SubmitEditPythonTransformationCommand(String id, String model, String newColumnName, String transformationCode, 
			String worksheetId, String hNodeId, 
			String errorDefaultValue, String targetHNodeId, String selectionId, boolean isJSONOutput) {
		super(id, model, newColumnName, transformationCode, worksheetId, hNodeId, errorDefaultValue, selectionId, isJSONOutput);
		this.targetHNodeId = targetHNodeId;
		this.pythonNodeId = targetHNodeId;
		logger.debug("SubmitEditPythonTransformationCommand:" + id + " newColumnName:" + newColumnName + ", code=" + transformationCode);
	}

	@Override
	public String getCommandName() {
		return this.getClass().getSimpleName();
	}

	@Override
	public String getTitle() {
		return "Edit Python Transformation";
	}

	@Override
	public String getDescription() {
		return newColumnName;
	}

	@Override
	public CommandType getCommandType() {
		return CommandType.undoable;
	}

	@Override
	public UpdateContainer doIt(Workspace workspace) throws CommandException {
		Worksheet worksheet = workspace.getWorksheet(worksheetId);
		RepFactory f = workspace.getFactory();
		HNode hNode = f.getHNode(hNodeId);
		inputColumns.clear();
		outputColumns.clear();
		outputColumns.add(targetHNodeId);
		ExecutionController ctrl = WorkspaceRegistry.getInstance().getExecutionController(
				workspace.getId());

		//this.previousPythonTransformationCommand = ctrl.getWorkspace().getCommandHistory().getCommand(previousCommandId);
		this.saveOrResetColumnValues(workspace, ctrl);

		try
		{
			UpdateContainer c = applyPythonTransformation(workspace, worksheet, f,
					hNode, ctrl, targetHNodeId);
			if (isJSONOutput) {
				HNode targetHNode = workspace.getFactory().getHNode(targetHNodeId);
				Map<String, String> mapping = gatherTransformedResults(workspace, targetHNodeId);
				handleJSONOutput(workspace, mapping, targetHNode);
			}
			WorksheetUpdateFactory.detectSelectionStatusChange(worksheetId, workspace, this);
			return c;
		}
		catch (Exception e )
		{
			logger.error("Error occured during python transformation.",e);
			return new UpdateContainer(new ErrorUpdate("Error occured while creating applying Python transformation to the column."));
		}
	}


	@Override
	public UpdateContainer undoIt(Workspace workspace) {
		for (Node node : affectedNodes) {
			HNode hNode = workspace.getFactory().getHNode(node.getHNodeId());
			hNode.removeNestedTable();
			node.setNestedTable(null, workspace.getFactory());
			node.clearValue(NodeStatus.original);
		}
		affectedNodes.clear();
		try {
			if(previousPythonTransformationCommand instanceof SubmitPythonTransformationCommand)
			{
				SubmitPythonTransformationCommand prevCommand = (SubmitPythonTransformationCommand)previousPythonTransformationCommand;
				//Previous python command exists, lets reset the values, and then start again
				prevCommand.resetColumnValues(workspace);
			}
			UpdateContainer uc = previousPythonTransformationCommand.doIt(workspace);
			return uc;
		} catch (CommandException e) {
			return new UpdateContainer(new ErrorUpdate("Error occured while  applying previous Python transformation to the column."));

		}
	}

}