1 # Copyright 2017 Wind River Systems, Inc.
2 # Copyright (c) 2017-2018 VMware, Inc.
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
20 # from django.core.files.uploadedfile import InMemoryUploadedFile
21 # from poster.encode import multipart_encode
22 # from poster.streaminghttp import register_openers
24 from rest_framework.views import APIView
25 from rest_framework.views import Response
26 from rest_framework.views import status
27 from multivimbroker.forwarder.base import BaseHandler
28 from multivimbroker.pub.utils.syscomm import originHeaders
29 from multivimbroker.pub.utils import syscomm
30 from multivimbroker.pub.msapi import extsys
33 class BaseServer(BaseHandler, APIView):
35 def get(self, request, vimid):
36 raise NotImplementedError()
38 def post(self, request, vimid):
39 raise NotImplementedError()
41 def put(self, request, vimid):
42 raise NotImplementedError()
44 def delete(self, request, vimid):
45 raise NotImplementedError()
47 def head(self, request, vimid):
48 raise NotImplementedError()
50 def patch(self, request, vimid):
51 raise NotImplementedError()
55 class Identity(BaseServer):
57 def get(self, request, vimid):
59 return self.send(vimid, request.get_full_path(), request.body, "GET",
60 headers=originHeaders(request))
62 def post(self, request, vimid):
64 return self.send(vimid, request.get_full_path(), request.body, "POST",
65 headers=originHeaders(request))
68 class Registry(BaseServer):
70 def post(self, request, vimid):
72 return self.send(vimid, request.get_full_path(), request.body, "POST",
73 headers=originHeaders(request))
75 def get(self, request, vimid):
77 return self.send(vimid, request.get_full_path(), request.body, "GET",
78 headers=originHeaders(request))
81 class UnRegistry(BaseServer):
83 def delete(self, request, vimid):
85 return self.send(vimid, request.get_full_path(), request.body,
86 "DELETE", headers=originHeaders(request))
89 class Extension(BaseServer):
91 def get(self, request, vimid):
93 return self.send(vimid, request.get_full_path(), request.body, "GET",
94 headers=originHeaders(request))
97 class VIMTypes(BaseServer):
99 def get(self, request):
100 return Response(data=syscomm.getVIMTypes(), status=status.HTTP_200_OK)
103 class CheckCapacity(BaseServer):
105 def post(self, request):
107 body = json.loads(request.body)
108 except ValueError as e:
110 data={'error': 'Invalidate request body %s.' % e},
111 status=status.HTTP_400_BAD_REQUEST)
115 "vCPU": body.get("vCPU", 0),
116 "Memory": body.get("Memory", 0),
117 "Storage": body.get("Storage", 0)
119 for vim in body.get("VIMs", []):
120 url = request.get_full_path().replace(
121 "check_vim_capacity", "%s/capacity_check" % vim)
122 resp = self.send(vim, url, json.dumps(newbody), "POST",
123 headers=originHeaders(request))
124 if int(resp.status_code) != status.HTTP_200_OK:
127 resp_body = json.loads(resp.content)
130 if not resp_body.get("result", False):
132 ret['VIMs'].append(vim)
133 return Response(data=ret, status=status.HTTP_200_OK)
137 class Forward(BaseServer):
139 def get(self, request, vimid):
141 return self.send(vimid, request.get_full_path(), request.body, "GET",
142 headers=originHeaders(request))
144 def post(self, request, vimid):
146 return self.send(vimid, request.get_full_path(), request.body, "POST",
147 headers=originHeaders(request))
149 def patch(self, request, vimid):
151 return self.send(vimid, request.get_full_path(), request.body, "PATCH",
152 headers=originHeaders(request))
154 def delete(self, request, vimid):
156 return self.send(vimid, request.get_full_path(), request.body,
157 "DELETE", headers=originHeaders(request))
159 def head(self, request, vimid):
161 return self.send(vimid, request.get_full_path(), request.body, "HEAD",
162 headers=originHeaders(request))
164 def put(self, request, vimid):
166 return self.send(vimid, request.get_full_path(), request.body, "PUT",
167 headers=originHeaders(request))
172 class APIv1Identity(Identity):
174 def get(self, request, cloud_owner, cloud_region_id):
175 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
176 return super(APIv1Identity, self).get(request, vimid)
178 def post(self, request, cloud_owner, cloud_region_id):
179 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
180 return super(APIv1Identity, self).post(request, vimid)
183 class APIv1Registry(Registry):
185 def get(self, request, cloud_owner, cloud_region_id):
186 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
187 return super(APIv1Registry, self).get(request, vimid)
189 def post(self, request, cloud_owner, cloud_region_id):
190 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
191 return super(APIv1Registry, self).post(request, vimid)
194 class APIv1UnRegistry(UnRegistry):
196 def delete(self, request, cloud_owner, cloud_region_id):
197 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
198 return super(APIv1UnRegistry, self).delete(request, vimid)
201 class APIv1Extension(Extension):
203 def get(self, request, cloud_owner, cloud_region_id):
204 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
205 return super(APIv1Extension, self).get(request, vimid)
208 class APIv1VIMTypes(VIMTypes):
210 def get(self, request):
211 return super(APIv1VIMTypes, self).get(request)
214 class APIv1CheckCapacity(CheckCapacity):
216 def post(self, request):
218 body = json.loads(request.body)
219 except ValueError as e:
221 data={'error': 'Invalidate request body %s.' % e},
222 status=status.HTTP_400_BAD_REQUEST)
226 "vCPU": body.get("vCPU", 0),
227 "Memory": body.get("Memory", 0),
228 "Storage": body.get("Storage", 0)
230 for vim in body.get("VIMs", []):
231 cloud_owner = vim["cloud-owner"]
232 cloud_region_id = vim["cloud-region-id"]
233 url = request.get_full_path().replace(
234 "check_vim_capacity", "%s/%s/capacity_check" %
235 (cloud_owner, cloud_region_id))
236 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
237 resp = self.send(vimid, url, json.dumps(newbody), "POST",
238 headers=originHeaders(request))
240 if int(resp.status_code) != status.HTTP_200_OK:
244 resp_body = json.loads(resp.content)
247 if not resp_body.get("result", False):
249 azCapInfo = resp_body.get("AZs", [])
250 vim["AZs"] = azCapInfo
251 ret['VIMs'].append(vim)
252 return Response(data=ret, status=status.HTTP_200_OK)
256 class APIv1Forward(Forward):
258 def get(self, request, cloud_owner, cloud_region_id):
259 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
260 return super(APIv1Forward, self).get(request, vimid)
262 def post(self, request, cloud_owner, cloud_region_id):
263 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
264 return super(APIv1Forward, self).post(request, vimid)
266 def patch(self, request, cloud_owner, cloud_region_id):
267 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
268 return super(APIv1Forward, self).patch(request, vimid)
270 def delete(self, request, cloud_owner, cloud_region_id):
271 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
272 return super(APIv1Forward, self).delete(request, vimid)
274 def head(self, request, cloud_owner, cloud_region_id):
275 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
276 return super(APIv1Forward, self).head(request, vimid)
278 def put(self, request, cloud_owner, cloud_region_id):
279 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
280 return super(APIv1Forward, self).put(request, vimid)
283 class APIv1InfraWorkload(BaseServer):
285 def post(self, request, cloud_owner, cloud_region_id):
286 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
287 content_type = request.META.get('CONTENT_TYPE', 'application/json')
288 if content_type.startswith("multipart"):
289 return self.__process_multipart(request, vimid)
290 return self.send(vimid, request.get_full_path(), request.body, "POST",
291 headers=originHeaders(request))
293 def get(self, request, cloud_owner, cloud_region_id):
294 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
295 return self.send(vimid, request.get_full_path(), request.body, "GET",
296 headers=originHeaders(request))
298 def delete(self, request, cloud_owner, cloud_region_id):
299 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
300 return self.send(vimid, request.get_full_path(), request.body,
301 "DELETE", headers=originHeaders(request))
303 def __process_multipart(self, request, vimid):
305 data={'error': 'multipart API is not supported yet'},
306 status=status.HTTP_400_BAD_REQUEST)
308 # API is depreciated due to poster not available in py3
310 # dataDict = dict(request.data.iterlists())
312 # for key in dataDict.keys():
313 # dataObj = dataDict[key][0]
314 # if isinstance(dataObj, InMemoryUploadedFile):
315 # f = tempfile.NamedTemporaryFile(prefix="django_",
316 # suffix=dataObj._name,
318 # f.write(dataObj.file.read())
319 # f.seek(dataObj.file.tell(), 0)
320 # dataObj.file.close()
321 # params[key] = open(f.name, 'rb')
323 # params[key] = dataObj
324 # datagen, headers = multipart_encode(params)
325 # regex = re.compile('^HTTP_')
326 # for key, value in request.META.iteritems():
327 # if key.startswith("HTTP_"):
328 # headers[regex.sub('', key).replace('_', '-')] = value
329 # resp = self.send(vimid, request.path, datagen, "POST",
330 # headers=headers, multipart=True)
332 # for key in params or {}:
333 # refobj = params[key]
334 # if type(refobj) is not unicode:
335 # if refobj.closed is False:
336 # print (refobj.close())
337 # os.remove(refobj.name)