*_string() enumerator to string functions: remove common prefix

[babeltrace.git] / src / bindings / python / bt2 / bt2 / __init__.py.in

diff --git a/src/bindings/python/bt2/bt2/__init__.py.in b/src/bindings/python/bt2/bt2/__init__.py.in
index 7859a22b459889f3b9265df2cbb5928ad346b789..e18188b8d7623eec9b34e044e3e06824a827e0ab 100644 (file)
--- a/src/bindings/python/bt2/bt2/__init__.py.in
+++ b/src/bindings/python/bt2/bt2/__init__.py.in
@@ -23,6 +23,7 @@
 __version__ = '@PACKAGE_VERSION@'
 
 
+# import all public names
 from bt2.clock_class import ClockClassOffset
 from bt2.clock_snapshot import _ClockSnapshot
 from bt2.clock_snapshot import _UnknownClockSnapshot
@@ -73,7 +74,7 @@ from bt2.field_class import _VariantFieldClassWithSignedSelector
 from bt2.field_class import _ArrayFieldClass
 from bt2.field_class import _StaticArrayFieldClass
 from bt2.field_class import _DynamicArrayFieldClass
-from bt2.field_path import Scope
+from bt2.field_path import FieldPathScope
 from bt2.field_path import _IndexFieldPathItem
 from bt2.field_path import _CurrentArrayElementFieldPathItem
 from bt2.graph import Graph
@@ -114,6 +115,48 @@ from bt2.value import ArrayValue
 from bt2.value import MapValue
 
 
+def _del_global_name(name):
+    if name in globals():
+        del globals()[name]
+
+
+# remove private module names from the package
+_del_global_name('_native_bt')
+_del_global_name('clock_class')
+_del_global_name('clock_snapshot')
+_del_global_name('component')
+_del_global_name('connection')
+_del_global_name('error')
+_del_global_name('event')
+_del_global_name('event_class')
+_del_global_name('field')
+_del_global_name('field_class')
+_del_global_name('field_path')
+_del_global_name('graph')
+_del_global_name('integer_range_set')
+_del_global_name('interrupter')
+_del_global_name('logging')
+_del_global_name('message')
+_del_global_name('message_iterator')
+_del_global_name('native_bt')
+_del_global_name('object')
+_del_global_name('packet')
+_del_global_name('plugin')
+_del_global_name('port')
+_del_global_name('py_plugin')
+_del_global_name('query_executor')
+_del_global_name('stream')
+_del_global_name('stream_class')
+_del_global_name('trace')
+_del_global_name('trace_class')
+_del_global_name('trace_collection_message_iterator')
+_del_global_name('utils')
+_del_global_name('value')
+
+# remove private `_del_global_name` name from the package
+del _del_global_name
+
+
 class _MemoryError(_Error):
     '''Raised when an operation fails due to memory issues.'''
 
@@ -122,7 +165,7 @@ class InvalidObject(Exception):
     pass
 
 
-class _OverflowError(OverflowError):
+class _OverflowError(_Error, OverflowError):
     pass
 
 
@@ -138,30 +181,15 @@ class _IncompleteUserClass(Exception):
     pass
 
 
-class _ListenerHandle:
-    def __init__(self, listener_id, obj):
-        self._listener_id = listener_id
-        self._obj = obj
-
-
 def _init_and_register_exit():
-    import bt2.native_bt as _native_bt
+    from bt2 import native_bt
     import atexit
 
-    atexit.register(_native_bt.bt2_cc_exit_handler)
-    version = (
-        _native_bt.version_get_major(),
-        _native_bt.version_get_minor(),
-        _native_bt.version_get_patch(),
-        _native_bt.version_get_extra(),
-    )
-    _native_bt.bt2_cc_init_from_bt2()
+    atexit.register(native_bt.bt2_cc_exit_handler)
+    native_bt.bt2_cc_init_from_bt2()
 
 
 _init_and_register_exit()
 
-
-try:
-    del native_bt
-except:
-    pass
+# remove private `_init_and_register_exit` name from the package
+del _init_and_register_exit