bt2: rename CreationError to MemoryError, handle it in and out of Python bindings

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

diff --git a/src/bindings/python/bt2/bt2/port.py b/src/bindings/python/bt2/bt2/port.py
index f7c236692e23d35951c00b85375ef58a887d9831..ecee18ce47b7eaa65dca8a218136572f9cd26c2a 100644 (file)
--- a/src/bindings/python/bt2/bt2/port.py
+++ b/src/bindings/python/bt2/bt2/port.py
@@ -32,7 +32,7 @@ def _create_from_ptr_and_get_ref(ptr, port_type):
     cls = _PORT_TYPE_TO_PYCLS.get(port_type, None)
 
     if cls is None:
-        raise bt2.Error('unknown port type: {}'.format(port_type))
+        raise TypeError('unknown port type: {}'.format(port_type))
 
     return cls._create_from_ptr_and_get_ref(ptr)
 
@@ -41,7 +41,7 @@ def _create_self_from_ptr_and_get_ref(ptr, port_type):
     cls = _PORT_TYPE_TO_USER_PYCLS.get(port_type, None)
 
     if cls is None:
-        raise bt2.Error('unknown port type: {}'.format(port_type))
+        raise TypeError('unknown port type: {}'.format(port_type))
 
     return cls._create_from_ptr_and_get_ref(ptr)
 
@@ -110,18 +110,24 @@ class _UserComponentPort(_Port):
 
 
 class _UserComponentInputPort(_UserComponentPort, _InputPort):
-    _as_self_port_ptr = staticmethod(native_bt.self_component_port_input_as_self_component_port)
+    _as_self_port_ptr = staticmethod(
+        native_bt.self_component_port_input_as_self_component_port
+    )
 
     def create_message_iterator(self):
-        msg_iter_ptr = native_bt.self_component_port_input_message_iterator_create(self._ptr)
+        msg_iter_ptr = native_bt.self_component_port_input_message_iterator_create(
+            self._ptr
+        )
         if msg_iter_ptr is None:
-            raise bt2.CreationError('cannot create message iterator object')
+            raise bt2.MemoryError('cannot create message iterator object')
 
         return bt2.message_iterator._UserComponentInputPortMessageIterator(msg_iter_ptr)
 
 
 class _UserComponentOutputPort(_UserComponentPort, _OutputPort):
-    _as_self_port_ptr = staticmethod(native_bt.self_component_port_output_as_self_component_port)
+    _as_self_port_ptr = staticmethod(
+        native_bt.self_component_port_output_as_self_component_port
+    )
 
 
 _PORT_TYPE_TO_PYCLS = {