Add keypair support
Change-Id: Iee1ddab7c5f2f21b453b69668554d8c1119d682d
This commit is contained in:
		
							
								
								
									
										178
									
								
								moganclient/osc/v1/keypair.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										178
									
								
								moganclient/osc/v1/keypair.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,178 @@
 | 
			
		||||
#   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 keypair action implementations"""
 | 
			
		||||
 | 
			
		||||
import copy
 | 
			
		||||
import logging
 | 
			
		||||
import os
 | 
			
		||||
import sys
 | 
			
		||||
 | 
			
		||||
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 CreateKeyPair(command.ShowOne):
 | 
			
		||||
    """Create new public or private key for baremetal server ssh access"""
 | 
			
		||||
 | 
			
		||||
    def get_parser(self, prog_name):
 | 
			
		||||
        parser = super(CreateKeyPair, self).get_parser(prog_name)
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "name",
 | 
			
		||||
            metavar="<name>",
 | 
			
		||||
            help=_("Name of key")
 | 
			
		||||
        )
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "--user",
 | 
			
		||||
            metavar="<user-id>",
 | 
			
		||||
            help=_("ID of user to whom to add key-pair (Admin only)")
 | 
			
		||||
        )
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "--key-type",
 | 
			
		||||
            metavar="<key-type>",
 | 
			
		||||
            help=_("Keypair type. Can be ssh or x509.")
 | 
			
		||||
        )
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "--public-key",
 | 
			
		||||
            metavar="<file>",
 | 
			
		||||
            help=_("Filename for public key to add. If not used, "
 | 
			
		||||
                   "creates a private key.")
 | 
			
		||||
        )
 | 
			
		||||
        return parser
 | 
			
		||||
 | 
			
		||||
    def take_action(self, parsed_args):
 | 
			
		||||
        bc_client = self.app.client_manager.baremetal_compute
 | 
			
		||||
        pub_key = parsed_args.public_key
 | 
			
		||||
        if pub_key:
 | 
			
		||||
            if pub_key == '-':
 | 
			
		||||
                pub_key = sys.stdin.read()
 | 
			
		||||
            else:
 | 
			
		||||
                try:
 | 
			
		||||
                    with open(os.path.expanduser(pub_key)) as f:
 | 
			
		||||
                        pub_key = f.read()
 | 
			
		||||
                except IOError as e:
 | 
			
		||||
                    raise exceptions.CommandError(
 | 
			
		||||
                        _("Can't open or read '%(key)s': %(exc)s")
 | 
			
		||||
                        % {'key': pub_key, 'exc': e}
 | 
			
		||||
                    )
 | 
			
		||||
 | 
			
		||||
        data = bc_client.keypair.create(parsed_args.name, parsed_args.user,
 | 
			
		||||
                                        pub_key, parsed_args.key_type)
 | 
			
		||||
        info = copy.copy(data._info)
 | 
			
		||||
        if not pub_key:
 | 
			
		||||
            private_key = info.pop('private_key')
 | 
			
		||||
            print(private_key)
 | 
			
		||||
        return zip(*sorted(info.items()))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class DeleteKeyPair(command.Command):
 | 
			
		||||
    """Delete baremetal server public or private key(s)"""
 | 
			
		||||
 | 
			
		||||
    def get_parser(self, prog_name):
 | 
			
		||||
        parser = super(DeleteKeyPair, self).get_parser(prog_name)
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            'key',
 | 
			
		||||
            metavar='<key>',
 | 
			
		||||
            nargs='+',
 | 
			
		||||
            help=_("Name of key(s) to delete (name only)")
 | 
			
		||||
        )
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "--user",
 | 
			
		||||
            metavar="<user-id>",
 | 
			
		||||
            help=_("ID of user to whom to add key-pair (Admin only)")
 | 
			
		||||
        )
 | 
			
		||||
        return parser
 | 
			
		||||
 | 
			
		||||
    def take_action(self, parsed_args):
 | 
			
		||||
        bc_client = self.app.client_manager.baremetal_compute
 | 
			
		||||
        result = 0
 | 
			
		||||
        for one_key in parsed_args.key:
 | 
			
		||||
            try:
 | 
			
		||||
                bc_client.keypair.delete(one_key, parsed_args.user)
 | 
			
		||||
            except Exception as e:
 | 
			
		||||
                result += 1
 | 
			
		||||
                LOG.error(_("Failed to delete keypair with name "
 | 
			
		||||
                            "'%(key)s': %(e)s") %
 | 
			
		||||
                          {'key': one_key, 'e': e})
 | 
			
		||||
 | 
			
		||||
        if result > 0:
 | 
			
		||||
            total = len(parsed_args.key)
 | 
			
		||||
            msg = (_("%(result)s of %(total)s keypairs failed "
 | 
			
		||||
                     "to delete.") % {'result': result, 'total': total})
 | 
			
		||||
            raise exceptions.CommandError(msg)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ListKeyPair(command.Lister):
 | 
			
		||||
    """List baremetal server key fingerprints"""
 | 
			
		||||
    def get_parser(self, prog_name):
 | 
			
		||||
        parser = super(ListKeyPair, self).get_parser(prog_name)
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "--user",
 | 
			
		||||
            metavar="<user-id>",
 | 
			
		||||
            help=_("ID of user to whom to add key-pair (Admin only)")
 | 
			
		||||
        )
 | 
			
		||||
        return parser
 | 
			
		||||
 | 
			
		||||
    def take_action(self, parsed_args):
 | 
			
		||||
        bc_client = self.app.client_manager.baremetal_compute
 | 
			
		||||
 | 
			
		||||
        data = bc_client.keypair.list(parsed_args.user)
 | 
			
		||||
 | 
			
		||||
        column_headers = (
 | 
			
		||||
            "Name",
 | 
			
		||||
            "Type",
 | 
			
		||||
            "Fingerprint",
 | 
			
		||||
        )
 | 
			
		||||
        columns = (
 | 
			
		||||
            "Name",
 | 
			
		||||
            "Type",
 | 
			
		||||
            "Fingerprint",
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        return (column_headers,
 | 
			
		||||
                (utils.get_item_properties(
 | 
			
		||||
                    s, columns,
 | 
			
		||||
                ) for s in data))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ShowKeyPair(command.ShowOne):
 | 
			
		||||
    """Display baremetal server key details"""
 | 
			
		||||
 | 
			
		||||
    def get_parser(self, prog_name):
 | 
			
		||||
        parser = super(ShowKeyPair, self).get_parser(prog_name)
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            'keypair',
 | 
			
		||||
            metavar='<keypair>',
 | 
			
		||||
            help=_("Keypair to display (name only)")
 | 
			
		||||
        )
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            "--user",
 | 
			
		||||
            metavar="<user-id>",
 | 
			
		||||
            help=_("ID of user to whom to add key-pair (Admin only)")
 | 
			
		||||
        )
 | 
			
		||||
        return parser
 | 
			
		||||
 | 
			
		||||
    def take_action(self, parsed_args):
 | 
			
		||||
        bc_client = self.app.client_manager.baremetal_compute
 | 
			
		||||
        data = bc_client.keypair.get(parsed_args.keypair, parsed_args.user)
 | 
			
		||||
        info = {}
 | 
			
		||||
        info.update(data._info)
 | 
			
		||||
        return zip(*sorted(info.items()))
 | 
			
		||||
@@ -16,6 +16,7 @@
 | 
			
		||||
from moganclient.common import http
 | 
			
		||||
from moganclient.v1 import availability_zone
 | 
			
		||||
from moganclient.v1 import flavor
 | 
			
		||||
from moganclient.v1 import keypair
 | 
			
		||||
from moganclient.v1 import server
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@@ -31,3 +32,4 @@ class Client(object):
 | 
			
		||||
        self.server = server.ServerManager(self.http_client)
 | 
			
		||||
        self.availability_zone = availability_zone.AvailabilityZoneManager(
 | 
			
		||||
            self.http_client)
 | 
			
		||||
        self.keypair = keypair.KeyPairManager(self.http_client)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										49
									
								
								moganclient/v1/keypair.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								moganclient/v1/keypair.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,49 @@
 | 
			
		||||
#   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.
 | 
			
		||||
#
 | 
			
		||||
 | 
			
		||||
from moganclient.common import base
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class KeyPair(base.Resource):
 | 
			
		||||
    pass
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class KeyPairManager(base.ManagerWithFind):
 | 
			
		||||
    resource_class = KeyPair
 | 
			
		||||
 | 
			
		||||
    def create(self, name, user_id=None, public_key=None, keypair_type=None):
 | 
			
		||||
        url = '/keypairs'
 | 
			
		||||
        data = {'name': name}
 | 
			
		||||
        if user_id:
 | 
			
		||||
            data['user_id'] = user_id
 | 
			
		||||
        if public_key:
 | 
			
		||||
            data['public_key'] = public_key
 | 
			
		||||
        if keypair_type:
 | 
			
		||||
            data['type'] = keypair_type
 | 
			
		||||
        return self._create(url, data=data)
 | 
			
		||||
 | 
			
		||||
    def delete(self, name, user_id=None):
 | 
			
		||||
        url = '/keypairs/%s' % name
 | 
			
		||||
        url = url + '?user_id=%s' % user_id if user_id else url
 | 
			
		||||
        return self._delete(url)
 | 
			
		||||
 | 
			
		||||
    def get(self, name, user_id=None):
 | 
			
		||||
        url = '/keypairs/%s' % name
 | 
			
		||||
        url = url + '?user_id=%s' % user_id if user_id else url
 | 
			
		||||
        return self._get(url)
 | 
			
		||||
 | 
			
		||||
    def list(self, user_id=None):
 | 
			
		||||
        url = '/keypairs' if not user_id else '/keypairs?user_id=%s' % user_id
 | 
			
		||||
        return self._list(url, response_key='keypairs')
 | 
			
		||||
@@ -49,6 +49,10 @@ openstack.baremetal_compute.v1 =
 | 
			
		||||
    baremetal_server_unlock = moganclient.osc.v1.server:UnLockServer
 | 
			
		||||
    baremetal_server_netinfo = moganclient.osc.v1.server:ShowServerNetworkInfo
 | 
			
		||||
    baremetal_availability_zone_list = moganclient.osc.v1.availability_zone:ListAvailabilityZone
 | 
			
		||||
    baremetal_keypair_create = moganclient.osc.v1.keypair:CreateKeyPair
 | 
			
		||||
    baremetal_keypair_show = moganclient.osc.v1.keypair:ShowKeyPair
 | 
			
		||||
    baremetal_keypair_list = moganclient.osc.v1.keypair:ListKeyPair
 | 
			
		||||
    baremetal_keypair_delete = moganclient.osc.v1.keypair:DeleteKeyPair
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
[build_sphinx]
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user