def __getitem__()

in distill/segmentation/segments.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def __getitem__(self, item):
        """
        Allows Segments to be subscriptable by segment name or numeric index.
        """
        if isinstance(item, str):
            segment_names = [segment.get_segment_name() for segment in self.segments]
            if item in segment_names:
                index = segment_names.index(item)
                return self.segments[index]
        else:
            return self.segments[item]