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=a5dd22c8af6fa796903dc3aeacb0426e376e5bb1;hb=d331bc744dc365d24818481a3ca7118ba1f05b29;hpb=63bf19e9be7fe29fc691bef4088ec544444a4a19 diff --git a/multivimbroker/multivimbroker/forwarder/views.py b/multivimbroker/multivimbroker/forwarder/views.py index a5dd22c..898acc5 100644 --- a/multivimbroker/multivimbroker/forwarder/views.py +++ b/multivimbroker/multivimbroker/forwarder/views.py @@ -13,11 +13,13 @@ # See the License for the specific language governing permissions and # limitations under the License. -import os +# import os import json -import tempfile -from poster.encode import multipart_encode -from poster.streaminghttp import register_openers +# 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 @@ -25,7 +27,7 @@ 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 +from multivimbroker.pub.msapi import extsys class BaseServer(BaseHandler, APIView): @@ -70,6 +72,11 @@ class Registry(BaseServer): 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): @@ -112,11 +119,12 @@ 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", + headers=originHeaders(request)) + 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): @@ -130,59 +138,201 @@ class Forward(BaseServer): def get(self, request, vimid): - return self.send(vimid, request.get_full_path(), request.body, "GET") + 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=None) + headers=originHeaders(request)) def patch(self, request, vimid): return self.send(vimid, request.get_full_path(), request.body, "PATCH", - headers=None) + headers=originHeaders(request)) def delete(self, request, vimid): return self.send(vimid, request.get_full_path(), request.body, - "DELETE", headers=None) + "DELETE", headers=originHeaders(request)) def head(self, request, vimid): - return self.send(vimid, request.get_full_path(), request.body, "HEAD") + 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=None) + headers=originHeaders(request)) -# Multipart view -class MultiPartView(BaseServer): +# API v1 +# proxy handler +class APIv1Identity(Identity): - parser_classes = (MultiPartParser, ) + 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, 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: - 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) - resp = self.send(vimid, request.path, datagen, "POST", + 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)) - finally: - for key in params: - fileRef = params[key] - if fileRef.closed is False: - fileRef.close() - os.remove(fileRef.name) - return resp + + 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): + + 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, cloud_owner, cloud_region_id): + vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id) + return super(APIv1Forward, self).post(request, vimid) + + 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 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) + + 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 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) + + +class APIv1InfraWorkload(BaseServer): + + 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 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)) + + 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