code refactor for genericparser
[modeling/etsicatalog.git] / genericparser / pub / utils / toscaparsers / vnfdmodel.py
1 # Copyright 2017 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 from genericparser.pub.config.config import VNFD_SCHEMA_VERSION_DEFAULT
18 from genericparser.pub.utils.toscaparsers.basemodel import BaseInfoModel
19 from genericparser.pub.utils.toscaparsers.vnfdparser import CreateVnfdSOLParser
20 # from genericparser.pub.exceptions import CatalogException
21
22 logger = logging.getLogger(__name__)
23
24 NFV_VNF_RELATIONSHIPS = [["tosca.relationships.nfv.VirtualLinksTo", "tosca.relationships.nfv.VduAttachesTo", "tosca.relationships.nfv.AttachesTo", "tosca.relationships.nfv.Vdu.AttachedTo", "tosca.relationships.DependsOn"],
25                          ["tosca.nodes.relationships.VirtualBindsTo", "tosca.relationships.nfv.VirtualBindsTo"]]
26
27
28 class EtsiVnfdInfoModel(BaseInfoModel):
29
30     def __init__(self, path, params):
31         self.vnf = {}
32         super(EtsiVnfdInfoModel, self).__init__(path, params)
33
34     def parseModel(self, tosca):
35         self.metadata = self.buildMetadata(tosca)
36         self.inputs = self.buildInputs(tosca)
37         nodeTemplates = map(functools.partial(self.buildNode, tosca=tosca), tosca.nodetemplates)
38         self.basepath = self.get_base_path(tosca)
39         node_types = tosca.topology_template.custom_defs
40         sol_version = self.metadata.get("VNFD_SCHEMA_VERSION", VNFD_SCHEMA_VERSION_DEFAULT) if isinstance(self.metadata, dict) else VNFD_SCHEMA_VERSION_DEFAULT
41         vnfd_sol_parser = CreateVnfdSOLParser(sol_version, self)
42         self.vnf = vnfd_sol_parser.build_vnf(tosca)
43         self.volume_storages = vnfd_sol_parser.get_all_volume_storage(nodeTemplates, node_types)
44         self.vdus = vnfd_sol_parser.get_all_vdu(nodeTemplates, node_types)
45         self.vls = vnfd_sol_parser.get_all_vl(nodeTemplates, node_types)
46         self.cps = vnfd_sol_parser.get_all_cp(nodeTemplates, node_types)
47         self.vnf_exposed = vnfd_sol_parser.get_all_endpoint_exposed()
48         self.graph = self.get_deploy_graph(tosca, NFV_VNF_RELATIONSHIPS)