From 7e2ad0109bb191b7e83f4e779b39440664ce4b5c Mon Sep 17 00:00:00 2001 From: Ajax Date: Mon, 25 Jan 2016 10:59:48 +0100 Subject: Graph: add_node return the previous status --- miasm2/core/graph.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'miasm2') diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py index ee5dc418..8199c22a 100644 --- a/miasm2/core/graph.py +++ b/miasm2/core/graph.py @@ -52,11 +52,16 @@ class DiGraph(object): sorted(self._edges) == sorted(graph.edges()))) def add_node(self, node): + """Add the node @node to the graph. + If the node was already present, return False. + Otherwise, return True + """ if node in self._nodes: - return + return False self._nodes.add(node) self._nodes_succ[node] = [] self._nodes_pred[node] = [] + return True def del_node(self, node): """Delete the @node of the graph; Also delete every edge to/from this -- cgit 1.4.1