def get_segments_after()

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

• 5 lines of code
• 5 McCabe index (conditional complexity)

    def get_segments_after(self, time):
        """
        Returns a new Segments object only including segments that have start times after the indicated time.

        :param time: An integer or datetime object that represents the time for which Segment start times should be after.
        :return: A new Segments object that contains Segment objects that have start times after the time indicated.
        """
        if not isinstance(time, int) and not isinstance(time, datetime.datetime):
            raise TypeError('Time must be an integer or datetime object.')

        segments = [segment for segment in self.segments if segment.start_end_val[0] > time]

        return Segments(segments)