node_hierarchy/cloud/Link.py

class Link(object):
    def __init__(self, LinkJsonObject, nodes):
        self.__jsonObject = LinkJsonObject
        self.linkType, self.isVpn = self.__getLinkType__()
        self.__nodes = nodes

    def __getLinkType__(self):
        types = [x['type'] for x in self.__jsonObject]
        ltype = types[0]
        lvpn = False
        for x in types:
            if x != 'unknown' and x != 'other':
                if x == 'l2tp' or x == 'tunnel':
                    lvpn = True
                val = x
        return ltype, lvpn

    def getEndpointNodes(self, getGateways = False):
        return self.__nodes

    def getEndpointNodeIDs(self, getGateways = True):
        return [x.nodeID for x in self.__nodes]

    def isNodeIDinLink(self, nodeID):
        for x in self.__nodes:
            if nodeID == x.nodeID:
                return True
        return False

    def isNodeInLink(self, node):
        return self.isNodeIDinLink(node.nodeID)