Support for Multipart has been added to Broker

[multicloud/framework.git] / multivimbroker / multivimbroker / forwarder / views.py

diff --git a/multivimbroker/multivimbroker/forwarder/views.py b/multivimbroker/multivimbroker/forwarder/views.py
index 82759e5..7575ffd 100644 (file)
--- a/multivimbroker/multivimbroker/forwarder/views.py
+++ b/multivimbroker/multivimbroker/forwarder/views.py
@@ -12,68 +12,136 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from rest_framework.views import  APIView
+import os
+import json
+from poster.encode import multipart_encode
+from poster.streaminghttp import register_openers
+import re
+
+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 rest_framework.parsers import MultiPartParser
 
 #
-class BaseServer(BaseHandler,APIView):
 
 
-    def get(self,request,vimid):
+class BaseServer(BaseHandler, APIView):
+
+    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")
+
+    def post(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "POST")
+
+
+class Registry(BaseServer):
+
+    def post(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "POST")
+
+
+class UnRegistry(BaseServer):
 
-        return self.send(vimid,request.get_full_path(),request.body,"GET")
+    def delete(self, request, vimid):
 
-    def post(self,request,vimid):
+        return self.send(vimid, request.get_full_path(), request.body,
+                         "DELETE")
 
-        return self.send(vimid,request.get_full_path(),request.body,"POST")
+
+class Extension(BaseServer):
+
+    def get(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "GET")
+
+
+class VIMTypes(BaseServer):
+
+    def get(self, request):
+        # Fix here unless we have plugin registry
+        json_file = os.path.join(os.path.dirname(__file__),
+                                 '../pub/config/provider-plugin.json')
+        with open(json_file, "r") as f:
+            plugins = json.load(f)
+        data = {"vim_types": plugins}
+        return Response(data=data, 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")
+
+    def post(self, request, vimid):
+
+        return self.send(vimid, request.get_full_path(), request.body, "POST",
+                         headers=None)
 
-        return self.send(vimid,request.get_full_path(),request.body,"GET")
+    def patch(self, request, vimid):
 
-    def post(self,request,vimid):
+        return self.send(vimid, request.get_full_path(), request.body, "PATCH",
+                         headers=None)
 
-        return self.send(vimid,request.get_full_path(),request.body,"POST",headers=None)
+    def delete(self, request, vimid):
 
-    def patch(self,request,vimid):
+        return self.send(vimid, request.get_full_path(), request.body,
+                         "DELETE", headers=None)
 
-        return self.send(vimid,request.get_full_path(),request.body,"PATCH",headers=None)
+    def head(self, request, vimid):
 
-    def delete(self,request,vimid):
+        return self.send(vimid, request.get_full_path(), request.body, "HEAD")
 
-        return self.send(vimid,request.get_full_path(),request.body,"DELETE",headers=None)
+    def put(self, request, vimid):
 
-    def head(self,request,vimid):
+        return self.send(vimid, request.get_full_path(), request.body, "PUT",
+                         headers=None)
 
-        return self.send(vimid,request.get_full_path(),request.body,"HEAD")
 
-    def put(self,request,vimid):
+# Multipart view
+class MultiPartView(BaseServer):
 
-        return self.send(vimid,request.get_full_path(),request.body,"PUT",headers=None)
+    parser_classes = (MultiPartParser, )
 
+    def post(self, request, vimid):
+        register_openers()
+        datagen, headers = multipart_encode(dict(request.data.iterlists()))
+        # will convert the datagen to be accepted by httplib2 body param
+        requestData = "".join(datagen)
+        # MultiPart parser store the header keys in request.META
+        # A custom header in request body(for ex: Cloud_Type) 
+        # will be transformed to HTTP_CLOUD_TYPE
+        regex = re.compile('^HTTP_')
+        for key, value in request.META.iteritems():
+            if key.startswith("HTTP_"):
+                headers[regex.sub('',key).replace('_','-')] = value
+        return self.send(vimid, request.path, requestData, "POST",
+                         headers=headers)