# See the License for the specific language governing permissions and
# limitations under the License.
-import os
+# import os
import json
+# 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
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 multivimbroker.pub.msapi import extsys
class BaseServer(BaseHandler, APIView):
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):
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
- 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)
- data = {"vim_types": plugins}
- return Response(data=data, status=status.HTTP_200_OK)
+ 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",
+ headers=originHeaders(request))
+ 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
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))
+
+
+# API v1
+# proxy handler
+class APIv1Identity(Identity):
+
+ 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, 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 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:
+ 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))
+
+ 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