0a1568ba38
Change the file structure to make appropriate for stand alone parser. Change-Id: I6d5259b61b1cd82949262fee4930fa66703ad036
47 lines
1.7 KiB
Python
47 lines
1.7 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):
|
|
if node not in self.vertices:
|
|
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 rel, nodetpls in relation.items():
|
|
for tpl in self.nodetemplates:
|
|
if tpl.name == nodetpls.name:
|
|
self._create_edge(node, tpl, rel)
|
|
self._create_vertex(node)
|