1 files changed, 64 insertions, 0 deletions
diff --git a/tosca2heat/heat-translator/translator/common/flavors.py b/tosca2heat/heat-translator/translator/common/flavors.py
new file mode 100644
index 0000000..c44f883
--- /dev/null
+++ b/tosca2heat/heat-translator/translator/common/flavors.py
@@ -0,0 +1,64 @@
+# 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
+
+try:
+    import novaclient.client
+    client_available = True
+except ImportError:
+    client_available = False
+    pass
+
+log = logging.getLogger('heat-translator')
+
+
+PREDEF_FLAVORS = {
+    'm1.xlarge': {'mem_size': 16384, 'disk_size': 160, 'num_cpus': 8},
+    'm1.large': {'mem_size': 8192, 'disk_size': 80, 'num_cpus': 4},
+    'm1.medium': {'mem_size': 4096, 'disk_size': 40, 'num_cpus': 2},
+    'm1.small': {'mem_size': 2048, 'disk_size': 20, 'num_cpus': 1},
+    'm1.tiny': {'mem_size': 512, 'disk_size': 1, 'num_cpus': 1},
+    'm1.micro': {'mem_size': 128, 'disk_size': 0, 'num_cpus': 1},
+    'm1.nano': {'mem_size': 64, 'disk_size': 0, 'num_cpus': 1}
+}
+
+SESSION = None
+
+FLAVORS = {}
+
+
+def get_flavors():
+    global FLAVORS
+
+    if FLAVORS:
+        return FLAVORS
+
+    if SESSION is not None and client_available:
+        try:
+            client = novaclient.client.Client("2", session=SESSION)
+        except Exception as e:
+            # Handles any exception coming from openstack
+            log.warn(_('Choosing predefined flavors since received '
+                       'Openstack Exception: %s') % str(e))
+        else:
+            for flv in client.flavors.list(detailed=True):
+                FLAVORS[str(flv.name)] = {
+                    "mem_size": flv.ram,
+                    "disk_size": flv.disk,
+                    "num_cpus": flv.vcpus
+                }
+
+    if not FLAVORS:
+        FLAVORS = PREDEF_FLAVORS
+
+    return FLAVORS