1 # Copyright (c) 2016 Cable Television Laboratories, Inc. ("CableLabs")
2 # and others. All rights reserved.
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at:
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
17 from novaclient.exceptions import NotFound
19 from snaps.openstack.utils import nova_utils
21 __author__ = 'spisarski'
23 logger = logging.getLogger('create_image')
26 class OpenStackFlavor:
28 Class responsible for creating a user in OpenStack
31 def __init__(self, os_creds, flavor_settings):
34 :param os_creds: The OpenStack connection credentials
35 :param flavor_settings: The flavor settings
38 self.__os_creds = os_creds
39 self.flavor_settings = flavor_settings
41 self.__nova = nova_utils.nova_client(self.__os_creds)
43 def create(self, cleanup=False):
45 Creates the image in OpenStack if it does not already exist
46 :param cleanup: Denotes whether or not this is being called for cleanup or not
47 :return: The OpenStack flavor object
49 self.__flavor = nova_utils.get_flavor_by_name(self.__nova, self.flavor_settings.name)
51 logger.info('Found flavor with name - ' + self.flavor_settings.name)
53 self.__flavor = nova_utils.create_flavor(self.__nova, self.flavor_settings)
55 logger.info('Did not create flavor due to cleanup mode')
61 Cleanse environment of all artifacts
66 nova_utils.delete_flavor(self.__nova, self.__flavor)
74 Returns the OpenStack flavor object
82 Configuration settings for OpenStack flavor creation
85 def __init__(self, config=None, name=None, flavor_id='auto', ram=None, disk=None, vcpus=None, ephemeral=0, swap=0,
86 rxtx_factor=1.0, is_public=True):
89 :param config: dict() object containing the configuration settings using the attribute names below as each
90 member's the key and overrides any of the other parameters.
91 :param name: the flavor's name (required)
92 :param flavor_id: the string ID (default 'auto')
93 :param ram: the required RAM in MB (required)
94 :param disk: the size of the root disk in GB (required)
95 :param vcpus: the number of virtual CPUs (required)
96 :param ephemeral: the size of the ephemeral disk in GB (default 0)
97 :param swap: the size of the dedicated swap disk in GB (default 0)
98 :param rxtx_factor: the receive/transmit factor to be set on ports if backend supports
99 QoS extension (default 1.0)
100 :param is_public: denotes whether or not the flavor is public (default True)
104 self.name = config.get('name')
106 if config.get('flavor_id'):
107 self.flavor_id = config['flavor_id']
109 self.flavor_id = flavor_id
111 self.ram = config.get('ram')
112 self.disk = config.get('disk')
113 self.vcpus = config.get('vcpus')
115 if config.get('ephemeral'):
116 self.ephemeral = config['ephemeral']
118 self.ephemeral = ephemeral
120 if config.get('swap'):
121 self.swap = config['swap']
125 if config.get('rxtx_factor'):
126 self.rxtx_factor = config['rxtx_factor']
128 self.rxtx_factor = rxtx_factor
130 if config.get('is_public') is not None:
131 self.is_public = config['is_public']
133 self.is_public = is_public
136 self.flavor_id = flavor_id
140 self.ephemeral = ephemeral
142 self.rxtx_factor = rxtx_factor
143 self.is_public = is_public
145 if not self.name or not self.ram or not self.disk or not self.vcpus:
146 raise Exception('The attributes name, ram, disk, and vcpus are required for FlavorSettings')
148 if not isinstance(self.ram, int):
149 raise Exception('The ram attribute must be a integer')
151 if not isinstance(self.disk, int):
152 raise Exception('The ram attribute must be a integer')
154 if not isinstance(self.vcpus, int):
155 raise Exception('The vcpus attribute must be a integer')
157 if self.ephemeral and not isinstance(self.ephemeral, int):
158 raise Exception('The ephemeral attribute must be an integer')
160 if self.swap and not isinstance(self.swap, int):
161 raise Exception('The swap attribute must be an integer')
163 if self.rxtx_factor and not isinstance(self.rxtx_factor, (int, float)):
164 raise Exception('The is_public attribute must be an integer or float')
166 if self.is_public and not isinstance(self.is_public, bool):
167 raise Exception('The is_public attribute must be a boolean')