/ - Diff - BIEN 3 - NCEAS Projects

« Previous | Next »

Revision 1635

streams.py: Added TimeoutInputStream and WrapStream. Changed StreamIter to use new WrapStream.

     # I/O
     class StreamIter:
     import timeout
     class WrapStream:
         '''Forwards close() to the underlying stream'''
         def __init__(self, stream): self.stream = stream
         def close(self): self.stream.close()
     class StreamIter(WrapStream):
         '''Iterates over the lines in a stream.
         Unlike stream.__iter__(), doesn't wait for EOF to start returning lines.
         Offers curr() method.
         Forwards close() to the underlying stream.
         '''
         def __init__(self, stream):
             self.stream = stream
             WrapStream.__init__(self, stream)
             self.line = None
         def __iter__(self): return self
-...
             line = self.curr()
             self.line = None
             return line
     class TimeoutInputStream(WrapStream):
         '''@param timeout_ sec'''
         def __init__(self, stream, timeout_):
             WrapStream.__init__(self, stream)
             self.timeout = timeout_
         def close(self): self.stream.close()
         def readline(self):
             return timeout.run(lambda: self.stream.readline(), self.timeout)
     class TracedOutputStream:
         '''Wraps an output stream, running a trace function on each string written

Also available in: Unified diff