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.
19 from rest_framework.views import APIView
20 from rest_framework.views import Response
21 from rest_framework.views import status
22 from multivimbroker.forwarder.base import BaseHandler
23 from multivimbroker.pub.utils.syscomm import originHeaders
26 class BaseServer(BaseHandler, APIView):
28 def get(self, request, vimid):
29 raise NotImplementedError()
31 def post(self, request, vimid):
32 raise NotImplementedError()
34 def put(self, request, vimid):
35 raise NotImplementedError()
37 def delete(self, request, vimid):
38 raise NotImplementedError()
40 def head(self, request, vimid):
41 raise NotImplementedError()
43 def patch(self, request, vimid):
44 raise NotImplementedError()
48 class Identity(BaseServer):
50 def get(self, request, vimid):
52 return self.send(vimid, request.get_full_path(), request.body, "GET",
53 headers=originHeaders(request))
55 def post(self, request, vimid):
57 return self.send(vimid, request.get_full_path(), request.body, "POST",
58 headers=originHeaders(request))
61 class Registry(BaseServer):
63 def post(self, request, vimid):
65 return self.send(vimid, request.get_full_path(), request.body, "POST",
66 headers=originHeaders(request))
69 class UnRegistry(BaseServer):
71 def delete(self, request, vimid):
73 return self.send(vimid, request.get_full_path(), request.body,
74 "DELETE", headers=originHeaders(request))
77 class Extension(BaseServer):
79 def get(self, request, vimid):
81 return self.send(vimid, request.get_full_path(), request.body, "GET",
82 headers=originHeaders(request))
85 class VIMTypes(BaseServer):
87 def get(self, request):
88 # Fix here unless we have plugin registry
89 json_file = os.path.join(os.path.dirname(__file__),
90 '../pub/config/provider-plugin.json')
91 with open(json_file, "r") as f:
92 plugins = json.load(f)
94 for k, v in plugins.items():
96 item["vim_type"] = v.get("vim_type")
97 item["versions"] = [k for k in v.get('versions', {})]
99 return Response(data=ret, status=status.HTTP_200_OK)
102 class CheckCapacity(BaseServer):
104 def post(self, request):
106 body = json.loads(request.body)
107 except json.JSONDecodeError as e:
109 data={'error': 'Invalidate request body %s.' % e},
110 status=status.HTTP_400_BAD_REQUEST)
114 "vCPU": body.get("vCPU", 0),
115 "Memory": body.get("Memory", 0),
116 "Storage": body.get("Storage", 0)
118 for vim in body.get("VIMs", []):
119 url = request.get_full_path().replace(
120 "check_vim_capacity", "%s/capacity_check" % vim)
121 resp = self.send(vim, url, newbody, "POST")
122 if resp.status_code != status.HTTP_200_OK:
125 resp_body = json.loads(resp.body)
126 except json.JSONDecodeError:
128 if not resp_body.get("result", False):
130 ret['VIMs'].append(vim)
131 return Response(data=ret, status=status.HTTP_200_OK)
135 class Forward(BaseServer):
137 def get(self, request, vimid):
139 return self.send(vimid, request.get_full_path(), request.body, "GET")
141 def post(self, request, vimid):
143 return self.send(vimid, request.get_full_path(), request.body, "POST",
146 def patch(self, request, vimid):
148 return self.send(vimid, request.get_full_path(), request.body, "PATCH",
151 def delete(self, request, vimid):
153 return self.send(vimid, request.get_full_path(), request.body,
154 "DELETE", headers=None)
156 def head(self, request, vimid):
158 return self.send(vimid, request.get_full_path(), request.body, "HEAD")
160 def put(self, request, vimid):
162 return self.send(vimid, request.get_full_path(), request.body, "PUT",