python: remove internal `import bt2` imports
[babeltrace.git] / src / bindings / python / bt2 / bt2 / value.py
index 77d1487979d4228d90187fe29e0978864ac1115e..d2cb4e076a11adc4f86eafecb9a9de3910a535c0 100644 (file)
@@ -2,13 +2,15 @@
 #
 # Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
 
-from bt2 import native_bt, object, utils
+from bt2 import native_bt
+from bt2 import object as bt2_object
+from bt2 import utils as bt2_utils
+from bt2 import error as bt2_error
 import collections.abc
 import functools
 import numbers
 import math
 import abc
-import bt2
 
 
 def _create_from_ptr_template(ptr, object_map):
@@ -81,9 +83,15 @@ def create_value(value):
     )
 
 
-class _ValueConst(object._SharedObject, metaclass=abc.ABCMeta):
-    _get_ref = staticmethod(native_bt.value_get_ref)
-    _put_ref = staticmethod(native_bt.value_put_ref)
+class _ValueConst(bt2_object._SharedObject, metaclass=abc.ABCMeta):
+    @staticmethod
+    def _get_ref(ptr):
+        native_bt.value_get_ref(ptr)
+
+    @staticmethod
+    def _put_ref(ptr):
+        native_bt.value_put_ref(ptr)
+
     _create_value_from_ptr = staticmethod(_create_from_const_ptr)
     _create_value_from_ptr_and_get_ref = staticmethod(
         _create_from_const_ptr_and_get_ref
@@ -94,8 +102,8 @@ class _ValueConst(object._SharedObject, metaclass=abc.ABCMeta):
 
     def _check_create_status(self, ptr):
         if ptr is None:
-            raise bt2._MemoryError(
-                'cannot create {} value object'.format(self._NAME.lower())
+            raise bt2_error._MemoryError(
+                "cannot create {} value object".format(self._NAME.lower())
             )
 
 
@@ -247,7 +255,7 @@ class _IntegralValue(_IntegralValueConst, _NumericValue):
 
 
 class _BoolValueConst(_IntegralValueConst):
-    _NAME = 'Const boolean'
+    _NAME = "Const boolean"
 
     def __bool__(self):
         return self._value
@@ -262,7 +270,7 @@ class _BoolValueConst(_IntegralValueConst):
 
 
 class BoolValue(_BoolValueConst, _IntegralValue):
-    _NAME = 'Boolean'
+    _NAME = "Boolean"
 
     def __init__(self, value=None):
         if value is None:
@@ -312,7 +320,7 @@ class _IntegerValue(_IntegerValueConst, _IntegralValue):
     @classmethod
     def _value_to_int(cls, value):
         if not isinstance(value, numbers.Integral):
-            raise TypeError('expecting an integral number object')
+            raise TypeError("expecting an integral number object")
 
         value = int(value)
         cls._check_int_range(value)
@@ -325,33 +333,33 @@ class _IntegerValue(_IntegerValueConst, _IntegralValue):
 
 
 class _UnsignedIntegerValueConst(_IntegerValueConst):
-    _NAME = 'Const unsigned integer'
+    _NAME = "Const unsigned integer"
     _get_value = staticmethod(native_bt.value_integer_unsigned_get)
 
 
 class UnsignedIntegerValue(_UnsignedIntegerValueConst, _IntegerValue):
-    _NAME = 'Unsigned integer'
-    _check_int_range = staticmethod(utils._check_uint64)
+    _NAME = "Unsigned integer"
+    _check_int_range = staticmethod(bt2_utils._check_uint64)
     _create_default_value = staticmethod(native_bt.value_integer_unsigned_create)
     _create_value = staticmethod(native_bt.value_integer_unsigned_create_init)
     _set_value = staticmethod(native_bt.value_integer_unsigned_set)
 
 
 class _SignedIntegerValueConst(_IntegerValueConst):
-    _NAME = 'Const signed integer'
+    _NAME = "Const signed integer"
     _get_value = staticmethod(native_bt.value_integer_signed_get)
 
 
 class SignedIntegerValue(_SignedIntegerValueConst, _IntegerValue):
-    _NAME = 'Signed integer'
-    _check_int_range = staticmethod(utils._check_int64)
+    _NAME = "Signed integer"
+    _check_int_range = staticmethod(bt2_utils._check_int64)
     _create_default_value = staticmethod(native_bt.value_integer_signed_create)
     _create_value = staticmethod(native_bt.value_integer_signed_create_init)
     _set_value = staticmethod(native_bt.value_integer_signed_set)
 
 
 class _RealValueConst(_NumericValueConst, numbers.Real):
-    _NAME = 'Const real number'
+    _NAME = "Const real number"
 
     @property
     def _value(self):
@@ -359,7 +367,7 @@ class _RealValueConst(_NumericValueConst, numbers.Real):
 
 
 class RealValue(_RealValueConst, _NumericValue):
-    _NAME = 'Real number'
+    _NAME = "Real number"
 
     def __init__(self, value=None):
         if value is None:
@@ -386,14 +394,14 @@ class RealValue(_RealValueConst, _NumericValue):
 
 @functools.total_ordering
 class _StringValueConst(collections.abc.Sequence, _Value):
-    _NAME = 'Const string'
+    _NAME = "Const string"
 
     @classmethod
     def _value_to_str(cls, value):
         if isinstance(value, _StringValueConst):
             value = value._value
 
-        utils._check_str(value)
+        bt2_utils._check_str(value)
         return value
 
     @property
@@ -429,7 +437,7 @@ class _StringValueConst(collections.abc.Sequence, _Value):
 
 
 class StringValue(_StringValueConst, _Value):
-    _NAME = 'String'
+    _NAME = "String"
 
     def __init__(self, value=None):
         if value is None:
@@ -442,7 +450,7 @@ class StringValue(_StringValueConst, _Value):
 
     def _set_value(self, value):
         status = native_bt.value_string_set(self._ptr, self._value_to_str(value))
-        utils._handle_func_status(status)
+        bt2_utils._handle_func_status(status)
 
     value = property(fset=_set_value)
 
@@ -464,7 +472,7 @@ class _Container(_ContainerConst):
 
 
 class _ArrayValueConst(_ContainerConst, collections.abc.Sequence, _ValueConst):
-    _NAME = 'Const array'
+    _NAME = "Const array"
     _borrow_element_by_index = staticmethod(
         native_bt.value_array_borrow_element_by_index_const
     )
@@ -501,7 +509,7 @@ class _ArrayValueConst(_ContainerConst, collections.abc.Sequence, _ValueConst):
         index = int(index)
 
         if index < 0 or index >= len(self):
-            raise IndexError('array value object index is out of range')
+            raise IndexError("array value object index is out of range")
 
     def __getitem__(self, index):
         self._check_index(index)
@@ -510,11 +518,11 @@ class _ArrayValueConst(_ContainerConst, collections.abc.Sequence, _ValueConst):
         return self._create_value_from_ptr_and_get_ref(ptr)
 
     def __repr__(self):
-        return '[{}]'.format(', '.join([repr(v) for v in self]))
+        return "[{}]".format(", ".join([repr(v) for v in self]))
 
 
 class ArrayValue(_ArrayValueConst, _Container, collections.abc.MutableSequence, _Value):
-    _NAME = 'Array'
+    _NAME = "Array"
     _borrow_element_by_index = staticmethod(
         native_bt.value_array_borrow_element_by_index
     )
@@ -540,7 +548,7 @@ class ArrayValue(_ArrayValueConst, _Container, collections.abc.MutableSequence,
             ptr = value._ptr
 
         status = native_bt.value_array_set_element_by_index(self._ptr, index, ptr)
-        utils._handle_func_status(status)
+        bt2_utils._handle_func_status(status)
 
     def append(self, value):
         value = create_value(value)
@@ -551,7 +559,7 @@ class ArrayValue(_ArrayValueConst, _Container, collections.abc.MutableSequence,
             ptr = value._ptr
 
         status = native_bt.value_array_append_element(self._ptr, ptr)
-        utils._handle_func_status(status)
+        bt2_utils._handle_func_status(status)
 
     def __iadd__(self, iterable):
         # Python will raise a TypeError if there's anything wrong with
@@ -572,7 +580,7 @@ class _MapValueKeyIterator(collections.abc.Iterator):
         keys_ptr = native_bt.value_map_get_keys(map_obj._ptr)
 
         if keys_ptr is None:
-            raise RuntimeError('unexpected error: cannot get map value object keys')
+            raise RuntimeError("unexpected error: cannot get map value object keys")
 
         self._keys = _create_from_ptr(keys_ptr)
 
@@ -586,7 +594,7 @@ class _MapValueKeyIterator(collections.abc.Iterator):
 
 
 class _MapValueConst(_ContainerConst, collections.abc.Mapping, _ValueConst):
-    _NAME = 'Const map'
+    _NAME = "Const map"
     _borrow_entry_value_ptr = staticmethod(native_bt.value_map_borrow_entry_value_const)
 
     def __ne__(self, other):
@@ -619,7 +627,7 @@ class _MapValueConst(_ContainerConst, collections.abc.Mapping, _ValueConst):
         return native_bt.value_map_has_entry(self._ptr, key)
 
     def _check_key_type(self, key):
-        utils._check_str(key)
+        bt2_utils._check_str(key)
 
     def _check_key(self, key):
         if key not in self:
@@ -635,12 +643,12 @@ class _MapValueConst(_ContainerConst, collections.abc.Mapping, _ValueConst):
         return _MapValueKeyIterator(self)
 
     def __repr__(self):
-        items = ['{}: {}'.format(repr(k), repr(v)) for k, v in self.items()]
-        return '{{{}}}'.format(', '.join(items))
+        items = ["{}: {}".format(repr(k), repr(v)) for k, v in self.items()]
+        return "{{{}}}".format(", ".join(items))
 
 
 class MapValue(_MapValueConst, _Container, collections.abc.MutableMapping, _Value):
-    _NAME = 'Map'
+    _NAME = "Map"
     _borrow_entry_value_ptr = staticmethod(native_bt.value_map_borrow_entry_value)
 
     def __init__(self, value=None):
@@ -664,7 +672,7 @@ class MapValue(_MapValueConst, _Container, collections.abc.MutableMapping, _Valu
             ptr = value._ptr
 
         status = native_bt.value_map_insert_entry(self._ptr, key, ptr)
-        utils._handle_func_status(status)
+        bt2_utils._handle_func_status(status)
 
 
 _TYPE_TO_OBJ = {
This page took 0.02849 seconds and 4 git commands to generate.