53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
class Link(object):
|
|
def __init__(self, LinkJsonObject, srcNode, dstNode):
|
|
self.__jsonObject__ = LinkJsonObject
|
|
self.__srcNode__ = srcNode
|
|
self.__dstNode__ = dstNode
|
|
self.linkType = self.__getLinkType__()
|
|
self.isVpn = self.__getLinkVpnState__()
|
|
|
|
|
|
def __getLinkType__(self):
|
|
if self.__srcNode__ != None:
|
|
for k, v in self.__srcNode__.interfaces.items():
|
|
if self.__jsonObject__['source']['interface_mac'] in v:
|
|
return k
|
|
if self.__dstNode__ != None:
|
|
for k, v in self.__dstNode__.interfaces.items():
|
|
if self.__jsonObject__['target']['interface_mac'] in v:
|
|
return k
|
|
return 'unknown'
|
|
|
|
def __getLinkVpnState__(self):
|
|
if self.__jsonObject__['vpn'] == True:
|
|
return True
|
|
for node in self.getEndpointNodes(getGateways = True):
|
|
if node.isGateway == True:
|
|
return True
|
|
return False
|
|
|
|
def getEndpointNodes(self, getGateways = False):
|
|
nodes = []
|
|
if self.__srcNode__ != None:
|
|
if getGateways == True or self.__srcNode__.isGateway == False:
|
|
nodes.append(self.__srcNode__)
|
|
if self.__dstNode__ != None:
|
|
if getGateways == True or self.__dstNode__.isGateway == False:
|
|
nodes.append(self.__dstNode__)
|
|
return nodes
|
|
|
|
def getEndpointNodeIDs(self, getGateways = True):
|
|
nodeIDs = []
|
|
for node in self.getEndpointNodes(getGateways):
|
|
nodeIDs.append(node.nodeID)
|
|
return nodeIDs
|
|
|
|
def isNodeIDinLink(self, nodeID):
|
|
for endpoint in self.getEndpointNodes():
|
|
if endpoint.nodeID == nodeID:
|
|
return True
|
|
return False
|
|
|
|
def isNodeInLink(self, node):
|
|
return self.isNodeIDinLink(node.nodeID)
|