python: remove internal `import bt2` imports

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

diff --git a/src/bindings/python/bt2/bt2/component.py b/src/bindings/python/bt2/bt2/component.py
index 9deb5721955428a23b15fa096b7813479242c3c5..7357eb7e413ca4774b3e40395ea2d8198fadf5cd 100644 (file)
--- a/src/bindings/python/bt2/bt2/component.py
+++ b/src/bindings/python/bt2/bt2/component.py
@@ -12,8 +12,9 @@ from bt2 import trace_class as bt2_trace_class
 from bt2 import clock_class as bt2_clock_class
 from bt2 import query_executor as bt2_query_executor
 from bt2 import port as bt2_port
+from bt2 import error as bt2_error
+from bt2 import integer_range_set as bt2_integer_range_set
 import sys
-import bt2
 
 
 class _IncompleteUserClass(Exception):
@@ -515,7 +516,7 @@ class _UserComponentType(type):
             )
         elif _UserSinkComponent in bases:
             if not hasattr(cls, "_user_consume"):
-                raise bt2._IncompleteUserClass(
+                raise _IncompleteUserClass(
                     "cannot create component class '{}': missing a _user_consume() method".format(
                         class_name
                     )
@@ -525,14 +526,14 @@ class _UserComponentType(type):
                 cls, comp_cls_name, comp_cls_descr, comp_cls_help
             )
         else:
-            raise bt2._IncompleteUserClass(
+            raise _IncompleteUserClass(
                 "cannot find a known component class base in the bases of '{}'".format(
                     class_name
                 )
             )
 
         if cc_ptr is None:
-            raise bt2._MemoryError(
+            raise bt2_error._MemoryError(
                 "cannot create component class '{}'".format(class_name)
             )
 
@@ -565,21 +566,21 @@ class _UserComponentType(type):
     @staticmethod
     def _bt_set_iterator_class(cls, iter_cls):
         if iter_cls is None:
-            raise bt2._IncompleteUserClass(
+            raise _IncompleteUserClass(
                 "cannot create component class '{}': missing message iterator class".format(
                     cls.__name__
                 )
             )
 
         if not issubclass(iter_cls, bt2_message_iterator._UserMessageIterator):
-            raise bt2._IncompleteUserClass(
+            raise _IncompleteUserClass(
                 "cannot create component class '{}': message iterator class does not inherit bt2._UserMessageIterator".format(
                     cls.__name__
                 )
             )
 
         if not hasattr(iter_cls, "__next__"):
-            raise bt2._IncompleteUserClass(
+            raise _IncompleteUserClass(
                 "cannot create component class '{}': message iterator class is missing a __next__() method".format(
                     cls.__name__
                 )
@@ -588,7 +589,7 @@ class _UserComponentType(type):
         if hasattr(iter_cls, "_user_can_seek_ns_from_origin") and not hasattr(
             iter_cls, "_user_seek_ns_from_origin"
         ):
-            raise bt2._IncompleteUserClass(
+            raise _IncompleteUserClass(
                 "cannot create component class '{}': message iterator class implements _user_can_seek_ns_from_origin but not _user_seek_ns_from_origin".format(
                     cls.__name__
                 )
@@ -597,7 +598,7 @@ class _UserComponentType(type):
         if hasattr(iter_cls, "_user_can_seek_beginning") and not hasattr(
             iter_cls, "_user_seek_beginning"
         ):
-            raise bt2._IncompleteUserClass(
+            raise _IncompleteUserClass(
                 "cannot create component class '{}': message iterator class implements _user_can_seek_beginning but not _user_seek_beginning".format(
                     cls.__name__
                 )
@@ -634,9 +635,9 @@ class _UserComponentType(type):
         # this can raise, but the native side checks the exception
         range_set = cls._user_get_supported_mip_versions(params, obj, log_level)
 
-        if type(range_set) is not bt2.UnsignedIntegerRangeSet:
+        if type(range_set) is not bt2_integer_range_set.UnsignedIntegerRangeSet:
             # this can raise, but the native side checks the exception
-            range_set = bt2.UnsignedIntegerRangeSet(range_set)
+            range_set = bt2_integer_range_set.UnsignedIntegerRangeSet(range_set)
 
         # return new reference
         range_set._get_ref(range_set._ptr)
@@ -669,7 +670,7 @@ class _UserComponentType(type):
             priv_query_exec._invalidate()
 
         # this can raise, but the native side checks the exception
-        results = bt2.create_value(results)
+        results = bt2_value.create_value(results)
 
         if results is None:
             results_ptr = native_bt.value_null
@@ -681,7 +682,7 @@ class _UserComponentType(type):
         return int(results_ptr)
 
     def _user_query(cls, priv_query_executor, object_name, params, method_obj):
-        raise bt2.UnknownObject
+        raise bt2_utils.UnknownObject
 
     def _bt_component_class_ptr(self):
         return self._bt_as_component_class_ptr(self._bt_cc_ptr)
@@ -791,7 +792,7 @@ class _UserComponent(metaclass=_UserComponentType):
         tc_ptr = native_bt.trace_class_create(ptr)
 
         if tc_ptr is None:
-            raise bt2._MemoryError("could not create trace class")
+            raise bt2_error._MemoryError("could not create trace class")
 
         tc = bt2_trace_class._TraceClass._create_from_ptr(tc_ptr)
         tc._assigns_automatic_stream_class_id = assigns_automatic_stream_class_id
@@ -816,7 +817,7 @@ class _UserComponent(metaclass=_UserComponentType):
         cc_ptr = native_bt.clock_class_create(ptr)
 
         if cc_ptr is None:
-            raise bt2._MemoryError("could not create clock class")
+            raise bt2_error._MemoryError("could not create clock class")
 
         cc = bt2_clock_class._ClockClass._create_from_ptr(cc_ptr)