~lttng/babeltrace/trunk

« back to all changes in this revision

Viewing changes to src/bindings/python/bt2/bt2/message_iterator.py

  • Committer: Philippe Proulx
  • Author(s): Simon Marchi
  • Date: 2023-09-21 17:29:44 UTC
  • Revision ID: git-v1:c345b07873b0cb0ed344bde32a322a1b1edf60ae
python: remove internal `import bt2` imports

Standardize the internal imports to import specific modules like this:

    from bt2 import foo as bt2_foo

... instead of importing the complete bt2 package.

Change-Id: I32eb223a8bcfe841020a2aafb976f121dadd8680
Signed-off-by: Simon Marchi <simon.marchi@efficios.com>
Reviewed-on: https://review.lttng.org/c/babeltrace/+/10388
Tested-by: jenkins <jenkins@lttng.org>
Reviewed-by: Philippe Proulx <eeppeliteloop@gmail.com>

Show diffs side-by-side

added added

removed removed

Lines of Context:
12
12
from bt2 import packet as bt2_packet
13
13
from bt2 import port as bt2_port
14
14
from bt2 import clock_class as bt2_clock_class
15
 
import bt2
 
15
from bt2 import error as bt2_error
16
16
 
17
17
 
18
18
class _MessageIterator(collections.abc.Iterator):
167
167
        pass
168
168
 
169
169
    def __next__(self):
170
 
        raise bt2.Stop
 
170
        raise bt2_utils.Stop
171
171
 
172
172
    def _bt_next_from_native(self):
173
173
        # this can raise anything: it's catched by the native part
174
174
        try:
175
175
            msg = next(self)
176
176
        except StopIteration:
177
 
            raise bt2.Stop
 
177
            raise bt2_utils.Stop
178
178
        except Exception:
179
179
            raise
180
180
 
280
280
                )
281
281
 
282
282
        if ptr is None:
283
 
            raise bt2._MemoryError("cannot create event message object")
 
283
            raise bt2_error._MemoryError("cannot create event message object")
284
284
 
285
285
        return bt2_message._EventMessage(ptr)
286
286
 
291
291
        )
292
292
 
293
293
        if ptr is None:
294
 
            raise bt2._MemoryError("cannot create inactivity message object")
 
294
            raise bt2_error._MemoryError("cannot create inactivity message object")
295
295
 
296
296
        return bt2_message._MessageIteratorInactivityMessage(ptr)
297
297
 
300
300
 
301
301
        ptr = native_bt.message_stream_beginning_create(self._bt_ptr, stream._ptr)
302
302
        if ptr is None:
303
 
            raise bt2._MemoryError("cannot create stream beginning message object")
 
303
            raise bt2_error._MemoryError(
 
304
                "cannot create stream beginning message object"
 
305
            )
304
306
 
305
307
        msg = bt2_message._StreamBeginningMessage(ptr)
306
308
 
314
316
 
315
317
        ptr = native_bt.message_stream_end_create(self._bt_ptr, stream._ptr)
316
318
        if ptr is None:
317
 
            raise bt2._MemoryError("cannot create stream end message object")
 
319
            raise bt2_error._MemoryError("cannot create stream end message object")
318
320
 
319
321
        msg = bt2_message._StreamEndMessage(ptr)
320
322
 
345
347
            ptr = native_bt.message_packet_beginning_create(self._bt_ptr, packet._ptr)
346
348
 
347
349
        if ptr is None:
348
 
            raise bt2._MemoryError("cannot create packet beginning message object")
 
350
            raise bt2_error._MemoryError(
 
351
                "cannot create packet beginning message object"
 
352
            )
349
353
 
350
354
        return bt2_message._PacketBeginningMessage(ptr)
351
355
 
371
375
            ptr = native_bt.message_packet_end_create(self._bt_ptr, packet._ptr)
372
376
 
373
377
        if ptr is None:
374
 
            raise bt2._MemoryError("cannot create packet end message object")
 
378
            raise bt2_error._MemoryError("cannot create packet end message object")
375
379
 
376
380
        return bt2_message._PacketEndMessage(ptr)
377
381
 
413
417
            ptr = native_bt.message_discarded_events_create(self._bt_ptr, stream._ptr)
414
418
 
415
419
        if ptr is None:
416
 
            raise bt2._MemoryError("cannot discarded events message object")
 
420
            raise bt2_error._MemoryError("cannot discarded events message object")
417
421
 
418
422
        msg = bt2_message._DiscardedEventsMessage(ptr)
419
423
 
460
464
            ptr = native_bt.message_discarded_packets_create(self._bt_ptr, stream._ptr)
461
465
 
462
466
        if ptr is None:
463
 
            raise bt2._MemoryError("cannot discarded packets message object")
 
467
            raise bt2_error._MemoryError("cannot discarded packets message object")
464
468
 
465
469
        msg = bt2_message._DiscardedPacketsMessage(ptr)
466
470