bt2: rename object's own BT class property to `cls`

[babeltrace.git] / bindings / python / bt2 / bt2 / object.py

diff --git a/bindings/python/bt2/bt2/object.py b/bindings/python/bt2/bt2/object.py
index 05c6deb75b63ac345aeabaa072ae63f836d83de5..218965de7799762839cae6120696a55ff21023b7 100644 (file)
--- a/bindings/python/bt2/bt2/object.py
+++ b/bindings/python/bt2/bt2/object.py
@@ -72,10 +72,14 @@ class _UniqueObject(_BaseObject):
     def _create_from_ptr_and_get_ref(cls, ptr, owner_ptr,
                                      owner_get_ref, owner_put_ref):
         obj = cls.__new__(cls)
+
         obj._ptr = ptr
         obj._owner_ptr = owner_ptr
+        obj._owner_get_ref = owner_get_ref
         obj._owner_put_ref = owner_put_ref
-        owner_get_ref(obj._owner_ptr)
+
+        obj._owner_get_ref(obj._owner_ptr)
+
         return obj
 
     def __del__(self):
@@ -123,5 +127,12 @@ class _SharedObject(_BaseObject):
         cls._get_ref(obj._ptr)
         return obj
 
+    def _release(self):
+        """Return the wrapped pointer, transfer its ownership to the
+        caller."""
+        ptr = self._ptr
+        self._ptr = None
+        return ptr
+
     def __del__(self):
         self._put_ref(self._ptr)