import os
import json
+import re
+import tempfile
+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 rest_framework.parsers import MultiPartParser
class BaseServer(BaseHandler, APIView):
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)
- ret = []
- for k, v in plugins.items():
- item = {}
- item["vim_type"] = v.get("vim_type")
- item["versions"] = [k for k in v.get('versions', {})]
- ret.append(item)
- return Response(data=ret, status=status.HTTP_200_OK)
+ return Response(data=syscomm.getVIMTypes(), status=status.HTTP_200_OK)
class CheckCapacity(BaseServer):
for vim in body.get("VIMs", []):
url = request.get_full_path().replace(
"check_vim_capacity", "%s/capacity_check" % vim)
- resp = self.send(vim, url, newbody, "POST")
- if resp.status_code != status.HTTP_200_OK:
+ resp = self.send(vim, url, json.dumps(newbody), "POST")
+ if int(resp.status_code) != status.HTTP_200_OK:
continue
try:
- resp_body = json.loads(resp.body)
+ resp_body = json.loads(resp.content)
except ValueError:
continue
if not resp_body.get("result", False):
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):
+ try:
+ register_openers()
+ fileDict = dict(request.FILES.iterlists())
+ params = {}
+ for key in fileDict.keys():
+ fileObj = fileDict[key][0]
+ f = tempfile.NamedTemporaryFile(prefix="django_",
+ suffix=fileObj._name,
+ delete=False)
+ f.write(fileObj.file.read())
+ f.seek(fileObj.file.tell(), 0)
+ fileObj.file.close()
+ params[key] = open(f.name, 'rb')
+ 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)
+ finally:
+ for key in params:
+ fileRef = params[key]
+ if fileRef.closed is False:
+ fileRef.close()
+ os.remove(fileRef.name)
+ return resp