M V P Nitesh 0de0c4f2c8 Replace six.iteritems() with .items()
1.As mentioned in [1], we should avoid using six.iteritems to achieve
iterators. We can use dict.items instead, as it will return iterators
in PY3 as well. And dict.items/keys will more readable.
2.In py2, the performance about list should be negligible, see the
link [2].
[1] https://wiki.openstack.org/wiki/Python3
[2] http://lists.openstack.org/pipermail/openstack-dev/2015-June/066391.html

Change-Id: Ifac571e10ee48b119bc4bbb6a5d92200d0c7e482
2017-03-30 15:58:56 +05:30

473 lines
16 KiB
Python

# Copyright 2016 Huawei, Inc. All 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.
#
"""Mogan v1 Baremetal server action implementations"""
import json
import logging
from osc_lib.cli import parseractions
from osc_lib.command import command
from osc_lib import exceptions
from osc_lib import utils
from moganclient.common.i18n import _
LOG = logging.getLogger(__name__)
class ServersActionBase(command.Command):
def _get_parser_with_action(self, prog_name, action):
parser = super(ServersActionBase, self).get_parser(prog_name)
parser.add_argument(
'server',
metavar='<server>',
nargs='+',
help=_("Baremetal server(s) to %s (name or UUID)") % action
)
return parser
def _action_multiple_items(self, parsed_args, action, method_name,
**kwargs):
bc_client = self.app.client_manager.baremetal_compute
result = 0
for one_server in parsed_args.server:
try:
data = utils.find_resource(
bc_client.server, one_server)
method = getattr(bc_client.server, method_name)
method(data.uuid, **kwargs)
except Exception as e:
result += 1
LOG.error(_("Failed to %(action)s server with name or UUID "
"'%(server)s': %(e)s") %
{'action': action, 'server': one_server, 'e': e})
if result > 0:
total = len(parsed_args.server)
msg = (_("%(result)s of %(total)s baremetal servers failed "
"to %(action)s.") % {'result': result, 'total': total,
'action': action})
raise exceptions.CommandError(msg)
class CreateServer(command.ShowOne):
"""Create a new baremetal server"""
def get_parser(self, prog_name):
parser = super(CreateServer, self).get_parser(prog_name)
parser.add_argument(
"name",
metavar="<name>",
help=_("New baremetal server name")
)
parser.add_argument(
"--flavor",
metavar="<flavor>",
required=True,
help=_("ID or Name of baremetal server flavor"),
)
parser.add_argument(
"--image",
metavar="<image>",
required=True,
help=_("ID or Name of image"),
)
parser.add_argument(
"--nic",
metavar="net-id=NETWORK[,port-type=PORT_TYPE]",
required=True,
required_keys=['net-id'],
optional_keys=['port-type'],
action=parseractions.MultiKeyValueAction,
help=_("Create a NIC on the server. "
"(repeat option to create multiple NICs)"),
)
parser.add_argument(
"--description",
metavar="<description>",
help=_("Baremetal server description"),
)
parser.add_argument(
"--availability-zone",
metavar="<zone-name>",
help=_("The availability zone for the baremetal server placement"),
)
parser.add_argument(
"--extra",
metavar="<extra>",
help=_("The extra information for baremetal server"),
)
parser.add_argument(
"--min-count",
type=int,
metavar="<number>",
help=_("Create at least <number> servers (limited by quota)"),
)
parser.add_argument(
"--max-count",
type=int,
metavar="<number>",
help=_("Create up to <number> servers (limited by quota)"),
)
return parser
def take_action(self, parsed_args):
bc_client = self.app.client_manager.baremetal_compute
flavor_data = utils.find_resource(
bc_client.flavor,
parsed_args.flavor)
image_data = utils.find_resource(
self.app.client_manager.image.images,
parsed_args.image)
for nic in parsed_args.nic:
if 'port-type' in nic:
nic['port_type'] = nic['port-type']
del nic['port-type']
if 'net-id' in nic:
nic['net_id'] = nic['net-id']
del nic['net-id']
data = bc_client.server.create(
name=parsed_args.name,
image_uuid=image_data.id,
flavor_uuid=flavor_data.uuid,
description=parsed_args.description,
networks=parsed_args.nic,
availability_zone=parsed_args.availability_zone,
extra=parsed_args.extra,
min_count=parsed_args.min_count,
max_count=parsed_args.max_count
)
info = {}
info.update(data._info)
return zip(*sorted(info.items()))
class DeleteServer(command.Command):
"""Delete existing baremetal erver(s)"""
def get_parser(self, prog_name):
parser = super(DeleteServer, self).get_parser(prog_name)
parser.add_argument(
'server',
metavar='<server>',
nargs='+',
help=_("Baremetal server(s) to delete (name or UUID)")
)
return parser
def take_action(self, parsed_args):
bc_client = self.app.client_manager.baremetal_compute
result = 0
for one_server in parsed_args.server:
try:
data = utils.find_resource(
bc_client.server, one_server)
bc_client.server.delete(data.uuid)
except Exception as e:
result += 1
LOG.error(_("Failed to delete server with name or UUID "
"'%(server)s': %(e)s") %
{'server': one_server, 'e': e})
if result > 0:
total = len(parsed_args.server)
msg = (_("%(result)s of %(total)s baremetal servers failed "
"to delete.") % {'result': result, 'total': total})
raise exceptions.CommandError(msg)
class ListServer(command.Lister):
"""List all baremetal servers"""
def get_parser(self, prog_name):
parser = super(ListServer, self).get_parser(prog_name)
parser.add_argument(
'--long',
action='store_true',
default=False,
help=_("List additional fields in output")
)
parser.add_argument(
'--all-projects',
action='store_true',
default=False,
help=_("List the baremetal servers of all projects, "
"only available for admin users.")
)
return parser
@staticmethod
def _networks_formatter(network_info):
return_info = []
for port_uuid in network_info:
port_ips = []
for fixed_ip in network_info[port_uuid]['fixed_ips']:
port_ips.append(fixed_ip['ip_address'])
return_info.append(', '.join(port_ips))
return '; '.join(return_info)
def take_action(self, parsed_args):
bc_client = self.app.client_manager.baremetal_compute
if parsed_args.long:
data = bc_client.server.list(detailed=True,
all_projects=parsed_args.all_projects)
formatters = {'network_info': self._networks_formatter}
# This is the easiest way to change column headers
column_headers = (
"UUID",
"Name",
"Flavor",
"Status",
"Power State",
"Image",
"Description",
"Availability Zone",
"Networks"
)
columns = (
"uuid",
"name",
"instance_type_uuid",
"status",
"power_state",
"image_uuid",
"description",
"availability_zone",
"network_info"
)
else:
data = bc_client.server.list(all_projects=parsed_args.all_projects)
formatters = None
column_headers = (
"UUID",
"Name",
"Status",
)
columns = (
"uuid",
"name",
"status",
)
return (column_headers,
(utils.get_item_properties(
s, columns, formatters=formatters
) for s in data))
class ShowServer(command.ShowOne):
"""Display baremetal server details"""
def get_parser(self, prog_name):
parser = super(ShowServer, self).get_parser(prog_name)
parser.add_argument(
'server',
metavar='<server>',
help=_("Baremetal server to display (name or UUID)")
)
return parser
def take_action(self, parsed_args):
bc_client = self.app.client_manager.baremetal_compute
data = utils.find_resource(
bc_client.server,
parsed_args.server,
)
info = {}
info.update(data._info)
return zip(*sorted(info.items()))
class UpdateServer(command.ShowOne):
"""Update a baremetal server"""
@staticmethod
def _partition_kv(kv_arg):
if ':' not in kv_arg:
msg = _("Input %s should be a pair of key/value combined "
"by ':'")
raise exceptions.CommandError(msg)
kv = kv_arg.partition(":")
return kv[0], kv[2]
def get_parser(self, prog_name):
parser = super(UpdateServer, self).get_parser(prog_name)
parser.add_argument(
'server',
metavar='<server>',
help=_("Baremetal server to update (name or UUID)")
)
parser.add_argument(
"--description",
metavar="<description>",
help=_("Baremetal Server description"),
)
parser.add_argument(
"--name",
metavar="<description>",
help=_("Baremetal server description"),
)
parser.add_argument(
"--add-extra",
action="append",
type=self._partition_kv,
metavar="<EXTRA_KEY:EXTRA_VALUE>",
help="A pair of key:value to be added to the extra "
"field of the server.")
parser.add_argument(
"--replace-extra",
action="append",
type=self._partition_kv,
metavar="<EXTRA_KEY:EXTRA_VALUE>",
help="A pair of key:value to be update to the extra "
"field of the serve.")
parser.add_argument(
"--remove-extra",
action="append",
metavar="<EXTRA_KEY>",
help="Delete an item of the field of the server with the key "
"specified.")
return parser
def take_action(self, parsed_args):
bc_client = self.app.client_manager.baremetal_compute
server = utils.find_resource(
bc_client.server,
parsed_args.server,
)
updates = []
if parsed_args.description:
updates.append({"op": "replace",
"path": "/description",
"value": parsed_args.description})
if parsed_args.name:
updates.append({"op": "replace",
"path": "/name",
"value": parsed_args.name})
for key, value in parsed_args.add_extra or []:
updates.append({"op": "add",
"path": "/extra/%s" % key,
"value": value})
for key, value in parsed_args.replace_extra or []:
updates.append({"op": "replace",
"path": "/extra/%s" % key,
"value": value})
for key in parsed_args.remove_extra or []:
updates.append({"op": "remove",
"path": "/extra/%s" % key})
data = bc_client.server.update(server_id=server.uuid,
updates=updates)
info = {}
info.update(data._info)
return zip(*sorted(info.items()))
class StartServer(ServersActionBase):
"""Start a baremetal server."""
def get_parser(self, prog_name):
return self._get_parser_with_action(prog_name, 'start')
def take_action(self, parsed_args):
self._action_multiple_items(parsed_args, 'start', 'set_power_state',
power_state='on')
class StopServer(ServersActionBase):
"""Stop baremetal server(s)."""
def get_parser(self, prog_name):
return self._get_parser_with_action(prog_name, 'stop')
def take_action(self, parsed_args):
self._action_multiple_items(parsed_args, 'stop', 'set_power_state',
power_state='off')
class RebootServer(ServersActionBase):
"""Reboot baremetal server(s)."""
def get_parser(self, prog_name):
return self._get_parser_with_action(prog_name, 'reboot')
def take_action(self, parsed_args):
self._action_multiple_items(parsed_args, 'reboot', 'set_power_state',
power_state='reboot')
class LockServer(ServersActionBase):
"""Lock baremetal server(s)."""
def get_parser(self, prog_name):
return self._get_parser_with_action(prog_name, 'lock')
def take_action(self, parsed_args):
self._action_multiple_items(parsed_args, 'lock', 'set_lock_state',
lock_state=True)
class UnLockServer(ServersActionBase):
"""UnLock baremetal server(s)."""
def get_parser(self, prog_name):
return self._get_parser_with_action(prog_name, 'unlock')
def take_action(self, parsed_args):
self._action_multiple_items(parsed_args, 'unlock', 'set_lock_state',
lock_state=False)
class ShowServerNetworkInfo(command.Lister):
"""Display baremetal server's network info"""
def get_parser(self, prog_name):
parser = super(ShowServerNetworkInfo, self).get_parser(prog_name)
parser.add_argument(
'server',
metavar='<server>',
help=_("Baremetal server to display its network information (name "
"or UUID)")
)
return parser
def take_action(self, parsed_args):
bc_client = self.app.client_manager.baremetal_compute
server = utils.find_resource(
bc_client.server,
parsed_args.server,
)
data = bc_client.server.get_network_info(server.uuid)
info = data._info
nics = []
for port_id in list(info):
nic = {'port_id': port_id}
nic.update(info[port_id])
nics.append(nic)
columns = ('network', 'port_id', 'mac_address', 'fixed_ips',
'floatingip', 'port_type')
formatters = {'fixed_ips': lambda s: json.dumps(s, indent=4)}
return (columns,
(utils.get_dict_properties(
s, columns, formatters=formatters) for s in nics))