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 poster.encode import multipart_encode
21 from poster.streaminghttp import register_openers
23 from rest_framework.views import APIView
24 from rest_framework.views import Response
25 from rest_framework.views import status
26 from multivimbroker.forwarder.base import BaseHandler
27 from multivimbroker.pub.utils.syscomm import originHeaders
28 from multivimbroker.pub.utils import syscomm
29 from rest_framework.parsers import MultiPartParser
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))
76 class UnRegistry(BaseServer):
78 def delete(self, request, vimid):
80 return self.send(vimid, request.get_full_path(), request.body,
81 "DELETE", headers=originHeaders(request))
84 class Extension(BaseServer):
86 def get(self, request, vimid):
88 return self.send(vimid, request.get_full_path(), request.body, "GET",
89 headers=originHeaders(request))
92 class VIMTypes(BaseServer):
94 def get(self, request):
95 return Response(data=syscomm.getVIMTypes(), status=status.HTTP_200_OK)
98 class CheckCapacity(BaseServer):
100 def post(self, request):
102 body = json.loads(request.body)
103 except ValueError as e:
105 data={'error': 'Invalidate request body %s.' % e},
106 status=status.HTTP_400_BAD_REQUEST)
110 "vCPU": body.get("vCPU", 0),
111 "Memory": body.get("Memory", 0),
112 "Storage": body.get("Storage", 0)
114 for vim in body.get("VIMs", []):
115 url = request.get_full_path().replace(
116 "check_vim_capacity", "%s/capacity_check" % vim)
117 resp = self.send(vim, url, json.dumps(newbody), "POST")
118 if int(resp.status_code) != status.HTTP_200_OK:
121 resp_body = json.loads(resp.content)
124 if not resp_body.get("result", False):
126 ret['VIMs'].append(vim)
127 return Response(data=ret, status=status.HTTP_200_OK)
131 class Forward(BaseServer):
133 def get(self, request, vimid):
135 return self.send(vimid, request.get_full_path(), request.body, "GET",
136 headers=originHeaders(request))
138 def post(self, request, vimid):
140 return self.send(vimid, request.get_full_path(), request.body, "POST",
141 headers=originHeaders(request))
143 def patch(self, request, vimid):
145 return self.send(vimid, request.get_full_path(), request.body, "PATCH",
146 headers=originHeaders(request))
148 def delete(self, request, vimid):
150 return self.send(vimid, request.get_full_path(), request.body,
151 "DELETE", headers=originHeaders(request))
153 def head(self, request, vimid):
155 return self.send(vimid, request.get_full_path(), request.body, "HEAD",
156 headers=originHeaders(request))
158 def put(self, request, vimid):
160 return self.send(vimid, request.get_full_path(), request.body, "PUT",
161 headers=originHeaders(request))
165 class MultiPartView(BaseServer):
167 parser_classes = (MultiPartParser, )
169 def post(self, request, vimid):
172 fileDict = dict(request.FILES.iterlists())
174 for key in fileDict.keys():
175 fileObj = fileDict[key][0]
176 f = tempfile.NamedTemporaryFile(prefix="django_",
177 suffix=fileObj._name,
179 f.write(fileObj.file.read())
180 f.seek(fileObj.file.tell(), 0)
182 params[key] = open(f.name, 'rb')
183 datagen, headers = multipart_encode(params)
184 regex = re.compile('^HTTP_')
185 for key, value in request.META.iteritems():
186 if key.startswith("HTTP_"):
187 headers[regex.sub('', key).replace('_', '-')] = value
188 resp = self.send(vimid, request.path, datagen, "POST",
192 fileRef = params[key]
193 if fileRef.closed is False:
195 os.remove(fileRef.name)
201 class APIv1Identity(Identity):
203 def get(self, request, cloud_owner, cloud_region_id):
204 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
205 return super(APIv1Identity, self).get(request, vimid)
207 def post(self, request, cloud_owner, cloud_region_id):
208 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
209 return super(APIv1Identity, self).post(request, vimid)
212 class APIv1Registry(Registry):
214 def post(self, request, cloud_owner, cloud_region_id):
215 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
216 return super(APIv1Registry, self).post(request, vimid)
219 class APIv1UnRegistry(UnRegistry):
221 def delete(self, request, cloud_owner, cloud_region_id):
222 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
223 return super(APIv1UnRegistry, self).delete(request, vimid)
226 class APIv1Extension(Extension):
228 def get(self, request, cloud_owner, cloud_region_id):
229 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
230 return super(APIv1Extension, self).get(request, vimid)
233 class APIv1VIMTypes(VIMTypes):
235 def get(self, request):
236 return super(APIv1VIMTypes, self).get(request)
239 class APIv1CheckCapacity(CheckCapacity):
241 def post(self, request):
243 body = json.loads(request.body)
244 except ValueError as e:
246 data={'error': 'Invalidate request body %s.' % e},
247 status=status.HTTP_400_BAD_REQUEST)
251 "vCPU": body.get("vCPU", 0),
252 "Memory": body.get("Memory", 0),
253 "Storage": body.get("Storage", 0)
255 for vim in body.get("VIMs", []):
256 cloud_owner = vim["cloud-owner"]
257 cloud_region_id = vim["cloud-region-id"]
258 url = request.get_full_path().replace(
259 "check_vim_capacity", "%s/%s/capacity_check" %
260 (cloud_owner, cloud_region_id))
261 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
262 resp = self.send(vimid, url, json.dumps(newbody), "POST")
263 if int(resp.status_code) != status.HTTP_200_OK:
266 resp_body = json.loads(resp.content)
269 if not resp_body.get("result", False):
271 ret['VIMs'].append(vim)
272 return Response(data=ret, status=status.HTTP_200_OK)
276 class APIv1Forward(Forward):
278 def get(self, request, cloud_owner, cloud_region_id):
279 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
280 return super(APIv1Forward, self).get(request, vimid)
282 def post(self, request, cloud_owner, cloud_region_id):
283 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
284 return super(APIv1Forward, self).post(request, vimid)
286 def patch(self, request, cloud_owner, cloud_region_id):
287 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
288 return super(APIv1Forward, self).patch(request, vimid)
290 def delete(self, request, cloud_owner, cloud_region_id):
291 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
292 return super(APIv1Forward, self).delete(request, vimid)
294 def head(self, request, cloud_owner, cloud_region_id):
295 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
296 return super(APIv1Forward, self).head(request, vimid)
298 def put(self, request, cloud_owner, cloud_region_id):
299 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
300 return super(APIv1Forward, self).put(request, vimid)
304 class APIv1MultiPartView(MultiPartView):
306 def post(self, request, cloud_owner, cloud_region_id):
308 vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id)
309 return super(APIv1MultiPartView, self).post(request, vimid)