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