node_hierarchy/cloud/GlobalGraph.py

47 lines
2 KiB
Python
Raw Normal View History

2016-05-29 18:56:50 +02:00
from cloud.LocalGraph import LocalGraph
from cloud.Graph import Graph
from exceptions.HieraException import HieraException
class GlobalGraph(Graph):
def __init__(self, nodes, links, debugPrint = False):
super().__init__(nodes, links)
self.__enableDebugPrinting__ = debugPrint
self.localGraphs = self.__buildLocalGraphs__()
if self.__enableDebugPrinting__:
self.__debugPrint__()
def __buildLocalGraphs__(self):
nodeIDs = self.getListOfNodeIDsOnline()
localGraphs = []
while len(nodeIDs) > 0:
connectedNodes = self.__getConnectedNodes__(nodeIDs[0])
try:
localGraphs.append(self.__createLocalCloudByNodesList__(connectedNodes))
print('Create LocalGraph object #',len(localGraphs), '\r',end = '')
except HieraException:
print('Was not able to add local cloud, because no VPN link was found.')
nodeIDs = [x for x in nodeIDs if x not in connectedNodes]
print('')
return localGraphs
def __getConnectedNodes__(self, nodeID, trace = []):
neighNodeIDs = self.getNeighbourNodeIDsForNodeID(nodeID)
trace_new = trace[:] + [x for x in neighNodeIDs if x not in trace]
for neighNodeID in neighNodeIDs:
if neighNodeID not in trace:
trace_new = trace_new + [x for x in self.__getConnectedNodes__(neighNodeID, trace_new) if x not in trace_new]
return trace_new
def __createLocalCloudByNodesList__(self, nodesIDList):
nodes = {}
links = []
for nodeID in nodesIDList:
nodes[nodeID] = self.__nodes__[nodeID]
links = links + [x for x in self.getLinksByNodeID(nodeID) if x not in links]
return LocalGraph(nodes, links, self.__enableDebugPrinting__)
def __debugPrint__(self):
for localGraph in self.localGraphs:
for node in localGraph.getNodesWithNoDependencies():
print(node.hostname, node.publicIPv6Addresses)