diff options
| -rw-r--r-- | lcm/pub/utils/toscaparser/basemodel.py | 8 | ||||
| -rw-r--r-- | lcm/pub/utils/toscaparser/vnfdmodel.py | 83 |
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 |