genericparser/pub/utils/toscaparsers/vnfdparser/vnfd_sol_251.py

   1 # Copyright 2019 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 functools
  16 import logging
  17 import os
  18 import base64
  19
  20
  21 logger = logging.getLogger(__name__)
  22
  23 SECTIONS = (VDU_COMPUTE_TYPE, VNF_VL_TYPE, VDU_CP_TYPE, VDU_STORAGE_TYPE) = \
  24            ('tosca.nodes.nfv.Vdu.Compute', 'tosca.nodes.nfv.VnfVirtualLink', 'tosca.nodes.nfv.VduCp', 'tosca.nodes.nfv.Vdu.VirtualStorage')
  25
  26
  27 class VnfdSOL251():
  28
  29     def __init__(self, model):
  30         self.model = model
  31
  32     def build_vnf(self, tosca):
  33         vnf = self.model.get_substitution_mappings(tosca)
  34         properties = vnf.get("properties", {})
  35         metadata = vnf.get("metadata", {})
  36
  37         for key, value in list(properties.items()):
  38             if isinstance(value, dict):
  39                 if value["type"] == "string":
  40                     properties[key] = value.get("default", "")
  41                 elif value["type"] == "list":
  42                     properties[key] = value.get("default", {})
  43                 else:
  44                     properties[key] = value.get("default", "")
  45         ptype = "descriptor_id"
  46         meta_types = ["descriptor_id", "id", "UUID"]
  47         self._get_property(properties, metadata, ptype, meta_types)
  48
  49         ptype = "descriptor_version"
  50         meta_types = ["template_version", "version"]
  51         self._get_property(properties, metadata, ptype, meta_types)
  52
  53         ptype = "provider"
  54         meta_types = ["template_author", "provider"]
  55         self._get_property(properties, metadata, ptype, meta_types)
  56
  57         ptype = "template_name"
  58         meta_types = ["template_name"]
  59         self._get_property(properties, metadata, ptype, meta_types)
  60
  61         ptype = "software_version"
  62         meta_types = ["software_version"]
  63         self._get_property(properties, metadata, ptype, meta_types)
  64
  65         ptype = "product_name"
  66         meta_types = ["product_name"]
  67         self._get_property(properties, metadata, ptype, meta_types)
  68
  69         ptype = "flavour_description"
  70         meta_types = ["flavour_description"]
  71         self._get_property(properties, metadata, ptype, meta_types)
  72
  73         ptype = "vnfm_info"
  74         meta_types = ["vnfm_info"]
  75         self._get_property(properties, metadata, ptype, meta_types)
  76
  77         ptype = "flavour_id"
  78         meta_types = ["flavour_id"]
  79         self._get_property(properties, metadata, ptype, meta_types)
  80
  81         logger.debug("vnf:%s", vnf)
  82
  83         return vnf
  84
  85     def get_all_vl(self, nodeTemplates, node_types):
  86         vls = []
  87         for node in nodeTemplates:
  88             if self.model.isNodeTypeX(node, node_types, VNF_VL_TYPE):
  89                 vl = dict()
  90                 vl['vl_id'] = node['name']
  91                 vl['description'] = node['description']
  92                 vl['properties'] = node['properties']
  93                 vls.append(vl)
  94         return vls
  95
  96     def get_all_cp(self, nodeTemplates, node_types):
  97         cps = []
  98         for node in nodeTemplates:
  99             if self.model.isNodeTypeX(node, node_types, VDU_CP_TYPE):
 100                 cp = {}
 101                 cp['cp_id'] = node['name']
 102                 cp['cpd_id'] = node['name']
 103                 cp['description'] = node['description']
 104                 cp['properties'] = node['properties']
 105                 cp['vl_id'] = self._get_node_vl_id(node)
 106                 cp['vdu_id'] = self._get_node_vdu_id(node)
 107                 vls = self._buil_cp_vls(node)
 108                 if len(vls) > 1:
 109                     cp['vls'] = vls
 110                 cps.append(cp)
 111         return cps
 112
 113     def get_all_volume_storage(self, nodeTemplates, node_types):
 114         rets = []
 115         for node in nodeTemplates:
 116             if self.model.isNodeTypeX(node, node_types, VDU_STORAGE_TYPE):
 117                 ret = {}
 118                 ret['volume_storage_id'] = node['name']
 119                 if 'description' in node:
 120                     ret['description'] = node['description']
 121                 ret['properties'] = node['properties']
 122                 rets.append(ret)
 123         return rets
 124
 125     def get_all_vdu(self, nodeTemplates, node_types):
 126         rets = []
 127         inject_files = []
 128         for node in nodeTemplates:
 129             logger.debug("nodeTemplates :%s", node)
 130             if self.model.isNodeTypeX(node, node_types, VDU_COMPUTE_TYPE):
 131                 ret = {}
 132                 ret['vdu_id'] = node['name']
 133                 ret['type'] = node['nodeType']
 134                 if 'description' in node:
 135                     ret['description'] = node['description']
 136                 ret['properties'] = node['properties']
 137                 if 'inject_files' in node['properties']:
 138                     inject_files = node['properties']['inject_files']
 139                 if inject_files is not None:
 140                     if isinstance(inject_files, list):
 141                         for inject_file in inject_files:
 142                             source_path = os.path.join(self.model.basepath, inject_file['source_path'])
 143                             with open(source_path, "rb") as f:
 144                                 source_data = f.read()
 145                                 source_data_base64 = base64.b64encode(source_data)
 146                                 inject_file["source_data_base64"] = source_data_base64.decode()
 147                     if isinstance(inject_files, dict):
 148                         source_path = os.path.join(self.model.basepath, inject_files['source_path'])
 149                         with open(source_path, "rb") as f:
 150                             source_data = f.read()
 151                             source_data_base64 = base64.b64encode(source_data)
 152                             inject_files["source_data_base64"] = source_data_base64.decode()
 153                 virtual_storages = self.model.getRequirementByName(node, 'virtual_storage')
 154                 ret['virtual_storages'] = list(map(functools.partial(self._trans_virtual_storage), virtual_storages))
 155                 ret['dependencies'] = [self.model.get_requirement_node_name(x) for x in self.model.getNodeDependencys(node)]
 156                 virtual_compute = self.model.getCapabilityByName(node, 'virtual_compute')
 157                 if virtual_compute is not None and 'properties' in virtual_compute:
 158                     ret['virtual_compute'] = virtual_compute['properties']
 159                 ret['vls'] = self._get_linked_vl_ids(node, nodeTemplates)
 160                 ret['cps'] = self._get_virtal_binding_cp_ids(node, nodeTemplates)
 161                 ret['artifacts'] = self.model.build_artifacts(node)
 162                 rets.append(ret)
 163         logger.debug("rets:%s", rets)
 164         return rets
 165
 166     def get_all_endpoint_exposed(self):
 167         if self.model.vnf:
 168             external_cps = self._get_external_cps(self.model.vnf.get('requirements', None))
 169             forward_cps = self._get_forward_cps(self.model.vnf.get('capabilities', None))
 170             return {"external_cps": external_cps, "forward_cps": forward_cps}
 171         return {}
 172
 173     def _get_property(self, properties, metadata, ptype, meta_types):
 174         if ptype not in properties or properties[ptype] == "":
 175             for mtype in meta_types:
 176                 data = metadata.get(mtype, "")
 177                 if data != "":
 178                     properties[ptype] = data
 179
 180     def _trans_virtual_storage(self, virtual_storage):
 181         if isinstance(virtual_storage, str):
 182             return {"virtual_storage_id": virtual_storage}
 183         else:
 184             ret = {}
 185             ret['virtual_storage_id'] = self.model.get_requirement_node_name(virtual_storage)
 186             return ret
 187
 188     def _get_linked_vl_ids(self, node, node_templates):
 189         vl_ids = []
 190         cps = self._get_virtal_binding_cps(node, node_templates)
 191         for cp in cps:
 192             vl_reqs = self.model.getRequirementByName(cp, 'virtual_link')
 193             for vl_req in vl_reqs:
 194                 vl_ids.append(self.model.get_requirement_node_name(vl_req))
 195         return vl_ids
 196
 197     def _get_virtal_binding_cp_ids(self, node, nodeTemplates):
 198         return [x['name'] for x in self._get_virtal_binding_cps(node, nodeTemplates)]
 199
 200     def _get_virtal_binding_cps(self, node, nodeTemplates):
 201         cps = []
 202         for tmpnode in nodeTemplates:
 203             if 'requirements' in tmpnode:
 204                 for item in tmpnode['requirements']:
 205                     for key, value in list(item.items()):
 206                         if key.upper().startswith('VIRTUAL_BINDING'):
 207                             req_node_name = self.model.get_requirement_node_name(value)
 208                             if req_node_name is not None and req_node_name == node['name']:
 209                                 cps.append(tmpnode)
 210         return cps
 211
 212     def _get_node_vdu_id(self, node):
 213         vdu_ids = [self.model.get_requirement_node_name(x) for x in self.model.getRequirementByName(node, 'virtual_binding')]
 214         if len(vdu_ids) > 0:
 215             return vdu_ids[0]
 216         return ""
 217
 218     def _get_node_vl_id(self, node):
 219         vl_ids = [self.model.get_requirement_node_name(x) for x in self.model.getRequirementByName(node, 'virtual_link')]
 220         if len(vl_ids) > 0:
 221             return vl_ids[0]
 222         return ""
 223
 224     def _buil_cp_vls(self, node):
 225         return [self._build_cp_vl(x) for x in self.model.getRequirementByName(node, 'virtual_link')]
 226
 227     def _build_cp_vl(self, req):
 228         cp_vl = {}
 229         cp_vl['vl_id'] = self.model.get_prop_from_obj(req, 'node')
 230         relationship = self.model.get_prop_from_obj(req, 'relationship')
 231         if relationship is not None:
 232             properties = self.model.get_prop_from_obj(relationship, 'properties')
 233             if properties is not None and isinstance(properties, dict):
 234                 for key, value in list(properties.items()):
 235                     cp_vl[key] = value
 236         return cp_vl
 237
 238     def _get_external_cps(self, vnf_requirements):
 239         external_cps = []
 240         if vnf_requirements:
 241             if isinstance(vnf_requirements, dict):
 242                 for key, value in list(vnf_requirements.items()):
 243                     if isinstance(value, list) and len(value) > 0:
 244                         external_cps.append({"key_name": key, "cpd_id": value[0]})
 245                     else:
 246                         external_cps.append({"key_name": key, "cpd_id": value})
 247             elif isinstance(vnf_requirements, list):
 248                 for vnf_requirement in vnf_requirements:
 249                     for key, value in list(vnf_requirement.items()):
 250                         if isinstance(value, list) and len(value) > 0:
 251                             external_cps.append({"key_name": key, "cpd_id": value[0]})
 252                         else:
 253                             external_cps.append({"key_name": key, "cpd_id": value})
 254         return external_cps
 255
 256     def _get_forward_cps(self, vnf_capabilities):
 257         forward_cps = []
 258         if vnf_capabilities:
 259             for key, value in list(vnf_capabilities.items()):
 260                 if isinstance(value, list) and len(value) > 0:
 261                     forward_cps.append({"key_name": key, "cpd_id": value[0]})
 262                 else:
 263                     forward_cps.append({"key_name": key, "cpd_id": value})
 264         return forward_cps