253 lines
9.9 KiB
Python
253 lines
9.9 KiB
Python
# Copyright 2015-2018 Red Hat, Inc.
|
|
#
|
|
# 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 argparse
|
|
import logging
|
|
import sys
|
|
|
|
from openstack import config as os_config
|
|
|
|
from metalsmith import _config
|
|
from metalsmith import _format
|
|
from metalsmith import _provisioner
|
|
from metalsmith import _utils
|
|
from metalsmith import sources
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class NICAction(argparse.Action):
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
assert option_string in ('--port', '--network', '--ip')
|
|
nics = getattr(namespace, self.dest, None) or []
|
|
if option_string == '--network':
|
|
nics.append({'network': values})
|
|
elif option_string == '--ip':
|
|
try:
|
|
network, ip = values.split(':', 1)
|
|
except ValueError:
|
|
raise argparse.ArgumentError(
|
|
self, '--ip format is NETWORK:IP, got %s' % values)
|
|
nics.append({'network': network, 'fixed_ip': ip})
|
|
else:
|
|
nics.append({'port': values})
|
|
setattr(namespace, self.dest, nics)
|
|
|
|
|
|
def _do_deploy(api, args, formatter):
|
|
wait = None if args.no_wait else args.wait
|
|
|
|
capabilities = dict(item.split('=', 1) for item in args.capability)
|
|
if args.ssh_public_key:
|
|
with open(args.ssh_public_key) as fp:
|
|
ssh_keys = [fp.read().strip()]
|
|
else:
|
|
ssh_keys = []
|
|
|
|
if args.hostname and not _utils.is_hostname_safe(args.hostname):
|
|
raise RuntimeError("%s cannot be used as a hostname" % args.hostname)
|
|
|
|
source = sources.detect(args.image,
|
|
kernel=args.image_kernel,
|
|
ramdisk=args.image_ramdisk,
|
|
checksum=args.image_checksum)
|
|
|
|
config = _config.InstanceConfig(ssh_keys=ssh_keys)
|
|
if args.user_name:
|
|
config.add_user(args.user_name, sudo=args.passwordless_sudo)
|
|
|
|
node = api.reserve_node(resource_class=args.resource_class,
|
|
conductor_group=args.conductor_group,
|
|
capabilities=capabilities,
|
|
traits=args.trait,
|
|
candidates=args.candidate)
|
|
instance = api.provision_node(node,
|
|
image=source,
|
|
nics=args.nics,
|
|
root_size_gb=args.root_size,
|
|
swap_size_mb=args.swap_size,
|
|
config=config,
|
|
hostname=args.hostname,
|
|
netboot=args.netboot,
|
|
wait=wait)
|
|
formatter.deploy(instance)
|
|
|
|
|
|
def _do_undeploy(api, args, formatter):
|
|
node = api.unprovision_node(args.node, wait=args.wait)
|
|
formatter.undeploy(node)
|
|
|
|
|
|
def _do_show(api, args, formatter):
|
|
instances = api.show_instances(args.instance)
|
|
formatter.show(instances)
|
|
|
|
|
|
def _do_wait(api, args, formatter):
|
|
instances = api.wait_for_provisioning(args.instance,
|
|
timeout=args.timeout)
|
|
formatter.show(instances)
|
|
|
|
|
|
def _do_list(api, args, formatter):
|
|
instances = api.list_instances()
|
|
formatter.show(instances)
|
|
|
|
|
|
def _parse_args(args, config):
|
|
parser = argparse.ArgumentParser(
|
|
description='Deployment and Scheduling tool for Bare Metal')
|
|
verbosity = parser.add_mutually_exclusive_group()
|
|
verbosity.add_argument('-q', '--quiet', action='store_true',
|
|
help='output only errors')
|
|
verbosity.add_argument('--debug', action='store_true',
|
|
help='output extensive logging')
|
|
verbosity.add_argument('-v', '--verbose', action='count', default=0,
|
|
dest='verbosity',
|
|
help='increase output verbosity, can be specified '
|
|
'up to three times')
|
|
parser.add_argument('--dry-run', action='store_true',
|
|
help='do not take any destructive actions')
|
|
parser.add_argument('--format', choices=list(_format.FORMATS),
|
|
default=_format.DEFAULT_FORMAT,
|
|
help='output format')
|
|
|
|
config.register_argparse_arguments(parser, sys.argv[1:])
|
|
|
|
subparsers = parser.add_subparsers()
|
|
|
|
deploy = subparsers.add_parser('deploy')
|
|
deploy.set_defaults(func=_do_deploy)
|
|
wait_grp = deploy.add_mutually_exclusive_group()
|
|
wait_grp.add_argument('--wait', type=int, default=1800,
|
|
help='time (in seconds) to wait for node to become '
|
|
'active')
|
|
wait_grp.add_argument('--no-wait', action='store_true',
|
|
help='disable waiting for deploy to finish')
|
|
deploy.add_argument('--image', help='image to use (name, UUID or URL)',
|
|
required=True)
|
|
deploy.add_argument('--image-checksum',
|
|
help='image MD5 checksum or URL with checksums')
|
|
deploy.add_argument('--image-kernel', help='URL of the image\'s kernel')
|
|
deploy.add_argument('--image-ramdisk', help='URL of the image\'s ramdisk')
|
|
deploy.add_argument('--network', help='network to use (name or UUID)',
|
|
dest='nics', action=NICAction)
|
|
deploy.add_argument('--port', help='port to attach (name or UUID)',
|
|
dest='nics', action=NICAction)
|
|
deploy.add_argument('--ip', help='attach IP from the network',
|
|
dest='nics', metavar='NETWORK:IP', action=NICAction)
|
|
deploy.add_argument('--netboot', action='store_true',
|
|
help='boot from network instead of local disk')
|
|
deploy.add_argument('--root-size', type=int,
|
|
help='root partition size (in GiB), defaults to '
|
|
'(local_gb - 1)')
|
|
deploy.add_argument('--swap-size', type=int,
|
|
help='swap partition size (in MiB), defaults to '
|
|
'no swap')
|
|
deploy.add_argument('--capability', action='append', metavar='NAME=VALUE',
|
|
default=[], help='capabilities the node should have')
|
|
deploy.add_argument('--trait', action='append',
|
|
default=[], help='trait the node should have')
|
|
deploy.add_argument('--ssh-public-key', help='SSH public key to load')
|
|
deploy.add_argument('--hostname', help='Host name to use, defaults to '
|
|
'Node\'s name or UUID')
|
|
deploy.add_argument('--resource-class',
|
|
help='node resource class to deploy')
|
|
deploy.add_argument('--conductor-group',
|
|
help='conductor group to pick the node from')
|
|
deploy.add_argument('--candidate', action='append',
|
|
help='A candidate node to use for scheduling (can be '
|
|
'specified several times)')
|
|
deploy.add_argument('--user-name', help='Name of the admin user to create')
|
|
deploy.add_argument('--passwordless-sudo', action='store_true',
|
|
help='allow password-less sudo for the user')
|
|
|
|
undeploy = subparsers.add_parser('undeploy')
|
|
undeploy.set_defaults(func=_do_undeploy)
|
|
undeploy.add_argument('node', help='node UUID')
|
|
undeploy.add_argument('--wait', type=int,
|
|
help='time (in seconds) to wait for node to become '
|
|
'available for deployment again')
|
|
|
|
show = subparsers.add_parser('show')
|
|
show.set_defaults(func=_do_show)
|
|
show.add_argument('instance', nargs='+', help='instance UUID(s)')
|
|
|
|
show = subparsers.add_parser('list')
|
|
show.set_defaults(func=_do_list)
|
|
|
|
wait = subparsers.add_parser('wait')
|
|
wait.set_defaults(func=_do_wait)
|
|
wait.add_argument('instance', nargs='+', help='instance UUID(s)')
|
|
wait.add_argument('--timeout', type=int,
|
|
help='time (in seconds) to wait for provisioning.')
|
|
|
|
return parser.parse_args(args)
|
|
|
|
|
|
_URLLIB3_LOGGER = 'urllib3.connectionpool'
|
|
|
|
|
|
def _configure_logging(args):
|
|
log_fmt = ('%(asctime)s %(levelname)s %(name)s: %(message)s'
|
|
if args.debug or args.verbosity
|
|
else '[%(asctime)s] %(message)s')
|
|
|
|
# Verbosity:
|
|
# 0 (the default) - warnings and errors
|
|
# 1 - info from metalsmith, warnings and errors from everything else
|
|
# 2 - debug from metalsmith, info from everything else
|
|
# 3 - the same as --debug
|
|
base_level = logging.WARNING
|
|
metalsmith_level = base_level
|
|
urllib_level = logging.CRITICAL
|
|
|
|
if args.quiet:
|
|
base_level = logging.CRITICAL
|
|
metalsmith_level = base_level
|
|
elif args.debug or args.verbosity > 2:
|
|
base_level = logging.DEBUG
|
|
metalsmith_level = base_level
|
|
urllib_level = logging.INFO
|
|
elif args.verbosity == 2:
|
|
base_level = logging.INFO
|
|
metalsmith_level = logging.DEBUG
|
|
elif args.verbosity == 1:
|
|
metalsmith_level = logging.INFO
|
|
|
|
logging.basicConfig(level=base_level, format=log_fmt)
|
|
logging.getLogger('metalsmith').setLevel(metalsmith_level)
|
|
logging.getLogger(_URLLIB3_LOGGER).setLevel(urllib_level)
|
|
|
|
|
|
def main(args=sys.argv[1:]):
|
|
config = os_config.OpenStackConfig()
|
|
args = _parse_args(args, config)
|
|
_configure_logging(args)
|
|
if args.quiet:
|
|
formatter = _format.NULL_FORMAT
|
|
else:
|
|
formatter = _format.FORMATS[args.format]
|
|
|
|
region = config.get_one(argparse=args)
|
|
api = _provisioner.Provisioner(cloud_region=region, dry_run=args.dry_run)
|
|
|
|
try:
|
|
args.func(api, args, formatter)
|
|
except Exception as exc:
|
|
LOG.critical('%s', exc, exc_info=args.debug)
|
|
sys.exit(1)
|