bt2: clean available `bt2` package names
[babeltrace.git] / src / bindings / python / bt2 / bt2 / component.py
index efbfc3439a1283f0a0276e238edfc98014bdf44f..d26adc8f8b06081e63bf1251d4722af325426163 100644 (file)
 # THE SOFTWARE.
 
 from bt2 import native_bt, object, utils
-import bt2.message_iterator
+from bt2 import message_iterator as bt2_message_iterator
 import collections.abc
-import bt2.value
-import bt2.trace_class
+from bt2 import value as bt2_value
+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
 import traceback
-import bt2.port
+from bt2 import port as bt2_port
 import sys
 import bt2
 import os
@@ -247,7 +249,7 @@ class _GenericSourceComponent(object._SharedObject, _SourceComponent):
             native_bt.component_source_borrow_output_port_by_name_const,
             native_bt.component_source_borrow_output_port_by_index_const,
             native_bt.component_source_get_output_port_count,
-            bt2.port._OutputPort,
+            bt2_port._OutputPort,
         )
 
 
@@ -264,7 +266,7 @@ class _GenericFilterComponent(object._SharedObject, _FilterComponent):
             native_bt.component_filter_borrow_output_port_by_name_const,
             native_bt.component_filter_borrow_output_port_by_index_const,
             native_bt.component_filter_get_output_port_count,
-            bt2.port._OutputPort,
+            bt2_port._OutputPort,
         )
 
     @property
@@ -274,7 +276,7 @@ class _GenericFilterComponent(object._SharedObject, _FilterComponent):
             native_bt.component_filter_borrow_input_port_by_name_const,
             native_bt.component_filter_borrow_input_port_by_index_const,
             native_bt.component_filter_get_input_port_count,
-            bt2.port._InputPort,
+            bt2_port._InputPort,
         )
 
 
@@ -291,7 +293,7 @@ class _GenericSinkComponent(object._SharedObject, _SinkComponent):
             native_bt.component_sink_borrow_input_port_by_name_const,
             native_bt.component_sink_borrow_input_port_by_index_const,
             native_bt.component_sink_get_input_port_count,
-            bt2.port._InputPort,
+            bt2_port._InputPort,
         )
 
 
@@ -515,7 +517,7 @@ class _UserComponentType(type):
 
         # call user's __init__() method
         if params_ptr is not None:
-            params = bt2.value._create_from_ptr_and_get_ref(params_ptr)
+            params = bt2_value._create_from_ptr_and_get_ref(params_ptr)
         else:
             params = None
 
@@ -536,7 +538,7 @@ class _UserComponentType(type):
                 )
             )
 
-        if not issubclass(iter_cls, bt2.message_iterator._UserMessageIterator):
+        if not issubclass(iter_cls, bt2_message_iterator._UserMessageIterator):
             raise bt2._IncompleteUserClass(
                 "cannot create component class '{}': message iterator class does not inherit bt2._UserMessageIterator".format(
                     cls.__name__
@@ -575,11 +577,13 @@ class _UserComponentType(type):
         # this can raise, in which case the native call to
         # bt_component_class_query() returns NULL
         if params_ptr is not None:
-            params = bt2.value._create_from_ptr_and_get_ref(params_ptr)
+            params = bt2_value._create_from_ptr_and_get_ref(params_ptr)
         else:
             params = None
 
-        query_exec = bt2.QueryExecutor._create_from_ptr_and_get_ref(query_exec_ptr)
+        query_exec = bt2_query_executor.QueryExecutor._create_from_ptr_and_get_ref(
+            query_exec_ptr
+        )
 
         # this can raise, but the native side checks the exception
         results = cls._query(query_exec, obj, params, log_level)
@@ -594,7 +598,7 @@ class _UserComponentType(type):
             results_ptr = results._ptr
 
         # We return a new reference.
-        bt2.value._Value._get_ref(results_ptr)
+        bt2_value._Value._get_ref(results_ptr)
 
         return int(results_ptr)
 
@@ -661,14 +665,14 @@ class _UserComponent(metaclass=_UserComponentType):
     def _bt_port_connected_from_native(
         self, self_port_ptr, self_port_type, other_port_ptr
     ):
-        port = bt2.port._create_self_from_ptr_and_get_ref(self_port_ptr, self_port_type)
+        port = bt2_port._create_self_from_ptr_and_get_ref(self_port_ptr, self_port_type)
 
         if self_port_type == native_bt.PORT_TYPE_OUTPUT:
             other_port_type = native_bt.PORT_TYPE_INPUT
         else:
             other_port_type = native_bt.PORT_TYPE_OUTPUT
 
-        other_port = bt2.port._create_from_ptr_and_get_ref(
+        other_port = bt2_port._create_from_ptr_and_get_ref(
             other_port_ptr, other_port_type
         )
         self._port_connected(port, other_port)
@@ -680,7 +684,7 @@ class _UserComponent(metaclass=_UserComponentType):
         if tc_ptr is None:
             raise bt2._MemoryError('could not create trace class')
 
-        tc = bt2.trace_class._TraceClass._create_from_ptr(tc_ptr)
+        tc = bt2_trace_class._TraceClass._create_from_ptr(tc_ptr)
         tc._assigns_automatic_stream_class_id = assigns_automatic_stream_class_id
 
         return tc
@@ -701,7 +705,7 @@ class _UserComponent(metaclass=_UserComponentType):
         if cc_ptr is None:
             raise bt2._MemoryError('could not create clock class')
 
-        cc = bt2.clock_class._ClockClass._create_from_ptr(cc_ptr)
+        cc = bt2_clock_class._ClockClass._create_from_ptr(cc_ptr)
 
         if frequency is not None:
             cc._frequency = frequency
@@ -745,7 +749,7 @@ class _UserSourceComponent(_UserComponent, _SourceComponent):
             native_bt.self_component_source_borrow_output_port_by_name,
             native_bt.self_component_source_borrow_output_port_by_index,
             get_output_port_count,
-            bt2.port._UserComponentOutputPort,
+            bt2_port._UserComponentOutputPort,
         )
 
     def _add_output_port(self, name, user_data=None):
@@ -756,7 +760,7 @@ class _UserSourceComponent(_UserComponent, _SourceComponent):
             comp_status, 'cannot add output port to source component object'
         )
         assert self_port_ptr is not None
-        return bt2.port._UserComponentOutputPort._create_from_ptr(self_port_ptr)
+        return bt2_port._UserComponentOutputPort._create_from_ptr(self_port_ptr)
 
 
 class _UserFilterComponent(_UserComponent, _FilterComponent):
@@ -778,7 +782,7 @@ class _UserFilterComponent(_UserComponent, _FilterComponent):
             native_bt.self_component_filter_borrow_output_port_by_name,
             native_bt.self_component_filter_borrow_output_port_by_index,
             get_output_port_count,
-            bt2.port._UserComponentOutputPort,
+            bt2_port._UserComponentOutputPort,
         )
 
     @property
@@ -792,7 +796,7 @@ class _UserFilterComponent(_UserComponent, _FilterComponent):
             native_bt.self_component_filter_borrow_input_port_by_name,
             native_bt.self_component_filter_borrow_input_port_by_index,
             get_input_port_count,
-            bt2.port._UserComponentInputPort,
+            bt2_port._UserComponentInputPort,
         )
 
     def _add_output_port(self, name, user_data=None):
@@ -803,7 +807,7 @@ class _UserFilterComponent(_UserComponent, _FilterComponent):
             comp_status, 'cannot add output port to filter component object'
         )
         assert self_port_ptr
-        return bt2.port._UserComponentOutputPort._create_from_ptr(self_port_ptr)
+        return bt2_port._UserComponentOutputPort._create_from_ptr(self_port_ptr)
 
     def _add_input_port(self, name, user_data=None):
         utils._check_str(name)
@@ -813,7 +817,7 @@ class _UserFilterComponent(_UserComponent, _FilterComponent):
             comp_status, 'cannot add input port to filter component object'
         )
         assert self_port_ptr
-        return bt2.port._UserComponentInputPort._create_from_ptr(self_port_ptr)
+        return bt2_port._UserComponentInputPort._create_from_ptr(self_port_ptr)
 
 
 class _UserSinkComponent(_UserComponent, _SinkComponent):
@@ -841,7 +845,7 @@ class _UserSinkComponent(_UserComponent, _SinkComponent):
             native_bt.self_component_sink_borrow_input_port_by_name,
             native_bt.self_component_sink_borrow_input_port_by_index,
             get_input_port_count,
-            bt2.port._UserComponentInputPort,
+            bt2_port._UserComponentInputPort,
         )
 
     def _add_input_port(self, name, user_data=None):
@@ -852,10 +856,10 @@ class _UserSinkComponent(_UserComponent, _SinkComponent):
             comp_status, 'cannot add input port to sink component object'
         )
         assert self_port_ptr
-        return bt2.port._UserComponentInputPort._create_from_ptr(self_port_ptr)
+        return bt2_port._UserComponentInputPort._create_from_ptr(self_port_ptr)
 
     def _create_input_port_message_iterator(self, input_port):
-        utils._check_type(input_port, bt2.port._UserComponentInputPort)
+        utils._check_type(input_port, bt2_port._UserComponentInputPort)
 
         msg_iter_ptr = native_bt.self_component_port_input_message_iterator_create_from_sink_component(
             self._bt_ptr, input_port._ptr
@@ -864,7 +868,7 @@ class _UserSinkComponent(_UserComponent, _SinkComponent):
         if msg_iter_ptr is None:
             raise bt2.CreationError('cannot create message iterator object')
 
-        return bt2.message_iterator._UserComponentInputPortMessageIterator(msg_iter_ptr)
+        return bt2_message_iterator._UserComponentInputPortMessageIterator(msg_iter_ptr)
 
     @property
     def _is_interrupted(self):
This page took 0.027994 seconds and 4 git commands to generate.