bt2: stream.py: add `trace` property to _Stream

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

diff --git a/src/bindings/python/bt2/bt2/stream.py b/src/bindings/python/bt2/bt2/stream.py
index 3a8821795caeb9c0aa5eeaf3cf75e89e9bc63044..308924e937384e003a074869c22a4abece677f52 100644 (file)
--- a/src/bindings/python/bt2/bt2/stream.py
+++ b/src/bindings/python/bt2/bt2/stream.py
@@ -24,6 +24,7 @@ from bt2 import native_bt, utils
 from bt2 import object as bt2_object
 from bt2 import packet as bt2_packet
 from bt2 import event as bt2_event
+from bt2 import trace as bt2_trace
 from bt2 import stream_class as bt2_stream_class
 import bt2
 
@@ -67,3 +68,9 @@ class _Stream(bt2_object._SharedObject):
             raise bt2._MemoryError('cannot create packet object')
 
         return bt2_packet._Packet._create_from_ptr(packet_ptr)
+
+    @property
+    def trace(self):
+        trace_ptr = native_bt.stream_borrow_trace(self._ptr)
+        assert trace_ptr is not None
+        return bt2_trace._Trace._create_from_ptr_and_get_ref(trace_ptr)