Add parser convert vnfd vdu logic

After parse the vnfd package, add
_get_all_vdu function to convert
the vdu info.

Change-Id: Ie4e59efda2a22222871cdacc8a6c14ff3feaecb5
Issue-ID: VFC-184
Signed-off-by: ying.yunlong <ying.yunlong@zte.com.cn>

2 files changed, 91 insertions, 0 deletions

diff --git a/lcm/pub/utils/toscaparser/basemodel.py b/lcm/pub/utils/toscaparser/basemodel.py
index 351f0935..218f9336 100644
--- a/lcm/pub/utils/toscaparser/basemodel.py
+++ b/lcm/pub/utils/toscaparser/basemodel.py
@@ -326,3 +326,11 @@ class BaseInfoModel(object):
 
     def is_nested_ns(self, node):
         return node['nodeType'].upper().find('.NS.') >= 0 or node['nodeType'].upper().endswith('.NS')
+
+    def isVdu(self, node):
+        return node['nodeType'].upper().find('.VDU.') >= 0 or node['nodeType'].upper().endswith('.VDU')
+
+    def getCapabilityByName(self, node, capabilityName):
+        if 'capabilities' in node and capabilityName in node['capabilities']:
+            return node['capabilities'][capabilityName]
+        return None
diff --git a/lcm/pub/utils/toscaparser/vnfdmodel.py b/lcm/pub/utils/toscaparser/vnfdmodel.py
index cd1ee565..59eab4e8 100644
--- a/lcm/pub/utils/toscaparser/vnfdmodel.py
+++ b/lcm/pub/utils/toscaparser/vnfdmodel.py
@@ -36,6 +36,7 @@ class EtsiVnfdInfoModel(EtsiNsdInfoModel):
         self.image_files = self._get_all_image_file(nodeTemplates)
         self.local_storages = self._get_all_local_storage(nodeTemplates)
         self.volume_storages = self._get_all_volume_storage(nodeTemplates)
+        self.vdus = self._get_all_vdu(nodeTemplates)
 
 
     def _get_all_services(self, nodeTemplates):
@@ -149,3 +150,85 @@ class EtsiVnfdInfoModel(EtsiNsdInfoModel):
     def _isVolumeStorage(self, node):
         return node['nodeType'].upper().find('.VOLUMESTORAGE.') >= 0 or node['nodeType'].upper().endswith(
             '.VOLUMESTORAGE')
+
+    def _get_all_vdu(self, nodeTemplates):
+        rets = []
+        for node in nodeTemplates:
+            if self.isVdu(node):
+                ret = {}
+                ret['vdu_id'] = node['name']
+                if 'description' in node:
+                    ret['description'] = node['description']
+                ret['properties'] = node['properties']
+                ret['image_file'] = self.get_node_image_file(node)
+                local_storages = self.getRequirementByName(node, 'local_storage')
+                ret['local_storages'] = map(lambda x: self.get_requirement_node_name(x), local_storages)
+                volume_storages = self.getRequirementByName(node, 'volume_storage')
+                ret['volume_storages'] = map(functools.partial(self._trans_volume_storage), volume_storages)
+                ret['dependencies'] = map(lambda x: self.get_requirement_node_name(x), self.getNodeDependencys(node))
+
+                nfv_compute = self.getCapabilityByName(node, 'nfv_compute')
+                if nfv_compute != None and 'properties' in nfv_compute:
+                    ret['nfv_compute'] = nfv_compute['properties']
+
+                ret['vls'] = self.get_linked_vl_ids(node, nodeTemplates)
+
+                scalable = self.getCapabilityByName(node, 'scalable')
+                if scalable != None and 'properties' in scalable:
+                    ret['scalable'] = scalable['properties']
+
+                ret['cps'] = self.getVirtalBindingCpIds(node, nodeTemplates)
+                ret['artifacts'] = self._build_artifacts(node)
+
+                rets.append(ret)
+        return rets
+
+    def get_node_image_file(self, node):
+        rets = map(lambda x: self.get_requirement_node_name(x), self.getRequirementByName(node, 'guest_os'))
+        if len(rets) > 0:
+            return rets[0]
+        return ""
+
+    def _trans_volume_storage(self, volume_storage):
+        if isinstance(volume_storage, str):
+            return {"volume_storage_id": volume_storage}
+        else:
+            ret = {}
+            ret['volume_storage_id'] = self.get_requirement_node_name(volume_storage)
+            if 'relationship' in volume_storage and 'properties' in volume_storage['relationship']:
+                if 'location' in volume_storage['relationship']['properties']:
+                    ret['location'] = volume_storage['relationship']['properties']['location']
+                if 'device' in volume_storage['relationship']['properties']:
+                    ret['device'] = volume_storage['relationship']['properties']['device']
+
+            return ret
+
+    def get_linked_vl_ids(self, node, node_templates):
+        vl_ids = []
+        cps = self.getVirtalBindingCps(node, node_templates)
+        for cp in cps:
+            vl_reqs = self.getVirtualLinks(cp)
+            for vl_req in vl_reqs:
+                vl_ids.append(self.get_requirement_node_name(vl_req))
+        return vl_ids
+
+    def _build_artifacts(self, node):
+        rets = []
+        if 'artifacts' in node and len(node['artifacts']) > 0:
+            artifacts = node['artifacts']
+            for name, value in artifacts.items():
+                ret = {}
+                if isinstance(value, dict):
+                    ret['artifact_name'] = name
+                    ret['type'] = value.get('type', '')
+                    ret['file'] = value.get('file', '')
+                    ret['repository'] = value.get('repository', '')
+                    ret['deploy_path'] = value.get('deploy_path', '')
+                else:
+                    ret['artifact_name'] = name
+                    ret['type'] = ''
+                    ret['file'] = value
+                    ret['repository'] = ''
+                    ret['deploy_path'] = ''
+                rets.append(ret)
+        return rets