prerna.reactor.frame.py.DateAddValueReactor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.reactor.frame.py;

import java.util.Arrays;
import java.util.List;
import java.util.Vector;

import prerna.algorithm.api.SemossDataType;
import prerna.ds.OwlTemporalEngineMeta;
import prerna.ds.py.PandasFrame;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;

public class DateAddValueReactor extends AbstractPyFrameReactor {

	/*
	 * Here are the keys that can be passed into the reactor options
	 */
	private static final String UNIT = "unit";
	private static final String VAL_TO_ADD = "val_to_add";
	private static final String NEW_COL = "new_col";
	
	/*
	 * Here are the units that can be used
	 * This is matched with values that are passed into the Python function
	 */

//	private static final String DAY = "day";
//	private static final String WEEK = "week";
//	private static final String MONTH = "month";
//	private static final String YEAR = "year";
	
	public DateAddValueReactor(){
		this.keysToGet = new String[]{ReactorKeysEnum.COLUMN.getKey(), NEW_COL, UNIT, VAL_TO_ADD};
	}
	
	@Override
	public NounMetadata execute() {
		organizeKeys();

		PandasFrame frame = (PandasFrame) getFrame();
		String table = frame.getName();
		String wrapperName = frame.getWrapperName();

		String srcCol = this.keyValue.get(this.keysToGet[0]);
		String newCol = this.keyValue.get(this.keysToGet[1]);
		if(newCol == null || newCol.isEmpty()) {
			newCol = srcCol;
		} else {
			newCol = getCleanNewColName(frame, newCol);
		}
		String unit = this.keyValue.get(this.keysToGet[2]).toLowerCase();
		int value = getValue();
		
		// make sure source column exists
		String[] startingColumns = getColumns(frame);
		List startingColumnsList = new Vector(startingColumns.length);
		startingColumnsList.addAll(Arrays.asList(startingColumns));
		if (srcCol == null || !startingColumnsList.contains(srcCol)) {
			throw new IllegalArgumentException("Need to define an existing date column.");
		}
		
		// python method
		// def add_to_date(this, date_column, output_column, unit_of_measure, value):
		
		StringBuilder script = new StringBuilder();
		script.append(wrapperName).append(".date_add_value(")
			.append(srcCol).append(",")
			.append(newCol).append(",")
			.append(unit).append(",")
			.append(value)
			;
		frame.runScript(script.toString());
		this.addExecutedCode(script.toString());

		NounMetadata retNoun;
		if(newCol.equals(srcCol)){
			retNoun = new NounMetadata(frame, PixelDataType.FRAME, PixelOperationType.FRAME_DATA_CHANGE);
		} else{
			retNoun = new NounMetadata(frame, PixelDataType.FRAME, PixelOperationType.FRAME_HEADERS_CHANGE, PixelOperationType.FRAME_DATA_CHANGE);
			// get src column data type
			String addedColumnDataType = SemossDataType.DATE.toString();
			OwlTemporalEngineMeta metaData = frame.getMetaData();
			metaData.addProperty(table, table + "__" + newCol);
			metaData.setAliasToProperty(table + "__" + newCol, newCol);
			metaData.setDataTypeToProperty(table + "__" + newCol, addedColumnDataType);
			metaData.setDerivedToProperty(table + "__" + newCol, true);
			frame.syncHeaders();
		}
		return retNoun;
	}
	
	private int getValue() {
		GenRowStruct grs = this.store.getNoun(VAL_TO_ADD);
		if(grs == null || grs.isEmpty()) {
			throw new IllegalArgumentException("Missing Necessary Value to Run");
		}
		NounMetadata noun = grs.getNoun(0);
		if(noun.getNounType() == PixelDataType.CONST_INT) {
			return (int) grs.get(0);
		}
		throw new IllegalArgumentException("Missing Necessary Value to Run");
	}
}