1 # Copyright 2019 ZTE Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
21 logger = logging.getLogger(__name__)
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')
29 def __init__(self, model):
32 def build_vnf(self, tosca):
33 vnf = self.model.get_substitution_mappings(tosca)
34 properties = vnf.get("properties", {})
35 metadata = vnf.get("metadata", {})
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", {})
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)
49 ptype = "descriptor_version"
50 meta_types = ["template_version", "version"]
51 self._get_property(properties, metadata, ptype, meta_types)
54 meta_types = ["template_author", "provider"]
55 self._get_property(properties, metadata, ptype, meta_types)
57 ptype = "template_name"
58 meta_types = ["template_name"]
59 self._get_property(properties, metadata, ptype, meta_types)
61 ptype = "software_version"
62 meta_types = ["software_version"]
63 self._get_property(properties, metadata, ptype, meta_types)
65 ptype = "product_name"
66 meta_types = ["product_name"]
67 self._get_property(properties, metadata, ptype, meta_types)
69 ptype = "flavour_description"
70 meta_types = ["flavour_description"]
71 self._get_property(properties, metadata, ptype, meta_types)
74 meta_types = ["vnfm_info"]
75 self._get_property(properties, metadata, ptype, meta_types)
78 meta_types = ["flavour_id"]
79 self._get_property(properties, metadata, ptype, meta_types)
81 logger.debug("vnf:%s", vnf)
85 def get_all_vl(self, nodeTemplates, node_types):
87 for node in nodeTemplates:
88 if self.model.isNodeTypeX(node, node_types, VNF_VL_TYPE):
90 vl['vl_id'] = node['name']
91 vl['description'] = node['description']
92 vl['properties'] = node['properties']
96 def get_all_cp(self, nodeTemplates, node_types):
98 for node in nodeTemplates:
99 if self.model.isNodeTypeX(node, node_types, VDU_CP_TYPE):
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)
113 def get_all_volume_storage(self, nodeTemplates, node_types):
115 for node in nodeTemplates:
116 if self.model.isNodeTypeX(node, node_types, VDU_STORAGE_TYPE):
118 ret['volume_storage_id'] = node['name']
119 if 'description' in node:
120 ret['description'] = node['description']
121 ret['properties'] = node['properties']
125 def get_all_vdu(self, nodeTemplates, node_types):
128 for node in nodeTemplates:
129 logger.debug("nodeTemplates :%s", node)
130 if self.model.isNodeTypeX(node, node_types, VDU_COMPUTE_TYPE):
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)
163 logger.debug("rets:%s", rets)
166 def get_all_endpoint_exposed(self):
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}
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, "")
178 properties[ptype] = data
180 def _trans_virtual_storage(self, virtual_storage):
181 if isinstance(virtual_storage, str):
182 return {"virtual_storage_id": virtual_storage}
185 ret['virtual_storage_id'] = self.model.get_requirement_node_name(virtual_storage)
188 def _get_linked_vl_ids(self, node, node_templates):
190 cps = self._get_virtal_binding_cps(node, node_templates)
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))
197 def _get_virtal_binding_cp_ids(self, node, nodeTemplates):
198 return [x['name'] for x in self._get_virtal_binding_cps(node, nodeTemplates)]
200 def _get_virtal_binding_cps(self, node, nodeTemplates):
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']:
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')]
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')]
224 def _buil_cp_vls(self, node):
225 return [self._build_cp_vl(x) for x in self.model.getRequirementByName(node, 'virtual_link')]
227 def _build_cp_vl(self, req):
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()):
238 def _get_external_cps(self, 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]})
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]})
253 external_cps.append({"key_name": key, "cpd_id": value})
256 def _get_forward_cps(self, 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]})
263 forward_cps.append({"key_name": key, "cpd_id": value})