bt2: make bt2 add graph listener wrappers append error causes

[babeltrace.git] / src / bindings / python / bt2 / bt2 / graph.py

diff --git a/src/bindings/python/bt2/bt2/graph.py b/src/bindings/python/bt2/bt2/graph.py
index 1f280ba119ff4de691398980b9c4cd6e54f9eb2c..870ddb381956406c6e79b878c8baa8db178f2e22 100644 (file)
--- a/src/bindings/python/bt2/bt2/graph.py
+++ b/src/bindings/python/bt2/bt2/graph.py
@@ -147,7 +147,8 @@ class Graph(object._SharedObject):
 
         listener_ids = fn(self._ptr, listener_from_native)
         if listener_ids is None:
-            utils._raise_bt2_error('cannot add listener to graph object')
+            raise bt2.Error('cannot add listener to graph object')
+
         return bt2._ListenerHandle(listener_ids, self)
 
     def add_ports_connected_listener(self, listener):
@@ -161,7 +162,8 @@ class Graph(object._SharedObject):
 
         listener_ids = fn(self._ptr, listener_from_native)
         if listener_ids is None:
-            utils._raise_bt2_error('cannot add listener to graph object')
+            raise bt2.Error('cannot add listener to graph object')
+
         return bt2._ListenerHandle(listener_ids, self)
 
     def run(self):