def add_edge(self,*args,**kwargs): if len(args)==1 and isinstance(args[0],Edge): edge = args[0] else: edge = Edge(*args,**kwargs) #end if src = edge.get_source() dst = edge.get_destination() self.edges[src,dst] = edge self.graph.add_edge(edge)
def delete_edge(self, edge: pydot.Edge): self.nodes_to_edges[edge.get_source()].remove(edge) self.nodes_to_edges[edge.get_destination()].remove(edge) self.graph.del_edge(edge.get_source(), edge.get_destination())
def add_edge(self, edge: pydot.Edge): self.nodes_to_edges[edge.get_source()] += [edge] self.nodes_to_edges[edge.get_destination()] += [edge] self.graph.add_edge(edge)