diff --git a/sahara/plugins/mapr/abstract/version_handler.py b/sahara/plugins/mapr/abstract/version_handler.py index 7d35b4bd..0931184a 100644 --- a/sahara/plugins/mapr/abstract/version_handler.py +++ b/sahara/plugins/mapr/abstract/version_handler.py @@ -56,6 +56,14 @@ class AbstractVersionHandler(object): def get_edp_engine(self, cluster, job_type): return + @abc.abstractmethod + def get_edp_job_types(self): + return [] + + @abc.abstractmethod + def get_edp_config_hints(self, job_type): + return {} + @abc.abstractmethod def get_context(self, cluster, added=None, removed=None): return diff --git a/sahara/plugins/mapr/base/base_version_handler.py b/sahara/plugins/mapr/base/base_version_handler.py index a512e528..24d88789 100644 --- a/sahara/plugins/mapr/base/base_version_handler.py +++ b/sahara/plugins/mapr/base/base_version_handler.py @@ -40,6 +40,12 @@ class BaseVersionHandler(vh.AbstractVersionHandler): return edp.MapROozieJobEngine(cluster) return None + def get_edp_job_types(self): + return edp.MapROozieJobEngine.get_supported_job_types() + + def get_edp_config_hints(self, job_type): + return edp.MapROozieJobEngine.get_possible_job_config(job_type) + def get_services(self): return self._services diff --git a/sahara/plugins/mapr/plugin.py b/sahara/plugins/mapr/plugin.py index 3c7106ff..5223085c 100644 --- a/sahara/plugins/mapr/plugin.py +++ b/sahara/plugins/mapr/plugin.py @@ -68,6 +68,18 @@ class MapRPlugin(p.ProvisioningPluginBase): v_handler = self._get_handler(cluster.hadoop_version) return v_handler.get_edp_engine(cluster, job_type) + def get_edp_job_types(self, versions=[]): + res = {} + for vers in self.get_versions(): + if not versions or vers in versions: + vh = self._get_handler(vers) + res[vers] = vh.get_edp_job_types() + return res + + def get_edp_config_hints(self, job_type, version): + v_handler = self._get_handler(version) + return v_handler.get_edp_config_hints(job_type) + def get_open_ports(self, node_group): v_handler = self._get_handler(node_group.cluster.hadoop_version) return v_handler.get_open_ports(node_group) diff --git a/sahara/plugins/mapr/versions/mapr_spark/version_handler.py b/sahara/plugins/mapr/versions/mapr_spark/version_handler.py index 37a36c37..19bf763a 100755 --- a/sahara/plugins/mapr/versions/mapr_spark/version_handler.py +++ b/sahara/plugins/mapr/versions/mapr_spark/version_handler.py @@ -48,3 +48,9 @@ class VersionHandler(bvh.BaseVersionHandler): if job_type in edp_engine.MapRSparkEngine.get_supported_job_types(): return edp_engine.MapRSparkEngine(cluster) return None + + def get_edp_job_types(self): + return edp_engine.MapRSparkEngine.get_supported_job_types() + + def get_edp_config_hints(self, job_type): + return edp_engine.MapRSparkEngine.get_possible_job_config(job_type) diff --git a/sahara/plugins/mapr/versions/v3_1_1/version_handler.py b/sahara/plugins/mapr/versions/v3_1_1/version_handler.py index 351a7d35..ab1ea864 100644 --- a/sahara/plugins/mapr/versions/v3_1_1/version_handler.py +++ b/sahara/plugins/mapr/versions/v3_1_1/version_handler.py @@ -73,3 +73,9 @@ class VersionHandler(bvh.BaseVersionHandler): if job_type in edp.MapR3OozieJobEngine.get_supported_job_types(): return edp.MapR3OozieJobEngine(cluster) return None + + def get_edp_job_types(self): + return edp.MapR3OozieJobEngine.get_supported_job_types() + + def get_edp_config_hints(self, job_type): + return edp.MapR3OozieJobEngine.get_possible_job_config(job_type)