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=59c566beb85a2ffa51d54db8c83003642d670e98;hb=69a315e508da7716c0c01caeaafd3da5a3d20599;hpb=9e76f8d1b1417fc06eb510cf668aa313ab94eb3d diff --git a/multivimbroker/multivimbroker/forwarder/views.py b/multivimbroker/multivimbroker/forwarder/views.py index 59c566b..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. @@ -12,12 +13,20 @@ # See the License for the specific language governing permissions and # limitations under the License. +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): @@ -46,18 +55,21 @@ class Identity(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") + 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") + return self.send(vimid, request.get_full_path(), request.body, "POST", + headers=originHeaders(request)) class UnRegistry(BaseServer): @@ -65,48 +77,53 @@ class UnRegistry(BaseServer): def delete(self, request, vimid): return self.send(vimid, request.get_full_path(), request.body, - "DELETE") + "DELETE", headers=originHeaders(request)) class Extension(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)) class VIMTypes(BaseServer): def get(self, request): - # Fix here unless we have plugin registry - data = { - "vim_types": [ - { - "vim_type": "openstack", - "versions": [ - { - "version": "titanium_cloud", - "extra_info_hint": "" - }, - { - "version": "ocata", - "extra_info_hint": "" - } - ] - }, - { - "vim_type": "vmware", - "versions": [ - { - "version": "4.0", - "extra_info_hint": "" - } - ] - } - ] - } + return Response(data=syscomm.getVIMTypes(), status=status.HTTP_200_OK) - return Response(data=data, 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 @@ -114,28 +131,65 @@ 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): + + 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