Files
python-mistralclient/mistralclient/commands/v2/tasks.py
Nikolay Mahotkin 1df04e7b6b Fixing tasks operations in mistralclient
* Deleted PUT/update operation from lib and CLI
 * Recovered filtering tasks by workflow execution:
   mistral task-list [workflow-execution-id]

Partially implements blueprint mistral-refactor-task-output

Change-Id: I0bf975f874865bbd178120cb3f1062812f3351ac
2015-03-23 14:03:17 +03:00

138 lines
3.4 KiB
Python

# Copyright 2014 - Mirantis, Inc.
# All Rights Reserved
#
# 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.
#
import json
import logging
from cliff import command
from cliff import show
from mistralclient.api.v2 import tasks
from mistralclient.commands.v2 import base
LOG = logging.getLogger(__name__)
def format(task=None):
columns = (
'ID',
'Name',
'Workflow name',
'Execution ID',
'State',
)
if task:
data = (
task.id,
task.name,
task.workflow_name,
task.workflow_execution_id,
task.state,
)
else:
data = (tuple('<none>' for _ in range(len(columns))),)
return columns, data
class List(base.MistralLister):
"""List all tasks."""
def get_parser(self, prog_name):
parser = super(List, self).get_parser(prog_name)
parser.add_argument(
'workflow_execution',
nargs='?',
help='Workflow execution ID associated with list of Tasks.')
return parser
def _get_format_function(self):
return format
def _get_resources(self, parsed_args):
return tasks.TaskManager(self.app.client).list(
parsed_args.workflow_execution
)
class Get(show.ShowOne):
"""Show specific task."""
def get_parser(self, prog_name):
parser = super(Get, self).get_parser(prog_name)
parser.add_argument(
'id',
help='Task identifier')
return parser
def take_action(self, parsed_args):
execution = tasks.TaskManager(self.app.client).get(
parsed_args.id)
return format(execution)
class GetResult(command.Command):
"""Show task output data."""
def get_parser(self, prog_name):
parser = super(GetResult, self).get_parser(prog_name)
parser.add_argument(
'id',
help='Task ID')
return parser
def take_action(self, parsed_args):
result = tasks.TaskManager(self.app.client).get(
parsed_args.id).result
try:
result = json.loads(result)
result = json.dumps(result, indent=4) + "\n"
except:
LOG.debug("Task result is not JSON.")
self.app.stdout.write(result or "\n")
class GetInput(command.Command):
"""Show task input."""
def get_parser(self, prog_name):
parser = super(GetInput, self).get_parser(prog_name)
parser.add_argument(
'id',
help='Task ID'
)
return parser
def take_action(self, parsed_args):
result = tasks.TaskManager(self.app.client).get(
parsed_args.id).input
try:
result = json.loads(result)
result = json.dumps(result, indent=4) + "\n"
except:
LOG.debug("Task result is not JSON.")
self.app.stdout.write(result or "\n")