83d94e50c4356faaa5d818f1cf775e64ff13577e
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / biz / grant_vnf.py
1 # Copyright 2018 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 json
16 import logging
17
18 from lcm.pub.database.models import NfInstModel
19 from lcm.pub.msapi.gvnfmdriver import apply_grant_to_nfvo
20 from lcm.pub.utils.values import ignore_case_get
21 from lcm.nf.const import GRANT_TYPE
22
23 logger = logging.getLogger(__name__)
24
25
26 def grant_resource(data, nf_inst_id, job_id, grant_type, vdus):
27     logger.info("Grant resource begin")
28     content_args = {
29         'vnfInstanceId': nf_inst_id,
30         'vnfLcmOpOccId': job_id,
31         # 'vnfdId': None,  # TODO
32         # 'flavourId': "default",  # TODO
33         'operation': grant_type,
34         'isAutomaticInvocation': True,  # TODO
35         # 'instantiationLevelId': None,  # TODO
36         # 'addResources': [],
37         # 'tmpResources': [],
38         # 'updateResources': [],
39         # 'removeResources': [],
40         # 'placementConstraints': [],
41         # 'vimConstraints': [],
42         'additionalParams': {}
43         # '_links': None  # TODO
44     }
45
46     if grant_type == GRANT_TYPE.TERMINATE:
47         res_index = 1
48         content_args['removeResources'] = []
49         for vdu in vdus:
50             res_def = {
51                 'id': str(res_index),
52                 'type': 'COMPUTE',
53                 # 'vduId': None,
54                 # 'resourceTemplateId': None,
55                 'resource': {
56                     #  'vimConnectionId': None,
57                     #  'resourceProviderId': None,
58                     'resourceId': vdu.resourceid,
59                     #  'vimLevelResourceType': None
60                 }
61             }
62             content_args['removeResources'].append(res_def)
63             res_index += 1
64         if vdus and vdus[0].vimid:
65             content_args['additionalParams']['vimid'] = vdus[0].vimid
66     elif grant_type == GRANT_TYPE.INSTANTIATE:
67         vim_id = ignore_case_get(ignore_case_get(data, "additionalParams"), "vimId")
68         res_index = 1
69         content_args['addResources'] = []
70         for vdu in vdus:
71             res_def = {
72                 'id': str(res_index),
73                 'type': 'COMPUTE',
74                 # 'vduId': vdu["vdu_id"],
75                 'resourceTemplateId': vdu["vdu_id"]  # , None,
76                 # 'resource': None
77             }
78             content_args['addResources'].append(res_def)
79             res_index += 1
80         if vim_id:
81             content_args['additionalParams']['vimid'] = vim_id
82     elif grant_type == GRANT_TYPE.OPERATE:
83         res_index = 1
84         content_args['updateResources'] = []
85         for vdu in vdus:
86             res_def = {
87                 'id': str(res_index),
88                 'type': 'COMPUTE',
89                 'vduId': None,
90                 'resourceTemplateId': None,
91                 'resource': {
92                     'vimConnectionId': None,
93                     'resourceProviderId': None,
94                     'resourceId': vdu.resourceid,
95                     'vimLevelResourceType': None
96                 }
97             }
98             content_args['updateResources'].append(res_def)
99             res_index += 1
100         if vdus and vdus[0].vimid:
101             content_args['additionalParams']['vimid'] = vdus[0].vimid
102
103     vnfInst_list = NfInstModel.objects.filter(nfinstid=nf_inst_id)
104     addition_paras = content_args['additionalParams']
105     for vnf in vnfInst_list:
106         if vnf.vnfminstid:
107             addition_paras['vnfmid'] = vnf.vnfminstid
108         if vnf.vimInfo and 'vimid' not in addition_paras:
109             vim_info = json.loads(vnf.vimInfo)
110             vimid = ""
111             for key in vim_info.iterkeys():
112                 vimid = key
113             addition_paras['vimid'] = vimid
114     logger.info('Grant request data=%s' % content_args)
115     apply_result = apply_grant_to_nfvo(json.dumps(content_args))
116     logger.info("apply_result: %s" % apply_result)
117     return apply_result