2933ff038e
TOSCA node template can have multiple relationships so property name providing relationship should be plural. Closes-Bug: #1404109 Change-Id: I12892b72f6b05b9b26ee1694c8d144e6beade524
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
# 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.
|
|
|
|
|
|
class ToscaGraph(object):
|
|
'''Graph of Tosca Node Templates.'''
|
|
def __init__(self, nodetemplates):
|
|
self.nodetemplates = nodetemplates
|
|
self.vertices = {}
|
|
self._create()
|
|
|
|
def _create_vertex(self, node):
|
|
self.vertices[node.name] = node
|
|
|
|
def _create_edge(self, node1, node2, relationship):
|
|
if node1 not in self.vertices:
|
|
self._create_vertex(node1)
|
|
self.vertices[node1.name]._add_next(node2,
|
|
relationship)
|
|
|
|
def vertex(self, node):
|
|
if node in self.vertices:
|
|
return self.vertices[node]
|
|
|
|
def __iter__(self):
|
|
return iter(self.vertices.values())
|
|
|
|
def _create(self):
|
|
for node in self.nodetemplates:
|
|
relation = node.relationships
|
|
if relation:
|
|
for relation, nodetpls in relation.items():
|
|
for tpl in self.nodetemplates:
|
|
if tpl.name == nodetpls.name:
|
|
self._create_edge(node, tpl, relation)
|
|
self._create_vertex(node)
|