X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=lcm%2Flcm%2Fnf%2Fbiz%2Fgrant_vnf.py;h=80ff035c9095181be39b7daa84017d7f47231eb1;hb=46ad7c172411214c5432ed93fda4271288077447;hp=cbfa88b0ddb93a4ae70adc1cbf43867c1ea6915b;hpb=ba6731a7525cea9aecfe4d5dcc90171c07965a43;p=vfc%2Fgvnfm%2Fvnflcm.git

diff --git a/lcm/lcm/nf/biz/grant_vnf.py b/lcm/lcm/nf/biz/grant_vnf.py
index cbfa88b0..80ff035c 100644
--- a/lcm/lcm/nf/biz/grant_vnf.py
+++ b/lcm/lcm/nf/biz/grant_vnf.py
@@ -15,38 +15,103 @@
 import json
 import logging
 
-from lcm.pub.database.models import NfInstModel, VmInstModel
+from lcm.pub.database.models import NfInstModel
 from lcm.pub.msapi.gvnfmdriver import apply_grant_to_nfvo
+from lcm.pub.utils.values import ignore_case_get
+from lcm.nf.const import GRANT_TYPE
 
 logger = logging.getLogger(__name__)
 
 
-def grant_resource(nf_inst_id, job_id):
+def grant_resource(data, nf_inst_id, job_id, grant_type, vdus):
     logger.info("Grant resource begin")
     content_args = {
         'vnfInstanceId': nf_inst_id,
-        'vnfDescriptorId': '',
-        'lifecycleOperation': 'Terminate',
         'vnfLcmOpOccId': job_id,
-        'addResource': [],
-        'removeResource': [],
-        'placementConstraint': [],
-        'additionalParam': {}
+        # 'vnfdId': None,  # TODO
+        # 'flavourId': "default",  # TODO
+        'operation': grant_type,
+        'isAutomaticInvocation': True,  # TODO
+        # 'instantiationLevelId': None,  # TODO
+        # 'addResources': [],
+        # 'tmpResources': [],
+        # 'updateResources': [],
+        # 'removeResources': [],
+        # 'placementConstraints': [],
+        # 'vimConstraints': [],
+        'additionalParams': {}
+        # '_links': None  # TODO
     }
 
-    vdus = VmInstModel.objects.filter(instid=nf_inst_id, is_predefined=1)
-    res_index = 1
-    for vdu in vdus:
-        res_def = {
-            'type': 'VDU',
-            'resDefId': str(res_index),
-            'resDesId': vdu.resouceid}
-        content_args['removeResource'].append(res_def)
-        res_index += 1
+    if grant_type == GRANT_TYPE.TERMINATE:
+        res_index = 1
+        content_args['removeResources'] = []
+        for vdu in vdus:
+            res_def = {
+                'id': str(res_index),
+                'type': 'COMPUTE',
+                # 'vduId': None,
+                # 'resourceTemplateId': None,
+                'resource': {
+                    #  'vimConnectionId': None,
+                    #  'resourceProviderId': None,
+                    'resourceId': vdu.resourceid,
+                    #  'vimLevelResourceType': None
+                }
+            }
+            content_args['removeResources'].append(res_def)
+            res_index += 1
+        if vdus and vdus[0].vimid:
+            content_args['additionalParams']['vimid'] = vdus[0].vimid
+    elif grant_type == GRANT_TYPE.INSTANTIATE:
+        vim_id = ignore_case_get(ignore_case_get(data, "additionalParams"), "vimId")
+        res_index = 1
+        content_args['addResources'] = []
+        for vdu in vdus:
+            res_def = {
+                'id': str(res_index),
+                'type': 'COMPUTE',
+                # 'vduId': vdu["vdu_id"],
+                'resourceTemplateId': vdu["vdu_id"]  # , None,
+                # 'resource': None
+            }
+            content_args['addResources'].append(res_def)
+            res_index += 1
+        if vim_id:
+            content_args['additionalParams']['vimid'] = vim_id
+    elif grant_type == GRANT_TYPE.OPERATE:
+        res_index = 1
+        content_args['updateResources'] = []
+        for vdu in vdus:
+            res_def = {
+                'id': str(res_index),
+                'type': 'COMPUTE',
+                'vduId': None,
+                'resourceTemplateId': None,
+                'resource': {
+                    'vimConnectionId': None,
+                    'resourceProviderId': None,
+                    'resourceId': vdu.resourceid,
+                    'vimLevelResourceType': None
+                }
+            }
+            content_args['updateResources'].append(res_def)
+            res_index += 1
+        if vdus and vdus[0].vimid:
+            content_args['additionalParams']['vimid'] = vdus[0].vimid
 
-    vnfInsts = NfInstModel.objects.filter(nfinstid=nf_inst_id)
-    content_args['additionalParam']['vnfmid'] = vnfInsts[0].vnfminstid
-    content_args['additionalParam']['vimid'] = vdus[0].vimid
+    vnfInst_list = NfInstModel.objects.filter(nfinstid=nf_inst_id)
+    addition_paras = content_args['additionalParams']
+    for vnf in vnfInst_list:
+        if vnf.vnfminstid:
+            addition_paras['vnfmid'] = vnf.vnfminstid
+        if vnf.vimInfo and 'vimid' not in addition_paras:
+            vim_info = json.loads(vnf.vimInfo)
+            vimid = ""
+            for key in list(vim_info.keys()):
+                vimid = key
+            addition_paras['vimid'] = vimid
     logger.info('Grant request data=%s' % content_args)
     apply_result = apply_grant_to_nfvo(json.dumps(content_args))
-    logger.info("Grant resource end, response: %s" % apply_result)
+    logger.info("apply_result: %s" % apply_result)
+    return apply_result