X-Git-Url: https://gerrit.onap.org/r/gitweb?p=multicloud%2Fframework.git;a=blobdiff_plain;f=multivimbroker%2Fmultivimbroker%2Fforwarder%2Fviews.py;h=898acc556501dab7aaf2df489ee76d9a22f8b3a8;hp=82759e56b21af44d85cae415d2c806fe95e1e85c;hb=d331bc744dc365d24818481a3ca7118ba1f05b29;hpb=0aeb8d832d2e8142083edef1d2dee2f6fc4f1ddc

diff --git a/multivimbroker/multivimbroker/forwarder/views.py b/multivimbroker/multivimbroker/forwarder/views.py
index 82759e5..898acc5 100644
--- a/multivimbroker/multivimbroker/forwarder/views.py
+++ b/multivimbroker/multivimbroker/forwarder/views.py
@@ -1,4 +1,5 @@
 # Copyright 2017 Wind River Systems, Inc.
+# Copyright (c) 2017-2018 VMware, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,68 +13,326 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from rest_framework.views import  APIView
+# import os
+import json
+# import re
+# import tempfile
+# from django.core.files.uploadedfile import InMemoryUploadedFile
+# from poster.encode import multipart_encode
+# from poster.streaminghttp import register_openers
+
+from rest_framework.views import APIView
+from rest_framework.views import Response
+from rest_framework.views import status
 from multivimbroker.forwarder.base import BaseHandler
+from multivimbroker.pub.utils.syscomm import originHeaders
+from multivimbroker.pub.utils import syscomm
+from multivimbroker.pub.msapi import extsys
 
-#
-class BaseServer(BaseHandler,APIView):
 
+class BaseServer(BaseHandler, APIView):
 
-    def get(self,request,vimid):
+    def get(self, request, vimid):
         raise NotImplementedError()
 
-    def post(self,request,vimid):
+    def post(self, request, vimid):
         raise NotImplementedError()
 
-    def put(self,request,vimid):
+    def put(self, request, vimid):
         raise NotImplementedError()
 
-    def delete(self,request,vimid):
+    def delete(self, request, vimid):
         raise NotImplementedError()
 
-    def head(self,request,vimid):
+    def head(self, request, vimid):
         raise NotImplementedError()
 
-    def patch(self,request,vimid):
+    def patch(self, request, vimid):
         raise NotImplementedError()
 
 
-#  vio proxy handler
+# proxy handler
 class Identity(BaseServer):
 
-    def get(self,request,vimid):
+    def get(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "GET",
+                         headers=originHeaders(request))
+
+    def post(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "POST",
+                         headers=originHeaders(request))
+
+
+class Registry(BaseServer):
+
+    def post(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "POST",
+                         headers=originHeaders(request))
+
+    def get(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "GET",
+                         headers=originHeaders(request))
+
+
+class UnRegistry(BaseServer):
+
+    def delete(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body,
+                         "DELETE", headers=originHeaders(request))
 
-        return self.send(vimid,request.get_full_path(),request.body,"GET")
 
-    def post(self,request,vimid):
+class Extension(BaseServer):
 
-        return self.send(vimid,request.get_full_path(),request.body,"POST")
+    def get(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "GET",
+                         headers=originHeaders(request))
+
+
+class VIMTypes(BaseServer):
+
+    def get(self, request):
+        return Response(data=syscomm.getVIMTypes(), status=status.HTTP_200_OK)
+
+
+class CheckCapacity(BaseServer):
+
+    def post(self, request):
+        try:
+            body = json.loads(request.body)
+        except ValueError as e:
+            return Response(
+                data={'error': 'Invalidate request body %s.' % e},
+                status=status.HTTP_400_BAD_REQUEST)
+
+        ret = {"VIMs": []}
+        newbody = {
+            "vCPU": body.get("vCPU", 0),
+            "Memory": body.get("Memory", 0),
+            "Storage": body.get("Storage", 0)
+        }
+        for vim in body.get("VIMs", []):
+            url = request.get_full_path().replace(
+                "check_vim_capacity", "%s/capacity_check" % vim)
+            resp = self.send(vim, url, json.dumps(newbody), "POST",
+                             headers=originHeaders(request))
+            if int(resp.status_code) != status.HTTP_200_OK:
+                continue
+            try:
+                resp_body = json.loads(resp.content)
+            except ValueError:
+                continue
+            if not resp_body.get("result", False):
+                continue
+            ret['VIMs'].append(vim)
+        return Response(data=ret, status=status.HTTP_200_OK)
 
 
 # forward  handler
 class Forward(BaseServer):
 
-    def get(self,request,vimid):
+    def get(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "GET",
+                         headers=originHeaders(request))
+
+    def post(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "POST",
+                         headers=originHeaders(request))
+
+    def patch(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "PATCH",
+                         headers=originHeaders(request))
+
+    def delete(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body,
+                         "DELETE", headers=originHeaders(request))
+
+    def head(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "HEAD",
+                         headers=originHeaders(request))
+
+    def put(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "PUT",
+                         headers=originHeaders(request))
+
+
+# API v1
+# proxy handler
+class APIv1Identity(Identity):
+
+    def get(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Identity, self).get(request, vimid)
+
+    def post(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Identity, self).post(request, vimid)
+
+
+class APIv1Registry(Registry):
+
+    def get(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Registry, self).get(request, vimid)
+
+    def post(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Registry, self).post(request, vimid)
+
+
+class APIv1UnRegistry(UnRegistry):
+
+    def delete(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1UnRegistry, self).delete(request, vimid)
+
+
+class APIv1Extension(Extension):
+
+    def get(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Extension, self).get(request, vimid)
+
+
+class APIv1VIMTypes(VIMTypes):
+
+    def get(self, request):
+        return super(APIv1VIMTypes, self).get(request)
+
+
+class APIv1CheckCapacity(CheckCapacity):
+
+    def post(self, request):
+        try:
+            body = json.loads(request.body)
+        except ValueError as e:
+            return Response(
+                data={'error': 'Invalidate request body %s.' % e},
+                status=status.HTTP_400_BAD_REQUEST)
+
+        ret = {"VIMs": []}
+        newbody = {
+            "vCPU": body.get("vCPU", 0),
+            "Memory": body.get("Memory", 0),
+            "Storage": body.get("Storage", 0)
+        }
+        for vim in body.get("VIMs", []):
+            cloud_owner = vim["cloud-owner"]
+            cloud_region_id = vim["cloud-region-id"]
+            url = request.get_full_path().replace(
+                "check_vim_capacity", "%s/%s/capacity_check" %
+                                      (cloud_owner, cloud_region_id))
+            vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+            resp = self.send(vimid, url, json.dumps(newbody), "POST",
+                             headers=originHeaders(request))
+
+            if int(resp.status_code) != status.HTTP_200_OK:
+                continue
+
+            try:
+                resp_body = json.loads(resp.content)
+            except ValueError:
+                continue
+            if not resp_body.get("result", False):
+                continue
+            azCapInfo = resp_body.get("AZs", [])
+            vim["AZs"] = azCapInfo
+            ret['VIMs'].append(vim)
+        return Response(data=ret, status=status.HTTP_200_OK)
+
+
+# forward  handler
+class APIv1Forward(Forward):
 
-        return self.send(vimid,request.get_full_path(),request.body,"GET")
+    def get(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Forward, self).get(request, vimid)
 
-    def post(self,request,vimid):
+    def post(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Forward, self).post(request, vimid)
 
-        return self.send(vimid,request.get_full_path(),request.body,"POST",headers=None)
+    def patch(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Forward, self).patch(request, vimid)
 
-    def patch(self,request,vimid):
+    def delete(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Forward, self).delete(request, vimid)
 
-        return self.send(vimid,request.get_full_path(),request.body,"PATCH",headers=None)
+    def head(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Forward, self).head(request, vimid)
 
-    def delete(self,request,vimid):
+    def put(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return super(APIv1Forward, self).put(request, vimid)
 
-        return self.send(vimid,request.get_full_path(),request.body,"DELETE",headers=None)
 
-    def head(self,request,vimid):
+class APIv1InfraWorkload(BaseServer):
 
-        return self.send(vimid,request.get_full_path(),request.body,"HEAD")
+    def post(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        content_type = request.META.get('CONTENT_TYPE', 'application/json')
+        if content_type.startswith("multipart"):
+            return self.__process_multipart(request, vimid)
+        return self.send(vimid, request.get_full_path(), request.body, "POST",
+                         headers=originHeaders(request))
 
-    def put(self,request,vimid):
+    def get(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return self.send(vimid, request.get_full_path(), request.body, "GET",
+                         headers=originHeaders(request))
 
-        return self.send(vimid,request.get_full_path(),request.body,"PUT",headers=None)
+    def delete(self, request, cloud_owner, cloud_region_id):
+        vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
+        return self.send(vimid, request.get_full_path(), request.body,
+                         "DELETE", headers=originHeaders(request))
 
+    def __process_multipart(self, request, vimid):
+        return Response(
+            data={'error': 'multipart API is not supported yet'},
+            status=status.HTTP_400_BAD_REQUEST)
+#        try:
+#            API is depreciated due to poster not available in py3
+#            register_openers()
+#            dataDict = dict(request.data.iterlists())
+#            params = {}
+#            for key in dataDict.keys():
+#                dataObj = dataDict[key][0]
+#                if isinstance(dataObj, InMemoryUploadedFile):
+#                    f = tempfile.NamedTemporaryFile(prefix="django_",
+#                                                    suffix=dataObj._name,
+#                                                    delete=False)
+#                    f.write(dataObj.file.read())
+#                    f.seek(dataObj.file.tell(), 0)
+#                    dataObj.file.close()
+#                    params[key] = open(f.name, 'rb')
+#                else:
+#                    params[key] = dataObj
+#             datagen, headers = multipart_encode(params)
+#             regex = re.compile('^HTTP_')
+#             for key, value in request.META.iteritems():
+#                if key.startswith("HTTP_"):
+#                    headers[regex.sub('', key).replace('_', '-')] = value
+#             resp = self.send(vimid, request.path, datagen, "POST",
+#                             headers=headers, multipart=True)
+#        finally:
+#             for key in params or {}:
+#                refobj = params[key]
+#                if type(refobj) is not unicode:
+#                    if refobj.closed is False:
+#                        print (refobj.close())
+#                    os.remove(refobj.name)
+#        return resp