133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
# Copyright 2017 AT&T Intellectual Property. All other 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.
|
|
"""Contains commands related to tasks against designs."""
|
|
import click
|
|
import json
|
|
import yaml
|
|
|
|
from drydock_provisioner.cli.task.actions import TaskList
|
|
from drydock_provisioner.cli.task.actions import TaskShow
|
|
from drydock_provisioner.cli.task.actions import TaskCreate
|
|
from drydock_provisioner.cli.task.actions import TaskBuildData
|
|
|
|
|
|
@click.group()
|
|
def task():
|
|
"""Drydock task commands."""
|
|
|
|
|
|
@task.command(name='create')
|
|
@click.option(
|
|
'--design-ref', '-d', help='The design reference for this action')
|
|
@click.option('--action', '-a', help='The action to perform')
|
|
@click.option(
|
|
'--node-names',
|
|
'-n',
|
|
help='The nodes targeted by this action, comma separated')
|
|
@click.option(
|
|
'--rack-names',
|
|
'-r',
|
|
help='The racks targeted by this action, comma separated')
|
|
@click.option(
|
|
'--node-tags',
|
|
'-t',
|
|
help='The nodes by tag name targeted by this action, comma separated')
|
|
@click.option(
|
|
'--block/--no-block',
|
|
'-b',
|
|
help='The CLI will wait until the created completes before exitting',
|
|
default=False)
|
|
@click.option(
|
|
'--poll-interval',
|
|
help='Polling interval to check task status in blocking mode.',
|
|
default=15)
|
|
@click.pass_context
|
|
def task_create(ctx,
|
|
design_ref=None,
|
|
action=None,
|
|
node_names=None,
|
|
rack_names=None,
|
|
node_tags=None,
|
|
block=False,
|
|
poll_interval=15):
|
|
"""Create a task."""
|
|
if not design_ref:
|
|
ctx.fail(
|
|
'Error: Design reference must be specified using --design-ref')
|
|
|
|
if not action:
|
|
ctx.fail('Error: Action must be specified using --action')
|
|
|
|
click.echo(
|
|
json.dumps(
|
|
TaskCreate(
|
|
ctx.obj['CLIENT'],
|
|
design_ref=design_ref,
|
|
action_name=action,
|
|
node_names=[x.strip() for x in node_names.split(',')]
|
|
if node_names else [],
|
|
rack_names=[x.strip() for x in rack_names.split(',')]
|
|
if rack_names else [],
|
|
node_tags=[x.strip()
|
|
for x in node_tags.split(',')] if node_tags else [],
|
|
block=block,
|
|
poll_interval=poll_interval).invoke()))
|
|
|
|
|
|
@task.command(name='list')
|
|
@click.pass_context
|
|
def task_list(ctx):
|
|
"""List tasks."""
|
|
click.echo(json.dumps(TaskList(ctx.obj['CLIENT']).invoke()))
|
|
|
|
|
|
@task.command(name='show')
|
|
@click.option('--task-id', '-t', help='The required task id')
|
|
@click.option(
|
|
'--block/--no-block',
|
|
'-b',
|
|
help='The CLI will wait until the created completes before exitting',
|
|
default=False)
|
|
@click.pass_context
|
|
def task_show(ctx, task_id=None, block=False):
|
|
"""show a task's details."""
|
|
if not task_id:
|
|
ctx.fail('The task id must be specified by --task-id')
|
|
|
|
click.echo(
|
|
json.dumps(TaskShow(ctx.obj['CLIENT'], task_id=task_id).invoke()))
|
|
|
|
|
|
@task.command(name='builddata')
|
|
@click.option('--task-id', '-t', help='The required task id')
|
|
@click.option(
|
|
'--output', '-o', help='The output format (yaml|json)', default='yaml')
|
|
@click.pass_context
|
|
def task_builddata(ctx, task_id=None, output='yaml'):
|
|
"""Show builddata assoicated with ``task_id``."""
|
|
if not task_id:
|
|
ctx.fail('The task id must be specified by --task-id')
|
|
|
|
task_bd = TaskBuildData(ctx.obj['CLIENT'], task_id=task_id).invoke()
|
|
|
|
if output == 'json':
|
|
click.echo(json.dumps(task_bd))
|
|
else:
|
|
if output != 'yaml':
|
|
click.echo(
|
|
'Invalid output format {}, defaulting to YAML.'.format(output))
|
|
click.echo(
|
|
yaml.safe_dump(
|
|
task_bd, allow_unicode=True, default_flow_style=False))
|