lib: graph: disallow recursive consuming

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

diff --git a/bindings/python/bt2/bt2/graph.py b/bindings/python/bt2/bt2/graph.py
index 56148395deafdea4d657896a7605f586febdf83a..ce861028f9388988457e9d2be96f8df95ba58144 100644 (file)
--- a/bindings/python/bt2/bt2/graph.py
+++ b/bindings/python/bt2/bt2/graph.py
@@ -106,6 +106,8 @@ class Graph(object._Object):
             raise bt2.TryAgain
         elif status == native_bt.GRAPH_STATUS_NO_SINK:
             raise bt2.NoSinkComponent
+        elif status == native_bt.GRAPH_STATUS_CANNOT_CONSUME:
+            raise bt2.CannotConsumeGraph
         elif status < 0:
             raise bt2.Error(gen_error_msg)