B bAZ>*@s~dZy ddlZWnek r(dZYnXddlmZddZddZeGdd d eZGd d d eZ Gd d d eZ dS)a werkzeug.contrib.iterio ~~~~~~~~~~~~~~~~~~~~~~~ This module implements a :class:`IterIO` that converts an iterator into a stream object and the other way round. Converting streams into iterators requires the `greenlet`_ module. To convert an iterator into a stream all you have to do is to pass it directly to the :class:`IterIO` constructor. In this example we pass it a newly created generator:: def foo(): yield "something\n" yield "otherthings" stream = IterIO(foo()) print stream.read() # read the whole iterator The other way round works a bit different because we have to ensure that the code execution doesn't take place yet. An :class:`IterIO` call with a callable as first argument does two things. The function itself is passed an :class:`IterIO` stream it can feed. The object returned by the :class:`IterIO` constructor on the other hand is not an stream object but an iterator:: def foo(stream): stream.write("some") stream.write("thing") stream.flush() stream.write("otherthing") iterator = IterIO(foo) print iterator.next() # prints something print iterator.next() # prints otherthing iterator.next() # raises StopIteration .. _greenlet: https://github.com/python-greenlet/greenlet :copyright: (c) 2014 by the Werkzeug Team, see AUTHORS for more details. :license: BSD, see LICENSE for more details. N)implements_iteratorcCs8t|}t||}t|tr*|d|S|d|S)z2concatenate any string type in an intelligent way.)iternext isinstancebytesjoin)iterablesentineliteratorZ first_itemr 6lib/python3.7/site-packages/werkzeug/contrib/iterio.py _mixed_join2s   rcCst|trdSdS)N  )rr)Zreference_stringr r r_newline;s rc@seZdZdZd ddZddZddZd d Zd!d d Zd"ddZ ddZ ddZ d#ddZ d$ddZ d%ddZddZddZdS)&IterIOaInstances of this object implement an interface compatible with the standard Python :class:`file` object. Streams are either read-only or write-only depending on how the object is created. If the first argument is an iterable a file like object is returned that returns the contents of the iterable. In case the iterable is empty read operations will return the sentinel value. If the first argument is a callable then the stream object will be created and passed to that function. The caller itself however will not receive a stream but an iterable. The function will be be executed step by step as something iterates over the returned iterable. Each call to :meth:`flush` will create an item for the iterable. If :meth:`flush` is called without any writes in-between the sentinel value will be yielded. Note for Python 3: due to the incompatible interface of bytes and streams you should set the sentinel value explicitly to an empty bytestring (``b''``) if you are expecting to deal with bytes as otherwise the end of the stream is marked with the wrong sentinel value. .. versionadded:: 0.9 `sentinel` parameter was added. rcCs2y t|}Wntk r&t||SXt||S)N)r TypeErrorIterIIterO)clsobjr r r r r__new__^s   zIterIO.__new__cCs|S)Nr )selfr r r__iter__eszIterIO.__iter__cCs|jrtd|jS)NzI/O operation on closed file)closed ValueErrorpos)rr r rtellhsz IterIO.tellcCs|jrtddS)NzI/O operation on closed fileF)rr)rr r risattymsz IterIO.isattyrcCs|jrtdtdddS)NzI/O operation on closed file zBad file descriptor)rrIOError)rrmoder r rseekrsz IterIO.seekNcCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rsizer r rtruncatewszIterIO.truncatecCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rsr r rwrite|sz IterIO.writecCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rlistr r r writelinesszIterIO.writelinescCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rnr r rreadsz IterIO.readcCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rsizehintr r r readlinesszIterIO.readlinescCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rlengthr r rreadlineszIterIO.readlinecCs|jrtdtdddS)NzI/O operation on closed filer!zBad file descriptor)rrr")rr r rflushsz IterIO.flushcCs"|jr t|}|st|S)N)r StopIterationr1)rliner r r__next__s zIterIO.__next__)r)r)N)r+)r)N)__name__ __module__ __qualname____doc__rrrr r$r&r(r*r-r/r1r2r5r r r rrAs      rc@sBeZdZdZdddZddZddZd d Zd d Zd dZ dS)rz#Convert an stream into an iterator.rc#s|tdkrtdt|t_g_d_|_d_ fdd}t|j}x| }|sjdS|dVqZWdS)NzIterI requires greenlet supportFrcsdS)N)closer )funcstreamr rrunszIterI.__new__..run) greenlet RuntimeErrorobjectrZ getcurrent_parent_bufferrr rswitch)rr;r r=grvr )r;r<rrs  z IterI.__new__cCs|jsd|_|dS)NT)r _flush_impl)rr r rr:sz IterI.closecCs4|jrtd|r0|jt|7_|j|dS)NzI/O operation on closed file)rrrlenrBappend)rr'r r rr(s z IterI.writecCsx|D]}||qWdS)N)r()rr)itemr r rr*s zIterI.writelinescCs|jrtd|dS)NzI/O operation on closed file)rrrF)rr r rr2sz IterI.flushcCs<t|j|j}g|_|s*|jr*|jn|j|fdS)N)rrBr rrArC)rdatar r rrFs   zIterI._flush_implN)r) r6r7r8r9rr:r(r*r2rFr r r rrs rc@sZeZdZdZdddZddZddZd d Zdd d ZdddZ dddZ dddZ dS)rzCIter output. Wrap an iterator and give it a stream like interface.rcCs,t|}||_d|_||_d|_d|_|S)NFr)r@r_gen_bufr rr)rgenr rr r rrs z IterO.__new__cCs|S)Nr )rr r rrszIterO.__iter__cCs |js||_n|j|7_dS)z[Replace string directly without appending to an empty string, avoiding type issues.N)rL)rstringr r r _buf_appendszIterO._buf_appendcCs&|js"d|_t|jdr"|jdS)NTr:)rhasattrrKr:)rr r rr:s z IterO.closercCs|jrtd|dkr"||j7}n8|dkrJ|t|j|j||_dS|dkrZtdg}y ImportErrorZwerkzeug._compatrrrr@rrrr r r r)s     a7