Fix: bt2: pass _TraceClassConst to destruction listeners

[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 c25154e8f9d9f4fb1584c26f5a83638a3d317d88..a691da30d1492748a950d12cff8ea0a06fb8614d 100644 (file)
--- a/src/bindings/python/bt2/bt2/trace_class.py
+++ b/src/bindings/python/bt2/bt2/trace_class.py
@@ -4,23 +4,24 @@
 # Copyright (c) 2018 Francis Deslauriers <francis.deslauriers@efficios.com>
 # Copyright (c) 2019 Simon Marchi <simon.marchi@efficios.com>
 
-from bt2 import native_bt
+import functools
+import collections.abc
+
+from bt2 import error as bt2_error
+from bt2 import trace as bt2_trace
 from bt2 import utils as bt2_utils
+from bt2 import value as bt2_value
 from bt2 import object as bt2_object
-from bt2 import stream_class as bt2_stream_class
+from bt2 import native_bt
 from bt2 import field_class as bt2_field_class
+from bt2 import stream_class as bt2_stream_class
 from bt2 import integer_range_set as bt2_integer_range_set
-from bt2 import trace as bt2_trace
-from bt2 import value as bt2_value
-import collections.abc
-import functools
-import bt2
 
 
 def _trace_class_destruction_listener_from_native(
     user_listener, handle, trace_class_ptr
 ):
-    trace_class = _TraceClass._create_from_ptr_and_get_ref(trace_class_ptr)
+    trace_class = _TraceClassConst._create_from_ptr_and_get_ref(trace_class_ptr)
     user_listener(trace_class)
     handle._invalidate()
 
@@ -149,7 +150,7 @@ class _TraceClass(_TraceClassConst):
         trace_ptr = native_bt.trace_create(self._ptr)
 
         if trace_ptr is None:
-            raise bt2._MemoryError("cannot create trace class object")
+            raise bt2_error._MemoryError("cannot create trace class object")
 
         trace = bt2_trace._Trace._create_from_ptr(trace_ptr)
 
@@ -281,7 +282,9 @@ class _TraceClass(_TraceClassConst):
 
     def _check_field_class_create_status(self, ptr, type_name):
         if ptr is None:
-            raise bt2._MemoryError("cannot create {} field class".format(type_name))
+            raise bt2_error._MemoryError(
+                "cannot create {} field class".format(type_name)
+            )
 
     @staticmethod
     def _set_field_class_user_attrs(fc, user_attributes):