X-Git-Url: https://gerrit.onap.org/r/gitweb?p=multicloud%2Fframework.git;a=blobdiff_plain;f=multivimbroker%2Fmultivimbroker%2Fforwarder%2Fviews.py;h=1e57d0ebb75a3e35e2a9f84388f5578bf2287612;hp=e1013c71f22011a28181f6d20b0a256fb1ee596c;hb=69a315e508da7716c0c01caeaafd3da5a3d20599;hpb=53ef35bf7af5bf1eec8d8e3908faace5ad2193e4 diff --git a/multivimbroker/multivimbroker/forwarder/views.py b/multivimbroker/multivimbroker/forwarder/views.py index e1013c7..1e57d0e 100644 --- a/multivimbroker/multivimbroker/forwarder/views.py +++ b/multivimbroker/multivimbroker/forwarder/views.py @@ -1,4 +1,5 @@ # 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. @@ -11,42 +12,184 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -import logging + +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): + + def get(self, request, vimid): + raise NotImplementedError() + + def post(self, request, vimid): + raise NotImplementedError() + + def put(self, request, vimid): + raise NotImplementedError() + + def delete(self, request, vimid): + raise NotImplementedError() + + def head(self, request, vimid): + raise NotImplementedError() + + def patch(self, request, vimid): + raise NotImplementedError() + + +# proxy handler +class Identity(BaseServer): + + 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, + "DELETE", headers=originHeaders(request)) + + +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): + + 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)) + + def patch(self, request, vimid): + + return self.send(vimid, request.get_full_path(), request.body, "PATCH", + headers=originHeaders(request)) + + def delete(self, request, vimid): + + return self.send(vimid, request.get_full_path(), request.body, + "DELETE", headers=originHeaders(request)) + + def head(self, request, vimid): + + 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=originHeaders(request)) + + +# Multipart view +class MultiPartView(BaseServer): + + parser_classes = (MultiPartParser, ) -from django.http import HttpResponse -from django.views.decorators.csrf import csrf_exempt - -from rest_framework import status - -from multivimbroker.pub.utils.restcall import req_by_msb -from multivimbroker.pub.msapi.extsys import get_vim_by_id - -logger = logging.getLogger(__name__) - - -@csrf_exempt -def route(request, vimid=''): - """ get vim info from vimid from local cache first - and then to ESR if cache miss - """ - content = "" - status_code = status.HTTP_200_OK - try: - vim = get_vim_by_id(vimid) - - # if vim type is openstack, use latest "newton" version as default - if vim["type"] == "openstack": - vim["type"] = "multivim-newton" - - route_uri = re.sub('multivim', vim["type"], request.get_full_path()) - retcode, content, status_code = \ - req_by_msb(route_uri, request.method, request.body) - if retcode != 0: - # Execptions are handled within req_by_msb - logger.error("Status code is %s, detail is %s.", - status_code, content) - except Exception as e: - content = e - status_code = status.HTTP_500_INTERNAL_SERVER_ERROR - return HttpResponse(content, status_code) + 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