Optimize vfc-ztevnfmdriver scale/heal code
[vfc/nfvo/driver/vnfm/svnfm.git] / zte / vmanager / driver / interfaces / views.py
1 # Copyright 2016-2017 ZTE Corporation.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 import inspect
16 import json
17 import logging
18 import traceback
19 import os
20
21 from rest_framework.decorators import api_view
22 from rest_framework.response import Response
23
24 from driver.pub.utils import restcall
25 from driver.pub.utils.restcall import req_by_msb
26 from driver.pub.config.config import VNF_FTP
27
28 logger = logging.getLogger(__name__)
29
30
31 def load_json_file(file_name):
32     json_file = os.path.join(os.path.dirname(__file__), "data/" + file_name)
33     f = open(json_file)
34     json_data = json.JSONDecoder().decode(f.read())
35     f.close()
36     return json_data
37
38
39 def fun_name():
40     return "=================%s==================" % inspect.stack()[1][3]
41
42
43 def ignorcase_get(args, key):
44     if not key:
45         return ""
46     if not args:
47         return ""
48     if key in args:
49         return args[key]
50     for old_key in args:
51         if old_key.upper() == key.upper():
52             return args[old_key]
53     return ""
54
55
56 def mapping_conv(keyword_map, rest_return):
57     resp_data = {}
58     for param in keyword_map:
59         if keyword_map[param]:
60             resp_data[keyword_map[param]] = ignorcase_get(rest_return, param)
61     return resp_data
62
63
64 # Query vnfm_info from nslcm
65 def get_vnfminfo_from_nslcm(vnfmid):
66     ret = req_by_msb("api/nslcm/v1/vnfms/%s" % vnfmid, "GET")
67     return ret
68
69
70 # Query vnfd_info from nslcm
71 def vnfd_get(vnfpackageid):
72     ret = req_by_msb("api/nslcm/v1/vnfpackage/%s" % vnfpackageid, "GET")
73     return ret
74
75
76 # Query vnfpackage_info from nslcm
77 def vnfpackage_get(csarid):
78     ret = req_by_msb("api/nslcm/v1/vnfpackage/%s" % csarid, "GET")
79     return ret
80
81
82 @api_view(http_method_names=['POST'])
83 def instantiate_vnf(request, *args, **kwargs):
84     try:
85         logger.debug("[%s] request.data=%s", fun_name(), request.data)
86         vnfm_id = ignorcase_get(kwargs, "vnfmid")
87         ret = get_vnfminfo_from_nslcm(vnfm_id)
88         if ret[0] != 0:
89             return Response(data={'error': ret[1]}, status=ret[2])
90         vnfm_info = json.JSONDecoder().decode(ret[1])
91         logger.debug("[%s] vnfm_info=%s", fun_name(), vnfm_info)
92         vnf_package_id = ignorcase_get(request.data, "vnfPackageId")
93         ret = vnfd_get(vnf_package_id)
94         if ret[0] != 0:
95             return Response(data={'error': ret[1]}, status=ret[2])
96         vnfd_info = json.JSONDecoder().decode(ret[1])
97         logger.debug("[%s] vnfd_info=%s", fun_name(), vnfd_info)
98         csar_id = ignorcase_get(vnfd_info, "csarId")
99         ret = vnfpackage_get(csar_id)
100         if ret[0] != 0:
101             return Response(data={'error': ret[1]}, status=ret[2])
102         vnf_package_info = json.JSONDecoder().decode(ret[1])
103         packageInfo = ignorcase_get(vnf_package_info, "packageInfo")
104         logger.debug("[%s] packageInfo=%s", fun_name(), packageInfo)
105         data = {}
106         data["NFVOID"] = 1
107         data["VNFMID"] = vnfm_id
108         vnfdModel = json.loads(ignorcase_get(packageInfo, "vnfdModel"))
109         metadata = ignorcase_get(vnfdModel, "metadata")
110         vnfd_name = ignorcase_get(metadata, "name")
111         # TODO  convert sdc vnf package to vnf vender package
112         inputs = []
113         if "SPGW" in vnfd_name.upper():
114             data["VNFD"] = VNF_FTP + "SPGW"
115             inputs = load_json_file("SPGW" + "_inputs.json")
116         elif "MME" in vnfd_name.upper():
117             data["VNFD"] = VNF_FTP + "MME"
118             inputs = load_json_file("MME" + "_inputs.json")
119         else:
120             data["VNFD"] = ignorcase_get(packageInfo, "downloadUri")
121
122         data["VNFURL"] = data["VNFD"]
123
124         data["extension"] = {}
125         for name, value in ignorcase_get(ignorcase_get(request.data, "additionalParam"), "inputs").items():
126             inputs.append({"name": name, "value": value})
127
128         data["extension"]["inputs"] = json.dumps(inputs)
129         additionalParam = ignorcase_get(request.data, "additionalParam")
130         data["extension"]["extVirtualLinks"] = ignorcase_get(additionalParam, "extVirtualLinks")
131         data["extension"]["vnfinstancename"] = ignorcase_get(request.data, "vnfInstanceName")
132         data["extension"]["vnfid"] = data["VNFD"]
133         data["extension"]["multivim"] = 0
134         logger.debug("[%s] call_req data=%s", fun_name(), data)
135
136         ret = restcall.call_req(
137             base_url=ignorcase_get(vnfm_info, "url"),
138             user=ignorcase_get(vnfm_info, "userName"),
139             passwd=ignorcase_get(vnfm_info, "password"),
140             auth_type=restcall.rest_no_auth,
141             resource="v1/vnfs",
142             method='post',
143             content=json.JSONEncoder().encode(data))
144
145         logger.debug("[%s] call_req ret=%s", fun_name(), ret)
146         if ret[0] != 0:
147             return Response(data={'error': ret[1]}, status=ret[2])
148         resp = json.JSONDecoder().decode(ret[1])
149         resp_data = {
150             "vnfInstanceId": ignorcase_get(resp, "VNFInstanceID"),
151             "jobId": ignorcase_get(resp, "JobId")
152         }
153         logger.info("[%s]resp_data=%s", fun_name(), resp_data)
154     except Exception as e:
155         logger.error("Error occurred when instantiating VNF")
156         raise e
157     return Response(data=resp_data, status=ret[2])
158
159
160 @api_view(http_method_names=['POST'])
161 def terminate_vnf(request, *args, **kwargs):
162     try:
163         logger.debug("[%s] request.data=%s", fun_name(), request.data)
164         vnfm_id = ignorcase_get(kwargs, "vnfmid")
165         ret = get_vnfminfo_from_nslcm(vnfm_id)
166         if ret[0] != 0:
167             return Response(data={'error': ret[1]}, status=ret[2])
168         vnfm_info = json.JSONDecoder().decode(ret[1])
169         logger.debug("[%s] vnfm_info=%s", fun_name(), vnfm_info)
170         ret = restcall.call_req(
171             base_url=ignorcase_get(vnfm_info, "url"),
172             user=ignorcase_get(vnfm_info, "userName"),
173             passwd=ignorcase_get(vnfm_info, "password"),
174             auth_type=restcall.rest_no_auth,
175             resource="v1/vnfs/%s" % (ignorcase_get(kwargs, "vnfInstanceID")),
176             method='delete',
177             content=json.JSONEncoder().encode(request.data))
178         if ret[0] != 0:
179             return Response(data={'error': ret[1]}, status=ret[2])
180         resp = json.JSONDecoder().decode(ret[1])
181         resp_data = {
182             "vnfInstanceId": ignorcase_get(resp, "VNFInstanceID"),
183             "jobId": ignorcase_get(resp, "JobId")
184         }
185         logger.debug("[%s]resp_data=%s", fun_name(), resp_data)
186     except Exception as e:
187         logger.error("Error occurred when terminating VNF")
188         raise e
189     return Response(data=resp_data, status=ret[2])
190
191
192 @api_view(http_method_names=['GET'])
193 def query_vnf(request, *args, **kwargs):
194     try:
195         logger.debug("[%s] request.data=%s", fun_name(), request.data)
196         vnfm_id = ignorcase_get(kwargs, "vnfmid")
197         ret = get_vnfminfo_from_nslcm(vnfm_id)
198         if ret[0] != 0:
199             return Response(data={'error': ret[1]}, status=ret[2])
200         vnfm_info = json.JSONDecoder().decode(ret[1])
201         logger.debug("[%s] vnfm_info=%s", fun_name(), vnfm_info)
202         ret = restcall.call_req(
203             base_url=ignorcase_get(vnfm_info, "url"),
204             user=ignorcase_get(vnfm_info, "userName"),
205             passwd=ignorcase_get(vnfm_info, "password"),
206             auth_type=restcall.rest_no_auth,
207             resource="v1/vnfs/%s" % (ignorcase_get(kwargs, "vnfInstanceID")),
208             method='get',
209             content=json.JSONEncoder().encode({}))
210         if ret[0] != 0:
211             return Response(data={'error': ret[1]}, status=ret[2])
212         resp = json.JSONDecoder().decode(ret[1])
213         vnf_status = ignorcase_get(resp, "vnfinstancestatus")
214         resp_data = {"vnfInfo": {"vnfStatus": vnf_status}}
215         logger.debug("[%s]resp_data=%s", fun_name(), resp_data)
216     except Exception as e:
217         logger.error("Error occurred when querying VNF information.")
218         raise e
219     return Response(data=resp_data, status=ret[2])
220
221
222 @api_view(http_method_names=['GET'])
223 def operation_status(request, *args, **kwargs):
224     data = {}
225     try:
226         logger.debug("[%s] request.data=%s", fun_name(), request.data)
227         vnfm_id = ignorcase_get(kwargs, "vnfmid")
228         ret = get_vnfminfo_from_nslcm(vnfm_id)
229         if ret[0] != 0:
230             return Response(data={'error': ret[1]}, status=ret[2])
231         vnfm_info = json.JSONDecoder().decode(ret[1])
232         logger.debug("[%s] vnfm_info=%s", fun_name(), vnfm_info)
233         operation_status_url = '/v1/jobs/{jobId}?NFVOID={nfvoId}&VNFMID={vnfmId}&ResponseID={responseId}'
234         ret = restcall.call_req(
235             base_url=ignorcase_get(vnfm_info, 'url'),
236             user=ignorcase_get(vnfm_info, 'userName'),
237             passwd=ignorcase_get(vnfm_info, 'password'),
238             auth_type=restcall.rest_no_auth,
239             resource=operation_status_url.format(jobId=ignorcase_get(kwargs, 'jobid'), nfvoId=1,
240                                                  vnfmId=ignorcase_get(kwargs, 'vnfmid'),
241                                                  responseId=ignorcase_get(request.GET, 'responseId')),
242             method='get',
243             content=json.JSONEncoder().encode(data))
244
245         if ret[0] != 0:
246             return Response(data={'error': ret[1]}, status=ret[2])
247         resp_data = json.JSONDecoder().decode(ret[1])
248         logger.info("[%s]resp_data=%s", fun_name(), resp_data)
249     except Exception as e:
250         logger.error("Error occurred when getting operation status information.")
251         raise e
252     return Response(data=resp_data, status=ret[2])
253
254
255 @api_view(http_method_names=['PUT'])
256 def grantvnf(request, *args, **kwargs):
257     logger.info("=====grantvnf=====")
258     try:
259         logger.info("req_data = %s", request.data)
260         grant_vnf_param_map = {
261             "VNFMID": "",
262             "NFVOID": "",
263             "VIMID": "",
264             "ExVIMIDList": "",
265             "ExVIMID": "",
266             "Tenant": "",
267             "VNFInstanceID": "vnfInstanceId",
268             "OperationRight": "",
269             "VMList": "",
270             "VMFlavor": "",
271             "VMNumber": ""
272         }
273         data = mapping_conv(grant_vnf_param_map, request.data)
274         data["vnfDescriptorId"] = ""
275         if ignorcase_get(request.data, "operationright") == 0:
276             data["lifecycleOperation"] = "Instantiate"
277             data["addresource"] = []
278             for vm in ignorcase_get(request.data, "vmlist"):
279                 for i in range(int(ignorcase_get(vm, "vmnumber"))):
280                     data["addresource"].append(
281                         {
282                             "type": "vdu",
283                             "resourceDefinitionId": i,
284                             "vdu": ignorcase_get(vm, "vmflavor"),
285                             "vimid": ignorcase_get(vm, "vimid"),
286                             "tenant": ignorcase_get(vm, "tenant")})
287
288         data["additionalparam"] = {}
289         data["additionalparam"]["vnfmid"] = ignorcase_get(request.data, "vnfmid")
290         data["additionalparam"]["vimid"] = ignorcase_get(request.data, "vimid")
291         data["additionalparam"]["tenant"] = ignorcase_get(request.data, "tenant")
292
293         ret = req_by_msb('api/nslcm/v1/ns/grantvnf', "POST", content=json.JSONEncoder().encode(data))
294         logger.info("ret = %s", ret)
295         if ret[0] != 0:
296             return Response(data={'error': ret[1]}, status=ret[2])
297         resp = json.JSONDecoder().decode(ret[1])
298         resp_data = {
299             'vimid': ignorcase_get(resp['vim'], 'vimid'),
300             'tenant': ignorcase_get(ignorcase_get(resp['vim'], 'accessinfo'), 'tenant')
301         }
302         logger.info("[%s]resp_data=%s", fun_name(), resp_data)
303     except Exception as e:
304         logger.error("Error occurred in Grant VNF.")
305         raise e
306     return Response(data=resp_data, status=ret[2])
307
308
309 @api_view(http_method_names=['POST'])
310 def notify(request, *args, **kwargs):
311     try:
312         logger.info("[%s]req_data = %s", fun_name(), request.data)
313         notify_param_map = {
314             "NFVOID": "",
315             "VNFMID": "VNFMID",
316             "VIMID": "vimid",
317             "VNFInstanceID": "vnfInstanceId",
318             "TimeStamp": "",
319             "EventType": "operation",
320             "VMList": "",
321             "VMFlavor": "",
322             "VMNumber": "",
323             "VMIDlist": "",
324             "VMUUID": ""
325         }
326         data = mapping_conv(notify_param_map, request.data)
327         logger.info("[%s]data = %s", fun_name(), data)
328
329         data["status"] = "result"
330         data["jobId"] = "notMust"
331         data["affectedVnfc"] = []
332         data["affectedVl"] = []
333         data["affectedVirtualStorage"] = []
334         data["affectedCp"] = []
335
336         extension = ignorcase_get(request.data, "extension")
337         openo_notification = ignorcase_get(extension, "openo_notification")
338         if openo_notification:
339             affectedvnfcs = ignorcase_get(openo_notification, "affectedVnfc")
340             affectedvls = ignorcase_get(openo_notification, "affectedvirtuallink")
341             affectedcps = ignorcase_get(openo_notification, "affectedCp")
342             vnfdmodule = ignorcase_get(openo_notification, "vnfdmodule")
343         else:
344             affectedvnfcs = ignorcase_get(ignorcase_get(request.data, "extension"), "affectedvnfc")
345             affectedvls = ignorcase_get(ignorcase_get(request.data, "extension"), "affectedvl")
346             affectedcps = ignorcase_get(ignorcase_get(request.data, "extension"), "affectedcp")
347             vnfdmodule = ignorcase_get(ignorcase_get(request.data, "extension"), "vnfdmodule")
348
349         data["vnfdmodule"] = vnfdmodule
350
351         for affectedvnfc in affectedvnfcs:
352             data["affectedVnfc"].append({
353                 "vnfcInstanceId": ignorcase_get(affectedvnfc, "vnfcInstanceId"),
354                 "vduId": ignorcase_get(affectedvnfc, "vduId"),
355                 "changeType": ignorcase_get(affectedvnfc, "changeType"),
356                 "vimid": ignorcase_get(ignorcase_get(affectedvnfc, "computeResource"), "vimId"),
357                 "vmId": ignorcase_get(ignorcase_get(affectedvnfc, "computeResource"), "resourceId"),
358                 "vmName": ignorcase_get(ignorcase_get(affectedvnfc, "computeResource"), "resourceName")
359             })
360
361         for affectedvl in affectedvls:
362             data["affectedVl"].append({
363                 "vlInstanceId": ignorcase_get(affectedvl, "virtuallinkinstanceid"),
364                 "vimid": ignorcase_get(ignorcase_get(affectedvl, "networkresource"), "vimid"),
365                 "vldid": ignorcase_get(affectedvl, "virtuallinkdescid"),
366                 "vllid": ignorcase_get(ignorcase_get(affectedvl, "networkresource"), "resourceid"),
367                 "vlName": ignorcase_get(ignorcase_get(affectedvl, "networkresource"), "resourcename")
368             })
369
370         for affectedcp in affectedcps:
371             data["affectedCp"].append(affectedcp)
372
373         notify_url = 'api/nslcm/v1/ns/{vnfmid}/vnfs/{vnfInstanceId}/Notify'
374         ret = req_by_msb(notify_url.format(vnfmid=ignorcase_get(data, 'VNFMID'),
375                                            vnfInstanceId=ignorcase_get(data, 'vnfinstanceid')),
376                          "POST", content=json.JSONEncoder().encode(data))
377
378         logger.info("[%s]data = %s", fun_name(), ret)
379         if ret[0] != 0:
380             return Response(data={'error': ret[1]}, status=ret[2])
381     except Exception as e:
382         logger.error("Error occurred in LCM notification.")
383         logger.error(traceback.format_exc())
384         raise e
385     return Response(data=None, status=ret[2])
386
387
388 @api_view(http_method_names=['POST'])
389 def scale(request, *args, **kwargs):
390     logger.info("====scale_vnf===")
391     try:
392         logger.info("request.data = %s", request.data)
393         logger.info("requested_url = %s", request.get_full_path())
394         vnfm_id = ignorcase_get(kwargs, "vnfmid")
395         nf_instance_id = ignorcase_get(kwargs, "vnfInstanceId")
396         ret = get_vnfminfo_from_nslcm(vnfm_id)
397         if ret[0] != 0:
398             return Response(data={'error': ret[1]}, status=ret[2])
399         vnfm_info = json.JSONDecoder().decode(ret[1])
400         scale_type = ignorcase_get(request.data, "type")
401         aspect_id = ignorcase_get(request.data, "aspectId")
402         number_of_steps = ignorcase_get(request.data, "numberOfSteps")
403         data = {
404             'vnfmid': vnfm_id,
405             'nfvoid': 1,
406             'scaletype': '0' if scale_type == 'SCALE_OUT' else '1',
407             'vmlist': [{'VMNumber': number_of_steps, 'VMFlavor': aspect_id}],
408             'extension': ''
409         }
410
411         logger.info("data = %s", data)
412         ret = restcall.call_req(
413             base_url=ignorcase_get(vnfm_info, "url"),
414             user=ignorcase_get(vnfm_info, "userName"),
415             passwd=ignorcase_get(vnfm_info, "password"),
416             auth_type=restcall.rest_no_auth,
417             resource='/v1/vnfs/{vnfInstanceID}/scale'.format(vnfInstanceID=nf_instance_id),
418             method='put',  # POST
419             content=json.JSONEncoder().encode(data))
420         logger.info("ret=%s", ret)
421         if ret[0] != 0:
422             return Response(data={'error': 'scale error'}, status=ret[2])
423         resp_data = json.JSONDecoder().decode(ret[1])
424         logger.info("resp_data=%s", resp_data)
425     except Exception as e:
426         logger.error("Error occurred when scaling VNF,error:%s", e.message)
427         logger.error(traceback.format_exc())
428         return Response(data={'error': 'scale expection'}, status='500')
429     return Response(data=resp_data, status=ret[2])
430
431
432 @api_view(http_method_names=['POST'])
433 def heal(request, *args, **kwargs):
434     logger.info("====heal_vnf===")
435     try:
436         logger.info("request.data = %s", request.data)
437         logger.info("requested_url = %s", request.get_full_path())
438         vnfm_id = ignorcase_get(kwargs, "vnfmid")
439         nf_instance_id = ignorcase_get(kwargs, "vnfInstanceId")
440         ret = get_vnfminfo_from_nslcm(vnfm_id)
441         if ret[0] != 0:
442             return Response(data={'error': ret[1]}, status=ret[2])
443         vnfm_info = json.JSONDecoder().decode(ret[1])
444         data = {}
445         data['action'] = ignorcase_get(request.data, 'action')
446         affectedvm = ignorcase_get(request.data, 'affectedvm')
447         data['affectedvm'] = []
448         if isinstance(affectedvm, list):
449             data['affectedvm'] = affectedvm
450         else:
451             data['affectedvm'].append(affectedvm)
452         data['lifecycleoperation'] = 'operate'
453         data['isgrace'] = 'force'
454
455         logger.info("data = %s", data)
456         ret = restcall.call_req(
457             base_url=ignorcase_get(vnfm_info, "url"),
458             user=ignorcase_get(vnfm_info, "userName"),
459             passwd=ignorcase_get(vnfm_info, "password"),
460             auth_type=restcall.rest_no_auth,
461             resource='/api/v1/nf_m_i/nfs/{vnfInstanceID}/vms/operation'.format(vnfInstanceID=nf_instance_id),
462             method='post',
463             content=json.JSONEncoder().encode(data))
464         logger.info("ret=%s", ret)
465         if ret[0] != 0:
466             return Response(data={'error': 'heal error'}, status=ret[2])
467         resp_data = json.JSONDecoder().decode(ret[1])
468         logger.info("resp_data=%s", resp_data)
469     except Exception as e:
470         logger.error("Error occurred when healing VNF,error:%s", e.message)
471         logger.error(traceback.format_exc())
472         return Response(data={'error': 'heal expection'}, status='500')
473     return Response(data=resp_data, status=ret[2])
474
475
476 def get_vdus(nf_model, aspect_id):
477     associated_group = ''
478     members = []
479     vnf_flavours = nf_model['vnf_flavours']
480     for vnf_flaour in vnf_flavours:
481         scaling_aspects = vnf_flaour['scaling_aspects']
482         for aspect in scaling_aspects:
483             if aspect_id == aspect['id']:
484                 associated_group = aspect['associated_group']
485                 break
486     if not associated_group:
487         logger.error('Cannot find the corresponding element group')
488         raise Exception('Cannot find the corresponding element group')
489     for element_group in nf_model['element_groups']:
490         if element_group['group_id'] == associated_group:
491             members = element_group['members']
492     if not members:
493         logger.error('Cannot find the corresponding members')
494         raise Exception('Cannot find the corresponding members')
495     return members
496
497
498 @api_view(http_method_names=['GET'])
499 def samples(request, *args, **kwargs):
500     return Response(data={"status": "ok"})