1 files changed, 87 insertions, 0 deletions
diff --git a/azure/multicloud_azure/swagger/views/registry/views.py b/azure/multicloud_azure/swagger/views/registry/views.py
new file mode 100644
index 0000000..8464ce4
--- /dev/null
+++ b/azure/multicloud_azure/swagger/views/registry/views.py
@@ -0,0 +1,87 @@
+# 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 logging
+import json
+
+from rest_framework import status
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from multicloud_azure.pub.exceptions import VimDriverAzureException
+from multicloud_azure.pub.msapi import extsys
+from multicloud_azure.pub.utils.restcall import AAIClient
+from multicloud_azure.pub.vim.vimapi.compute import OperateFlavors
+
+
+logger = logging.getLogger(__name__)
+
+
+class Registry(APIView):
+    def _get_flavors(self, auth_info):
+        flavors_op = OperateFlavors.OperateFlavors()
+        try:
+            flavors = flavors_op.list_flavors(auth_info)
+        except Exception as e:
+            logger.exception("get flavors error %(e)s", {"e": e})
+            raise e
+
+        rsp = {"flavors": flavors}
+        return rsp
+
+    def post(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)
+        cloud_extra_info = json.loads(vim_info['cloud_extra_info'])
+        data = {
+            'subscription_id': cloud_extra_info['subscription_id'],
+            'username': vim_info['username'],
+            'password': vim_info['password'],
+            'tenant_id': vim_info['default_tenant'],
+            'region_id': vim_info['cloud-region-id']
+        }
+        rsp = {}
+        try:
+            logger.debug('Getting flavors')
+            flavors = self._get_flavors(data)
+            rsp.update(flavors)
+            # update A&AI
+            logger.debug('Put data into A&AI')
+            cloud_owner, cloud_region = extsys.split_vim_to_owner_region(
+                vimid)
+            aai_adapter = AAIClient(cloud_owner, cloud_region)
+            aai_adapter.update_vim(rsp)
+        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)
+
+        return Response(data="", status=status.HTTP_200_OK)
+
+
+class UnRegistry(APIView):
+
+    def delete(self, request, vimid):
+        try:
+            cloud_owner, cloud_region = extsys.split_vim_to_owner_region(
+                    vimid)
+            aai_adapter = AAIClient(cloud_owner, cloud_region)
+            aai_adapter.delete_vim()
+        except Exception as e:
+            return Response(data=e.message,
+                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        return Response(data="", status=status.HTTP_204_NO_CONTENT)