Adding Job Types support to CLI

Adding Job Types commands to Sahara
OpenstackClient plugin:

$ dataprocessing job type list
$ dataprocessing job type configs get

Partially implements: blueprint cli-as-openstackclient-plugin

Change-Id: I259b006dc47da92952cdc2289ce0af8c200b39ef

1 files changed, 133 insertions, 0 deletions

diff --git a/saharaclient/osc/v1/job_types.py b/saharaclient/osc/v1/job_types.py
new file mode 100644
index 0000000..ca483b8
--- /dev/null
+++ b/saharaclient/osc/v1/job_types.py
@@ -0,0 +1,133 @@
+# Copyright (c) 2015 Mirantis Inc.
+#
+# 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.
+
+from os import path
+
+from cliff import command
+from cliff import lister
+from openstackclient.common import exceptions
+from openstackclient.common import utils as osc_utils
+from oslo_log import log as logging
+from oslo_serialization import jsonutils
+
+from saharaclient.osc.v1.job_templates import JOB_TYPES_CHOICES
+from saharaclient.osc.v1 import utils
+
+
+class ListJobTypes(lister.Lister):
+    """Lists job types supported by plugins"""
+
+    log = logging.getLogger(__name__ + ".ListJobTypes")
+
+    def get_parser(self, prog_name):
+        parser = super(ListJobTypes, self).get_parser(prog_name)
+        parser.add_argument(
+            '--type',
+            metavar="<type>",
+            choices=JOB_TYPES_CHOICES,
+            help="Get information about specific job type"
+        )
+        parser.add_argument(
+            '--plugin',
+            metavar="<plugin>",
+            help="Get only job types supported by this plugin"
+        )
+        parser.add_argument(
+            '--version',
+            metavar="<version>",
+            help="Get only job types supported by specific version of the "
+                 "plugin. This parameter will be taken into account only if "
+                 "plugin is provided"
+        )
+
+        return parser
+
+    def take_action(self, parsed_args):
+        self.log.debug("take_action(%s)" % parsed_args)
+        client = self.app.client_manager.data_processing
+
+        search_opts = {}
+        if parsed_args.type:
+            search_opts['type'] = parsed_args.type
+        if parsed_args.plugin:
+            search_opts['plugin'] = parsed_args.plugin
+            if parsed_args.version:
+                search_opts['version'] = parsed_args.version
+        elif parsed_args.version:
+            raise exceptions.CommandError(
+                '--version argument should be specified with --plugin '
+                'argument')
+
+        data = client.job_types.list(search_opts=search_opts)
+        for job in data:
+            plugins = []
+            for plugin in job.plugins:
+                versions = ", ".join(sorted(plugin["versions"].keys()))
+                if versions:
+                    versions = "(" + versions + ")"
+                plugins.append(plugin["name"] + versions)
+            job.plugins = ', '.join(plugins)
+
+        columns = ('name', 'plugins')
+        column_headers = utils.prepare_column_headers(columns)
+
+        return (
+            column_headers,
+            (osc_utils.get_item_properties(
+                s,
+                columns
+            ) for s in data)
+        )
+
+
+class GetJobTypeConfigs(command.Command):
+    """Get job type configs"""
+
+    log = logging.getLogger(__name__ + ".GetJobTypeConfigs")
+
+    def get_parser(self, prog_name):
+        parser = super(GetJobTypeConfigs, self).get_parser(prog_name)
+        parser.add_argument(
+            "job_type",
+            metavar="<job-type>",
+            choices=JOB_TYPES_CHOICES,
+            help="Type of the job to provide config information about",
+        )
+        parser.add_argument(
+            '--file',
+            metavar="<file>",
+            help='Destination file (defaults to job type)',
+        )
+        return parser
+
+    def take_action(self, parsed_args):
+        self.log.debug("take_action(%s)" % parsed_args)
+        client = self.app.client_manager.data_processing
+
+        if not parsed_args.file:
+            parsed_args.file = parsed_args.job_type
+
+        data = client.jobs.get_configs(parsed_args.job_type).to_dict()
+
+        if path.exists(parsed_args.file):
+            self.log.error('File "%s" already exists. Choose another one with '
+                           '--file argument.' % parsed_args.file)
+        else:
+            with open(parsed_args.file, 'w') as f:
+                jsonutils.dump(data, f, indent=4)
+                self.log.info(
+                    '"%(type)s" job configs were saved in "%(file)s"'
+                    'file' % {'type': parsed_args.job_type,
+                              'file': parsed_args.file})