68 lines
2.4 KiB
Python
68 lines
2.4 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 __future__ import print_function
|
|
|
|
import logging
|
|
|
|
from cliff import command as cmd
|
|
from fuelclient.objects import environment as environment_obj
|
|
from fuelclient.objects import release as release_obj
|
|
|
|
from octane.util import env as env_util
|
|
from octane.util import helpers
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def load_network_template(network_template):
|
|
try:
|
|
data = helpers.load_yaml(network_template)
|
|
except Exception:
|
|
LOG.exception("Cannot open network template from %s",
|
|
network_template)
|
|
raise
|
|
return data
|
|
|
|
|
|
def upgrade_env(env_id, release_id, network_template=None):
|
|
release = release_obj.Release(release_id)
|
|
seed_id = env_util.clone_env(env_id, release)
|
|
env_util.copy_fuel_keys(env_id, seed_id)
|
|
if network_template:
|
|
network_template_data = load_network_template(network_template)
|
|
seed_env = environment_obj.Environment(seed_id)
|
|
seed_env.set_network_template_data(network_template_data)
|
|
return seed_id
|
|
|
|
|
|
class UpgradeEnvCommand(cmd.Command):
|
|
"""Create upgrade seed env for env ENV_ID and copy settings to it"""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(UpgradeEnvCommand, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'env_id', type=int, metavar='ENV_ID',
|
|
help="ID of environment to be upgraded")
|
|
parser.add_argument(
|
|
'release_id', type=int, metavar='RELEASE_ID',
|
|
help="ID of a release for the new environment.")
|
|
parser.add_argument(
|
|
'--template', type=str, metavar='TEMPLATE_FILE',
|
|
help="Use network template from file")
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
seed_id = upgrade_env(parsed_args.env_id, parsed_args.release_id,
|
|
network_template=parsed_args.template)
|
|
print(seed_id) # TODO: This shouldn't be needed
|