Read input word by word in Python

word_reader.py

class WordReader(Iterator):
    def __init__(self, fp=sys.stdin):
        self.fp = fp

    def __next__(self):
        word = ''
        while True:
            c = self.fp.read(1)
            if c is None:
                if word:
                    return word
                else:
                    raise StopIteration
            elif c == ' ' or c == '\n':
                if word:
                    return word
            else:
                word += c

    def get_int(self):
        return int(next(self))