47 lines
2 KiB
Python
47 lines
2 KiB
Python
|
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)
|