# 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.
# 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 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 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))
+
+
+class UnRegistry(BaseServer):
+
+ def delete(self, request, vimid):
- return self.send(vimid,request.get_full_path(),request.body,"GET")
+ return self.send(vimid, request.get_full_path(), request.body,
+ "DELETE", headers=originHeaders(request))
- def post(self,request,vimid):
- 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",
+ 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")
+ 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")
+
+ 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):
+ 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