bt2: import public names into `__init__.py`

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

diff --git a/src/bindings/python/bt2/bt2/trace_class.py b/src/bindings/python/bt2/bt2/trace_class.py
index 7e7c179770508058846e8b9295e6bff49c839520..5c4da5c7086670b3e4d2e4e71cd5a6b1bff39bf3 100644 (file)
--- a/src/bindings/python/bt2/bt2/trace_class.py
+++ b/src/bindings/python/bt2/bt2/trace_class.py
@@ -28,6 +28,7 @@ import bt2
 from bt2 import native_bt, utils, object
 import bt2.stream_class
 import bt2.field_class
+import bt2.trace
 import collections.abc
 import functools
 
@@ -69,7 +70,7 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
         trace_ptr = native_bt.trace_create(self._ptr)
 
         if trace_ptr is None:
-            raise bt2.CreationError('cannot create trace class object')
+            raise bt2._MemoryError('cannot create trace class object')
 
         trace = bt2.trace._Trace._create_from_ptr(trace_ptr)
 
@@ -204,7 +205,7 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
 
     def _check_create_status(self, ptr, type_name):
         if ptr is None:
-            raise bt2.CreationError('cannot create {} field class'.format(type_name))
+            raise bt2._MemoryError('cannot create {} field class'.format(type_name))
 
     def _create_integer_field_class(
         self, create_func, py_cls, type_name, field_value_range, preferred_display_base
@@ -333,10 +334,9 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
             _trace_class_destruction_listener_from_native, listener
         )
 
-        listener_id = fn(self._ptr, listener_from_native)
-        if listener_id is None:
-            utils._raise_bt2_error(
-                'cannot add destruction listener to trace class object'
-            )
+        status, listener_id = fn(self._ptr, listener_from_native)
+        utils._handle_func_status(
+            status, 'cannot add destruction listener to trace class object'
+        )
 
         return bt2._ListenerHandle(listener_id, self)