~lttng/babeltrace/trunk

« back to all changes in this revision

Viewing changes to src/bindings/python/bt2/bt2/component.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 clock_class as bt2_clock_class
13
13
from bt2 import query_executor as bt2_query_executor
14
14
from bt2 import port as bt2_port
 
15
from bt2 import error as bt2_error
 
16
from bt2 import integer_range_set as bt2_integer_range_set
15
17
import sys
16
 
import bt2
17
18
 
18
19
 
19
20
class _IncompleteUserClass(Exception):
515
516
            )
516
517
        elif _UserSinkComponent in bases:
517
518
            if not hasattr(cls, "_user_consume"):
518
 
                raise bt2._IncompleteUserClass(
 
519
                raise _IncompleteUserClass(
519
520
                    "cannot create component class '{}': missing a _user_consume() method".format(
520
521
                        class_name
521
522
                    )
525
526
                cls, comp_cls_name, comp_cls_descr, comp_cls_help
526
527
            )
527
528
        else:
528
 
            raise bt2._IncompleteUserClass(
 
529
            raise _IncompleteUserClass(
529
530
                "cannot find a known component class base in the bases of '{}'".format(
530
531
                    class_name
531
532
                )
532
533
            )
533
534
 
534
535
        if cc_ptr is None:
535
 
            raise bt2._MemoryError(
 
536
            raise bt2_error._MemoryError(
536
537
                "cannot create component class '{}'".format(class_name)
537
538
            )
538
539
 
565
566
    @staticmethod
566
567
    def _bt_set_iterator_class(cls, iter_cls):
567
568
        if iter_cls is None:
568
 
            raise bt2._IncompleteUserClass(
 
569
            raise _IncompleteUserClass(
569
570
                "cannot create component class '{}': missing message iterator class".format(
570
571
                    cls.__name__
571
572
                )
572
573
            )
573
574
 
574
575
        if not issubclass(iter_cls, bt2_message_iterator._UserMessageIterator):
575
 
            raise bt2._IncompleteUserClass(
 
576
            raise _IncompleteUserClass(
576
577
                "cannot create component class '{}': message iterator class does not inherit bt2._UserMessageIterator".format(
577
578
                    cls.__name__
578
579
                )
579
580
            )
580
581
 
581
582
        if not hasattr(iter_cls, "__next__"):
582
 
            raise bt2._IncompleteUserClass(
 
583
            raise _IncompleteUserClass(
583
584
                "cannot create component class '{}': message iterator class is missing a __next__() method".format(
584
585
                    cls.__name__
585
586
                )
588
589
        if hasattr(iter_cls, "_user_can_seek_ns_from_origin") and not hasattr(
589
590
            iter_cls, "_user_seek_ns_from_origin"
590
591
        ):
591
 
            raise bt2._IncompleteUserClass(
 
592
            raise _IncompleteUserClass(
592
593
                "cannot create component class '{}': message iterator class implements _user_can_seek_ns_from_origin but not _user_seek_ns_from_origin".format(
593
594
                    cls.__name__
594
595
                )
597
598
        if hasattr(iter_cls, "_user_can_seek_beginning") and not hasattr(
598
599
            iter_cls, "_user_seek_beginning"
599
600
        ):
600
 
            raise bt2._IncompleteUserClass(
 
601
            raise _IncompleteUserClass(
601
602
                "cannot create component class '{}': message iterator class implements _user_can_seek_beginning but not _user_seek_beginning".format(
602
603
                    cls.__name__
603
604
                )
634
635
        # this can raise, but the native side checks the exception
635
636
        range_set = cls._user_get_supported_mip_versions(params, obj, log_level)
636
637
 
637
 
        if type(range_set) is not bt2.UnsignedIntegerRangeSet:
 
638
        if type(range_set) is not bt2_integer_range_set.UnsignedIntegerRangeSet:
638
639
            # this can raise, but the native side checks the exception
639
 
            range_set = bt2.UnsignedIntegerRangeSet(range_set)
 
640
            range_set = bt2_integer_range_set.UnsignedIntegerRangeSet(range_set)
640
641
 
641
642
        # return new reference
642
643
        range_set._get_ref(range_set._ptr)
669
670
            priv_query_exec._invalidate()
670
671
 
671
672
        # this can raise, but the native side checks the exception
672
 
        results = bt2.create_value(results)
 
673
        results = bt2_value.create_value(results)
673
674
 
674
675
        if results is None:
675
676
            results_ptr = native_bt.value_null
681
682
        return int(results_ptr)
682
683
 
683
684
    def _user_query(cls, priv_query_executor, object_name, params, method_obj):
684
 
        raise bt2.UnknownObject
 
685
        raise bt2_utils.UnknownObject
685
686
 
686
687
    def _bt_component_class_ptr(self):
687
688
        return self._bt_as_component_class_ptr(self._bt_cc_ptr)
791
792
        tc_ptr = native_bt.trace_class_create(ptr)
792
793
 
793
794
        if tc_ptr is None:
794
 
            raise bt2._MemoryError("could not create trace class")
 
795
            raise bt2_error._MemoryError("could not create trace class")
795
796
 
796
797
        tc = bt2_trace_class._TraceClass._create_from_ptr(tc_ptr)
797
798
        tc._assigns_automatic_stream_class_id = assigns_automatic_stream_class_id
816
817
        cc_ptr = native_bt.clock_class_create(ptr)
817
818
 
818
819
        if cc_ptr is None:
819
 
            raise bt2._MemoryError("could not create clock class")
 
820
            raise bt2_error._MemoryError("could not create clock class")
820
821
 
821
822
        cc = bt2_clock_class._ClockClass._create_from_ptr(cc_ptr)
822
823