b6d7af07fa
Rather than always returning JSON or YAML, add functionality to return table responses when applicable for ease of reading. Adds some marker exceptions to the Shipyard API Client to handle cases where the user of the client would generally want to take a specific and repeatable course of action instead of handling the response in a case-by-case basis. Moved cli action testing to use the responses library instead of as many internal mocks. Added test response generators for standard api responses. Change-Id: I3a593fb29b6e76d971adc7f3bb3a4b7f378ed091
75 lines
2.8 KiB
Python
75 lines
2.8 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-1.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.
|
|
"""CLI value checks invoked from commands"""
|
|
import arrow
|
|
from arrow.parser import ParserError
|
|
|
|
|
|
def check_action_command(ctx, action_command):
|
|
"""Verifies the action command is valid"""
|
|
if action_command not in ['deploy_site', 'update_site', 'redeploy_server']:
|
|
ctx.fail('Invalid action command. The action commands available are '
|
|
'deploy_site, update_site, and redeploy_server.')
|
|
|
|
|
|
def check_control_action(ctx, action):
|
|
"""Verifies the control action is valid"""
|
|
if action not in ['pause', 'unpause', 'stop']:
|
|
ctx.fail('Invalid action. Please enter pause, unpause, or stop.')
|
|
|
|
|
|
def check_id(ctx, action_id):
|
|
"""Verifies a ULID id is in a valid format"""
|
|
if action_id is None:
|
|
ctx.fail('Invalid ID. None is not a valid action ID.')
|
|
if len(action_id) != 26:
|
|
ctx.fail('Invalid ID. ID can only be 26 characters.')
|
|
if not action_id.isalnum():
|
|
ctx.fail('Invalid ID. ID can only contain letters and numbers.')
|
|
|
|
|
|
def check_workflow_id(ctx, workflow_id):
|
|
"""Verifies that a workflow id matches the desired format"""
|
|
if workflow_id is None:
|
|
ctx.fail('Invalid ID. None is not a valid workflow ID.')
|
|
if '__' not in workflow_id:
|
|
ctx.fail('Invalid ID. The ID must cotain a double underscore '
|
|
'separating the workflow name from the execution date')
|
|
input_date_string = workflow_id.split('__')[1]
|
|
date_format_ok = True
|
|
try:
|
|
parsed_dt = arrow.get(input_date_string)
|
|
if input_date_string != parsed_dt.format('YYYY-MM-DDTHH:mm:ss.SSSSSS'):
|
|
date_format_ok = False
|
|
except ParserError:
|
|
date_format_ok = False
|
|
|
|
if not date_format_ok:
|
|
ctx.fail('Invalid ID. The date portion of the ID must conform to '
|
|
'YYYY-MM-DDTHH:mm:ss.SSSSSS')
|
|
|
|
|
|
def check_reformat_parameter(ctx, param):
|
|
"""Checks for <name>=<value> format"""
|
|
param_dictionary = {}
|
|
try:
|
|
for p in param:
|
|
values = p.split('=')
|
|
param_dictionary[values[0]] = values[1]
|
|
except Exception:
|
|
ctx.fail(
|
|
"Invalid parameter or parameter format for " + p +
|
|
". Please utilize the format: <parameter name>=<parameter value>")
|
|
return param_dictionary
|