bt2: add user attributes property support

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

diff --git a/src/bindings/python/bt2/bt2/clock_class.py b/src/bindings/python/bt2/bt2/clock_class.py
index a8247dfbf4244ea72b53fbed26d480b4549b67d3..953a2185b2d8c3926d1b57873c959237379d4b2c 100644 (file)
--- a/src/bindings/python/bt2/bt2/clock_class.py
+++ b/src/bindings/python/bt2/bt2/clock_class.py
@@ -21,6 +21,7 @@
 # THE SOFTWARE.
 
 from bt2 import native_bt, object, utils
+from bt2 import value as bt2_value
 import uuid as uuidp
 
 
@@ -51,6 +52,19 @@ class _ClockClass(object._SharedObject):
     _get_ref = staticmethod(native_bt.clock_class_get_ref)
     _put_ref = staticmethod(native_bt.clock_class_put_ref)
 
+    @property
+    def user_attributes(self):
+        ptr = native_bt.clock_class_borrow_user_attributes(self._ptr)
+        assert ptr is not None
+        return bt2_value._create_from_ptr_and_get_ref(ptr)
+
+    def _user_attributes(self, user_attributes):
+        value = bt2_value.create_value(user_attributes)
+        utils._check_type(value, bt2_value.MapValue)
+        native_bt.clock_class_set_user_attributes(self._ptr, value._ptr)
+
+    _user_attributes = property(fset=_user_attributes)
+
     @property
     def name(self):
         return native_bt.clock_class_get_name(self._ptr)