Source code for Edge
#-*- coding: utf-8 -*-
from Item import *
[docs]class Edge(Item):
"""
"""
def __init__(self, node1, node2, directed = False):
"""
"""
super(Edge, self).__init__()
self.node1 = node1
self.node2 = node2
self.node1._edges.append(self)
self.node2._edges.append(self)
self._directed = directed
[docs] def node(self, node):
"""
Return the other node
"""
if node == self.node1:
return self.node2
elif node == self.node2:
return self.node1
else:
return None
[docs] def parent(self):
"""
"""
return self.node1
[docs] def child(self):
"""
"""
return self.node2
[docs] def directed(self, dir):
"""
"""
return self._directed
[docs] def set_directed(self, dir):
"""
"""
self._directed = dir