bt2: add interrupter support

[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..e9659bd2f242bedbad3509a567be476507b36fb8 100644 (file)
--- a/src/bindings/python/bt2/bt2/graph.py
+++ b/src/bindings/python/bt2/bt2/graph.py
@@ -73,7 +73,7 @@ class Graph(object._SharedObject):
         ptr = native_bt.graph_create()
 
         if ptr is None:
-            raise bt2.CreationError('cannot create graph object')
+            raise bt2._MemoryError('cannot create graph object')
 
         super().__init__(ptr)
 
@@ -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):
@@ -194,6 +196,6 @@ class Graph(object._SharedObject):
         )
 
         if msg_iter_ptr is None:
-            raise bt2.CreationError('cannot create output port message iterator')
+            raise bt2._MemoryError('cannot create output port message iterator')
 
         return bt2.message_iterator._OutputPortMessageIterator(msg_iter_ptr)