aboutsummaryrefslogtreecommitdiffstats
path: root/catalog/pub/utils/toscaparser/pnfmodel.py
blob: 1acf6d4c44f4765e23944e0fb97881f785e88c2f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# Copyright 2018 ZTE Corporation.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import functools
import logging
from catalog.pub.utils.toscaparser.basemodel import BaseInfoModel
logger = logging.getLogger(__name__)


class PnfdInfoModel(BaseInfoModel):

    def __init__(self, path, params):
        super(PnfdInfoModel, self).__init__(path, params)

    def parseModel(self, tosca):
        self.metadata = self.buildMetadata(tosca)
        self.inputs = self.buildInputs(tosca)
        nodeTemplates = map(functools.partial(self.buildNode, tosca=tosca),
                            tosca.nodetemplates)
        self.basepath = self.get_base_path(tosca)
        self.pnf = {}
        self.get_all_cp(nodeTemplates)

    def get_substitution_mappings(self, tosca):
        pnf_substitution_mappings = tosca.tpl['topology_template']['substitution_mappings']
        if pnf_substitution_mappings:
            self.pnf['type'] = pnf_substitution_mappings['node_type']
            self.pnf['properties'] = pnf_substitution_mappings['properties']

    def get_all_cp(self, nodeTemplates):
        self.pnf['ExtPorts'] = []
        for node in nodeTemplates:
            if self.isPnfExtPort(node):
                cp = {}
                cp['id'] = node['name']
                cp['type'] = node['nodeType']
                cp['properties'] = node['properties']
                self.pnf['ExtPorts'].append(cp)

    def isPnfExtPort(self, node):
        return node['nodeType'].find('tosca.nodes.nfv.PnfExtPort') >= 0