Merge "Added V0 Registry API"

[multicloud/azure.git] / azure / multicloud_azure / swagger / views / flavor / views.py

diff --git a/azure/multicloud_azure/swagger/views/flavor/views.py b/azure/multicloud_azure/swagger/views/flavor/views.py
new file mode 100644 (file)
index 0000000..c4eb060
--- /dev/null
+++ b/azure/multicloud_azure/swagger/views/flavor/views.py
@@ -0,0 +1,98 @@
+# Copyright (c) 2018 Amdocs
+#
+# 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.
+
+import json
+
+from rest_framework import status
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from multicloud_azure.pub.msapi import extsys
+from multicloud_azure.pub.vim.vimapi.compute import OperateFlavors
+from multicloud_azure.swagger import compute_utils
+from multicloud_azure.pub.exceptions import VimDriverAzureException
+
+
+class FlavorsView(APIView):
+
+    def post(self, request, vimid):
+        try:
+            create_req = json.loads(request.body)
+        except Exception as e:
+            return Response(data={'error': 'Fail to decode request body.'},
+                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+        try:
+            vim_info = extsys.get_vim_by_id(vimid)
+        except VimDriverAzureException as e:
+            return Response(data={'error': str(e)}, status=e.status_code)
+
+        data = {'subscription_id': vim_info['cloud_extra_info'],
+                'username': vim_info['username'],
+                'password': vim_info['password'],
+                'tenant_id': vim_info['default_tenant'],
+                'region_id': vim_info['cloud_region']}
+        rsp = {'vimId': vim_info['cloud_extra_info'],
+               'vimName': vim_info['name']}
+        flavor_name = create_req.get('name', None)
+        flavor_id = create_req.get('id', None)
+        flavors_op = OperateFlavors.OperateFlavors()
+        try:
+            target = flavor_id or flavor_name
+            flavor = flavors_op.find_flavor(data, target)
+            if flavor:
+                flavor, extra_specs = flavors_op.get_flavor(
+                    data, flavor.id)
+                rsp['returnCode'] = 0
+            else:
+                rsp['returnCode'] = 1
+                flavor, extra_specs = flavors_op.create_flavor(
+                    data, create_req)
+            flavor_dict = compute_utils.flavor_formatter(flavor, extra_specs)
+        except Exception as e:
+            if hasattr(e, "http_status"):
+                return Response(data={'error': str(e)}, status=e.http_status)
+            else:
+                return Response(data={'error': str(e)},
+                                status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        rsp.update(flavor_dict)
+        return Response(data=rsp, status=status.HTTP_200_OK)
+
+    def get(self, request, vimid):
+        try:
+            vim_info = extsys.get_vim_by_id(vimid)
+        except VimDriverAzureException as e:
+            return Response(data={'error': str(e)}, status=e.status_code)
+
+        data = {'subscription_id': vim_info['cloud_extra_info'],
+                'username': vim_info['username'],
+                'password': vim_info['password'],
+                'tenant_id': vim_info['default_tenant'],
+                'region_id': vim_info['cloud_region_id']}
+        query = dict(request.query_params)
+        flavors_op = OperateFlavors.OperateFlavors()
+        try:
+            flavors_result = flavors_op.list_flavors(data, **query)
+            flavors_dict = [compute_utils.flavor_formatter(flavor, extra)
+                            for flavor, extra in flavors_result]
+        except Exception as e:
+            if hasattr(e, "http_status"):
+                return Response(data={'error': str(e)}, status=e.http_status)
+            else:
+                return Response(data={'error': str(e)},
+                                status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+        rsp = {'vimId': vim_info['cloud_extra_info'],
+               'vimName': vim_info['name'],
+               'flavors': flavors_dict}
+
+        return Response(data=rsp, status=status.HTTP_200_OK)