bt2: Add bindings for trace classes

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

diff --git a/bindings/python/bt2/bt2/stream_class.py b/bindings/python/bt2/bt2/stream_class.py
index 97cd593fe2d90b325399e647fcb03ef61a76142d..e94c3b2297d3d23fc74a05fc4fa72ff417b7242c 100644 (file)
--- a/bindings/python/bt2/bt2/stream_class.py
+++ b/bindings/python/bt2/bt2/stream_class.py
@@ -49,6 +49,9 @@ class _EventClassIterator(collections.abc.Iterator):
 
 
 class StreamClass(object._SharedObject, collections.abc.Mapping):
+    _get_ref = staticmethod(native_bt.stream_class_get_ref)
+    _put_ref = staticmethod(native_bt.stream_class_put_ref)
+
     def __init__(self, name=None, id=None, packet_context_field_class=None,
                  event_header_field_class=None, event_context_field_class=None,
                  event_classes=None):