prerna.reactor.codeexec.PyReactor 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.codeexec;

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

import org.apache.logging.log4j.Logger;

import prerna.algorithm.api.ICodeExecution;
import prerna.ds.py.PyTranslator;
import prerna.ds.py.PyUtils;
import prerna.om.Variable.LANGUAGE;
import prerna.reactor.frame.py.AbstractPyFrameReactor;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Constants;
import prerna.util.DIHelper;
import prerna.util.Utility;

public class PyReactor extends AbstractPyFrameReactor implements ICodeExecution {
	
//	private static transient SecurityManager defaultManager = System.getSecurityManager();
	private static final String CLASS_NAME = PyReactor.class.getName();
	// the code that was executed
	private String code = null;

	@Override
	public NounMetadata execute() {
		String disable_terminal =  DIHelper.getInstance().getProperty(Constants.DISABLE_TERMINAL);
		if(disable_terminal != null && !disable_terminal.isEmpty() ) {
			if(Boolean.parseBoolean(disable_terminal)) {
				throw new IllegalArgumentException("Terminal and user code execution has been disabled.");
			}
		}

		if(!PyUtils.pyEnabled()) {
			throw new IllegalArgumentException("Python is not enabled to use the following command");
		}
		
		//check if py terminal is disabled
		String disable_py_terminal =  DIHelper.getInstance().getProperty(Constants.DISABLE_PY_TERMINAL);
		if(disable_py_terminal != null && !disable_py_terminal.isEmpty() ) {
			if(Boolean.parseBoolean(disable_py_terminal)) {
				throw new IllegalArgumentException("Python terminal has been disabled.");
			}
		}
		Logger logger = getLogger(CLASS_NAME);

		this.code = Utility.decodeURIComponent(this.curRow.get(0).toString());
		int tokens = code.split("\\n").length;

		this.code = fillVars(this.code);
		
		PyTranslator pyTranslator = this.insight.getPyTranslator();
		pyTranslator.setLogger(logger);
		//String output = pyTranslator.runPyAndReturnOutput(code);
		String output = null;
		
		if(code.startsWith("sns.")) {
			return new NounMetadata("Please use PyPlot to plot your chart", PixelDataType.CONST_STRING);
		}
		
		NounMetadata execNoun = null;
//		try
//		{
			//if(tokens > 1) 
			{
//				if(nativePyServer)
//				{
//					Map appMap = insight.getUser().getVarMap();
//					
//					if (appMap != null && appMap.containsKey("PY_VAR_STRING"))
//					{
//						String varFolderAssignment = appMap.get("PY_VAR_STRING").toString();
//						varFolderAssignment = varFolderAssignment.replace("\n", ";");
//						pyTranslator.runScript(varFolderAssignment);
//					}
//					output = pyTranslator.runScript(code, this.insight) + "";
//				}
//				else
					output = pyTranslator.runSingle(code, this.insight) + "";
					execNoun = new NounMetadata(output, PixelDataType.CONST_STRING);
			} 
			/*else {
				//output = pyTranslator.runScript(code) + "";
				output = pyTranslator.runScript(insight.getUser().getVarMap(), code) + "";
			}*/
//		}catch(SemossPixelException ex)
//		{
//			output = ex.getMessage();
//			execNoun = new NounMetadata(output, PixelDataType.ERROR, PixelOperationType.ERROR);
//		}
		List outputs = new ArrayList<>(2);
		outputs.add(execNoun);
		
		boolean smartSync = (insight.getProperty("SMART_SYNC") != null) && insight.getProperty("SMART_SYNC").equalsIgnoreCase("true");
		//forcing smart sync to true
		smartSync = true;

		if(smartSync) {
			// if this returns true
			if(smartSync(pyTranslator)) {
				outputs.add(new NounMetadata(this.insight.getCurFrame(), PixelDataType.FRAME, PixelOperationType.FRAME_HEADERS_CHANGE));
			}
		}
		// call it here.. and if it return true
		// regenerate the metadata. 
		return new NounMetadata(outputs, PixelDataType.CODE, PixelOperationType.CODE_EXECUTION);
	}
	
	
	@Override
	public String getExecutedCode() {
		return this.code;
	}

	@Override
	public LANGUAGE getLanguage() {
		return LANGUAGE.PYTHON;
	}

	@Override
	public boolean isUserScript() {
		return true;
	}
	
}