diff options
Diffstat (limited to 'miasm2/core/graph.py')
| -rw-r--r-- | miasm2/core/graph.py | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py index ad159771..d29517cc 100644 --- a/miasm2/core/graph.py +++ b/miasm2/core/graph.py @@ -27,13 +27,16 @@ class DiGraph: self._nodes_to[n] = [] self._nodes_from[n] = [] - def del_node(self, n): - if n in self._nodes: - self._nodes.remove(n) - for pred in self.predecessors(n): - self.del_edge(pred, n) - for succ in self.successors(n): - self.del_edge(n, succ) + def del_node(self, node): + """Delete the @node of the graph; Also delete every edge to/from this + @node""" + + if node in self._nodes: + self._nodes.remove(node) + for pred in self.predecessors(node): + self.del_edge(pred, node) + for succ in self.successors(node): + self.del_edge(node, succ) def add_edge(self, a, b): if not a in self._nodes: |