prerna.reactor.codeexec.LoadPyFromFileProjectPyReactor 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 org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import prerna.project.api.IProject;
import prerna.reactor.AbstractReactor;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.execptions.SemossPixelException;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.AssetUtility;
import prerna.util.Constants;
import prerna.util.UploadInputUtility;
import prerna.util.Utility;

public class LoadPyFromFileProjectPyReactor extends AbstractReactor {

	private static final Logger classLogger = LogManager.getLogger(LoadPyFromFileReactor.class);
	
	
	public LoadPyFromFileProjectPyReactor() {
		this.keysToGet = new String[] {ReactorKeysEnum.FILE_PATH.getKey(), ReactorKeysEnum.ALIAS.getKey(), ReactorKeysEnum.SPACE.getKey()};
	}
	
	@Override
	public NounMetadata execute() {
		organizeKeys();
		
		String appFolder = null;
		String space = keyValue.get(keysToGet[2]);
		
		if(space != null && !space.isEmpty() && !space.equals(AssetUtility.INSIGHT_SPACE_KEY) && !space.equals(AssetUtility.USER_SPACE_KEY))
		{
			appFolder = AssetUtility.getProjectAssetFolder(space) + "/" + Constants.PY_BASE_FOLDER;
			appFolder = appFolder.replace("\\", "/");
		}
		else {
			throw new IllegalArgumentException("Project space is needed");
		}
		
		
		// this also validates
		String filePath = UploadInputUtility.getFilePath(this.store, this.insight);		
		
		String alias = keyValue.get(keysToGet[1]);
		
		IProject project = Utility.getProject(space);

		try {
			String script = alias + " = smssutil.load_module_from_file(module_name='" + alias + "', file_path='" + filePath +"', search='" + appFolder + "')";
			project.getProjectPyTranslator(this.insight).runScript(script);
			return new NounMetadata("Variable set " + alias, PixelDataType.CONST_STRING);

		}
		catch (Exception e) {
			classLogger.error(Constants.STACKTRACE, e);
			throw new SemossPixelException("Unable to load python file as module");
		}
	}
}