Add the function of getting endpoint

[snaps.git] / snaps / openstack / create_flavor.py

diff --git a/snaps/openstack/create_flavor.py b/snaps/openstack/create_flavor.py
index 68a7080..2d78be4 100644 (file)
--- a/snaps/openstack/create_flavor.py
+++ b/snaps/openstack/create_flavor.py
@@ -22,7 +22,8 @@ __author__ = 'spisarski'
 
 logger = logging.getLogger('create_image')
 
-DEFAULT_METADATA = {'hw:mem_page_size': 'any'}
+MEM_PAGE_SIZE_ANY = {'hw:mem_page_size': 'any'}
+MEM_PAGE_SIZE_LARGE = {'hw:mem_page_size': 'large'}
 
 
 class OpenStackFlavor:
@@ -40,7 +41,7 @@ class OpenStackFlavor:
         self.__os_creds = os_creds
         self.flavor_settings = flavor_settings
         self.__flavor = None
-        self.__nova = nova_utils.nova_client(self.__os_creds)
+        self.__nova = None
 
     def create(self, cleanup=False):
         """
@@ -48,6 +49,7 @@ class OpenStackFlavor:
         :param cleanup: Denotes whether or not this is being called for cleanup or not
         :return: The OpenStack flavor object
         """
+        self.__nova = nova_utils.nova_client(self.__os_creds)
         self.__flavor = nova_utils.get_flavor_by_name(self.__nova, self.flavor_settings.name)
         if self.__flavor:
             logger.info('Found flavor with name - ' + self.flavor_settings.name)
@@ -88,7 +90,7 @@ class FlavorSettings:
     """
 
     def __init__(self, config=None, name=None, flavor_id='auto', ram=None, disk=None, vcpus=None, ephemeral=0, swap=0,
-                 rxtx_factor=1.0, is_public=True, metadata=DEFAULT_METADATA):
+                 rxtx_factor=1.0, is_public=True, metadata=None):
         """
         Constructor
         :param config: dict() object containing the configuration settings using the attribute names below as each