diff options
author | spisarski <s.pisarski@cablelabs.com> | 2017-02-15 09:13:54 -0700 |
---|---|---|
committer | spisarski <s.pisarski@cablelabs.com> | 2017-02-15 09:15:34 -0700 |
commit | 57777f3df521553a06cd01a3861b415d2905ceca (patch) | |
tree | f3b3be457baec7b5231309989aa3ffa9658cd25d /snaps/openstack/create_keypairs.py | |
parent | 73ef791a1cde68e0d8d69cddf63534fbb90f3e2d (diff) |
Initial patch with all code from CableLabs repository.
Change-Id: I70a2778718c5e7f21fd14e4ad28c9269d3761cc7
Signed-off-by: spisarski <s.pisarski@cablelabs.com>
Diffstat (limited to 'snaps/openstack/create_keypairs.py')
-rw-r--r-- | snaps/openstack/create_keypairs.py | 121 |
1 files changed, 121 insertions, 0 deletions
diff --git a/snaps/openstack/create_keypairs.py b/snaps/openstack/create_keypairs.py new file mode 100644 index 0000000..ea7c811 --- /dev/null +++ b/snaps/openstack/create_keypairs.py @@ -0,0 +1,121 @@ +# Copyright (c) 2016 Cable Television Laboratories, Inc. ("CableLabs") +# and others. 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. +import logging +import os + +from Crypto.PublicKey import RSA +from novaclient.exceptions import NotFound + +from snaps.openstack.utils import nova_utils + +__author__ = 'spisarski' + +logger = logging.getLogger('OpenStackKeypair') + + +class OpenStackKeypair: + """ + Class responsible for creating a keypair in OpenStack + """ + + def __init__(self, os_creds, keypair_settings): + """ + Constructor - all parameters are required + :param os_creds: The credentials to connect with OpenStack + :param keypair_settings: The settings used to create a keypair + """ + self.__os_creds = os_creds + self.keypair_settings = keypair_settings + self.__nova = nova_utils.nova_client(os_creds) + + # Attributes instantiated on create() + self.__keypair = None + + def create(self, cleanup=False): + """ + Responsible for creating the keypair object. + :param cleanup: Denotes whether or not this is being called for cleanup or not + """ + logger.info('Creating keypair %s...' % self.keypair_settings.name) + + try: + self.__keypair = nova_utils.get_keypair_by_name(self.__nova, self.keypair_settings.name) + + if not self.__keypair and not cleanup: + if self.keypair_settings.public_filepath and os.path.isfile(self.keypair_settings.public_filepath): + logger.info("Uploading existing keypair") + self.__keypair = nova_utils.upload_keypair_file(self.__nova, self.keypair_settings.name, + self.keypair_settings.public_filepath) + else: + logger.info("Creating new keypair") + # TODO - Make this value configurable + keys = RSA.generate(1024) + self.__keypair = nova_utils.upload_keypair(self.__nova, self.keypair_settings.name, + keys.publickey().exportKey('OpenSSH')) + nova_utils.save_keys_to_files(keys, self.keypair_settings.public_filepath, + self.keypair_settings.private_filepath) + + return self.__keypair + except Exception as e: + logger.error('Unexpected error creating keypair named - ' + self.keypair_settings.name) + self.clean() + raise Exception(e.message) + + def clean(self): + """ + Removes and deletes the keypair. + """ + if self.__keypair: + try: + nova_utils.delete_keypair(self.__nova, self.__keypair) + except NotFound: + pass + self.__keypair = None + + def get_keypair(self): + """ + Returns the OpenStack keypair object + :return: + """ + return self.__keypair + + +class KeypairSettings: + """ + Class representing a keypair configuration + """ + + def __init__(self, config=None, name=None, public_filepath=None, private_filepath=None): + """ + Constructor - all parameters are optional + :param config: Should be a dict object containing the configuration settings using the attribute names below + as each member's the key and overrides any of the other parameters. + :param name: The keypair name. + :param public_filepath: The path to/from the filesystem where the public key file is or will be stored + :param private_filepath: The path where the generated private key file will be stored + :return: + """ + + if config: + self.name = config.get('name') + self.public_filepath = config.get('public_filepath') + self.private_filepath = config.get('private_filepath') + else: + self.name = name + self.public_filepath = public_filepath + self.private_filepath = private_filepath + + if not self.name: + raise Exception('The attributes name, public_filepath, and private_filepath are required') |