101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
# 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 cliff import show
|
|
import six
|
|
|
|
from tuning_box.cli import base
|
|
|
|
|
|
class EnvironmentsCommand(base.BaseCommand):
|
|
entity_name = 'environment'
|
|
base_url = '/environments'
|
|
columns = ('id', 'components', 'hierarchy_levels')
|
|
|
|
|
|
class ListEnvironments(EnvironmentsCommand, base.BaseListCommand):
|
|
pass
|
|
|
|
|
|
class ShowEnvironment(EnvironmentsCommand, base.BaseShowCommand):
|
|
pass
|
|
|
|
|
|
class DeleteEnvironment(EnvironmentsCommand, base.BaseDeleteCommand):
|
|
pass
|
|
|
|
|
|
class CreateEnvironment(EnvironmentsCommand, show.ShowOne):
|
|
|
|
def get_parser(self, *args, **kwargs):
|
|
parser = super(CreateEnvironment, self).get_parser(
|
|
*args, **kwargs)
|
|
parser.add_argument(
|
|
'-i', '--components',
|
|
type=str,
|
|
help="Comma separated components IDs",
|
|
)
|
|
parser.add_argument(
|
|
'-l', '--levels',
|
|
type=str,
|
|
help="Comma separated levels names",
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
levels = self._parse_comma_separated(
|
|
parsed_args, 'levels', six.text_type)
|
|
components = self._parse_comma_separated(
|
|
parsed_args, 'components', int)
|
|
|
|
result = self.get_client().post(
|
|
self.base_url,
|
|
{'hierarchy_levels': levels, 'components': components}
|
|
)
|
|
return zip(*result.items())
|
|
|
|
|
|
class UpdateEnvironment(EnvironmentsCommand, base.BaseOneCommand):
|
|
|
|
def get_parser(self, *args, **kwargs):
|
|
parser = super(UpdateEnvironment, self).get_parser(
|
|
*args, **kwargs)
|
|
parser.add_argument(
|
|
'-i', '--components',
|
|
dest='components',
|
|
type=str,
|
|
help="Comma separated components IDs. "
|
|
"Set parameter to [] if you want to pass empty list",
|
|
)
|
|
parser.add_argument(
|
|
'-l', '--levels',
|
|
type=str,
|
|
dest='levels',
|
|
help="Comma separated levels names "
|
|
"Set parameter to [] if you want to pass empty list",
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
data = {}
|
|
if parsed_args.levels is not None:
|
|
data['hierarchy_levels'] = self._parse_comma_separated(
|
|
parsed_args, 'levels', six.text_type)
|
|
if parsed_args.components is not None:
|
|
data['components'] = self._parse_comma_separated(
|
|
parsed_args, 'components', int)
|
|
|
|
result = self.get_client().patch(self.get_url(parsed_args), data)
|
|
if result is None:
|
|
result = self.get_update_message(parsed_args)
|
|
self.app.stdout.write(six.text_type(result))
|