heat-translator/translator/toscalib/tpl_relationship_graph.py
Ton Ngo da3373aa01 New custom type, unit test and template for use case
Add TOSCA template for testing:  this represents the first part
of the monitoring use case, deploying mongodb and nodejs onto
2 servers.  A nodejs application is also installed and started.
For testing, the nodejs application is set as sample, but for real
deployment, change it to some real application like:
https://github.com/paypal/rest-api-sample-app-nodejs.git
The new unit test will translate to Heat and validate the output.
Also include change for the keyword input -> inputs in the
template.

Change-Id: Ia2eabe030669fc0b0f52a9c2c1cc754a4b68a394
2015-02-23 10:13:47 -08:00

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)