Breaking News: Grepper is joining You.com. Read the official announcement!

python peek next item from iterator

Amused Albatross answered on March 7, 2020 Popularity 4/10 Helpfulness 1/10

answer python peek next item from iterator

python peek next item from iterator

Comment

class Peekorator(object):

    def __init__(self, generator):
        self.empty = False
        self.peek = None
        self.generator = generator
        try:
            self.peek = self.generator.next()
        except StopIteration:
            self.empty = True

    def __iter__(self):
        return self

    def next(self):
        """
        Return the self.peek element, or raise StopIteration
        if empty
        """
        if self.empty:
            raise StopIteration()
        to_return = self.peek
        try:
            self.peek = self.generator.next()
        except StopIteration:
            self.peek = None
            self.empty = True
        return to_return

def simple_iterator():
    for x in range(10):
        yield x*3

pkr = Peekorator(simple_iterator())
for i in pkr:
    print i, pkr.peek, pkr.empty

Popularity 4/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: iterator next peek python

Link to this answer
Share Copy Link