range_with_timeout.py

class RangeWithTimeout(object):

    def __init__(self, n=None, timeout_seconds=None):
        self.n = n
        self.timeout_seconds = timeout_seconds

        self.i = 0

    def __iter__(self):
        self.start_datetime = datetime.datetime.now()
        return self

    def __next__(self):
        i = self.i
        self.i += 1

        if self.n is not None and i >= self.n:
            raise StopIteration

        if self.timeout_seconds is not None:
            elapsed_timedelta = datetime.datetime.now() - self.start_datetime
            elapsed_seconds = elapsed_timedelta.total_seconds()
            if elapsed_seconds >= self.timeout_seconds:
                raise StopIteration

        return i