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 623aaa8..7575ffd 100644 (file)
--- a/multivimbroker/multivimbroker/forwarder/views.py
+++ b/multivimbroker/multivimbroker/forwarder/views.py
@@ -12,10 +12,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+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
 
 #
 
@@ -59,6 +66,9 @@ class Registry(BaseServer):
 
         return self.send(vimid, request.get_full_path(), request.body, "POST")
 
+
+class UnRegistry(BaseServer):
+
     def delete(self, request, vimid):
 
         return self.send(vimid, request.get_full_path(), request.body,
@@ -76,33 +86,11 @@ class VIMTypes(BaseServer):
 
     def get(self, request):
         # Fix here unless we have plugin registry
-        data = {
-            "vim_types": [
-                {
-                    "vim_type": "openstack",
-                    "versions": [
-                        {
-                            "version": "mitaka",
-                            "extra_info_hint": ""
-                        },
-                        {
-                            "version": "ocata",
-                            "extra_info_hint": ""
-                        }
-                    ]
-                },
-                {
-                    "vim_type": "vmware",
-                    "versions": [
-                        {
-                            "version": "4.0",
-                            "extra_info_hint": ""
-                        }
-                    ]
-                }
-            ]
-        }
-
+        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)
 
 
@@ -136,3 +124,24 @@ class Forward(BaseServer):
 
         return self.send(vimid, request.get_full_path(), request.body, "PUT",
                          headers=None)
+
+
+# Multipart view
+class MultiPartView(BaseServer):
+
+    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)