3 [ð]°*ã @s–dZddlZddlmZy ddlZWnek r<dZYnXejdedddd„Zd d „Z eGd d „d e ƒƒZ Gd d„de ƒZ Gdd„de ƒZ 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: 2007 Pallets :license: BSD-3-Clause éNé)Úimplements_iteratorz^'werkzeug.contrib.iterio' is deprecated as of version 0.15 and will be removed in version 1.0.)Ú stacklevelcCs8t|ƒ}t||ƒ}t|tƒr*|dj|ƒS|dj|ƒS)z2concatenate any string type in an intelligent way.óÚ)ÚiterÚnextÚ isinstanceÚbytesÚjoin)ÚiterableÚsentinelÚiteratorZ first_item©rú;/tmp/pip-build-w1as9q70/Werkzeug/werkzeug/contrib/iterio.pyÚ _mixed_join;s   rcCst|tƒrdSdS)Nó Ú )r r )Zreference_stringrrrÚ_newlineDs rc@s„eZdZdZd dd„Zdd„Zdd„Zd d „Zd!d d „Zd"dd„Z dd„Z dd„Z d$dd„Z d%dd„Z d&dd„Zdd„Zdd„ZdS)'ÚIterIOaÁInstances 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 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. rc Cs2y t|ƒ}Wntk r&t||ƒSXt||ƒS)N)rÚ TypeErrorÚIterIÚIterO)ÚclsÚobjr rrrrÚ__new__fs   zIterIO.__new__cCs|S)Nr)ÚselfrrrÚ__iter__mszIterIO.__iter__cCs|jrtdƒ‚|jS)NzI/O operation on closed file)ÚclosedÚ ValueErrorÚpos)rrrrÚtellpsz IterIO.tellcCs|jrtdƒ‚dS)NzI/O operation on closed fileF)rr)rrrrÚisattyusz IterIO.isattyrcCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed fileé zBad file descriptor)rrÚIOError)rr ÚmoderrrÚseekzsz IterIO.seekNcCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rÚsizerrrÚtruncateszIterIO.truncatecCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rÚsrrrÚwrite„sz IterIO.writecCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rÚlistrrrÚ writelines‰szIterIO.writelinesécCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rÚnrrrÚreadŽsz IterIO.readcCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rÚsizehintrrrÚ readlines“szIterIO.readlinescCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rÚlengthrrrÚreadline˜szIterIO.readlinecCs|jrtdƒ‚tddƒ‚dS)NzI/O operation on closed filer#zBad file descriptor)rrr$)rrrrÚflushsz IterIO.flushcCs"|jr tƒ‚|jƒ}|stƒ‚|S)N)rÚ StopIterationr3)rÚlinerrrÚ__next__¢s zIterIO.__next__)r)r)Néÿÿÿÿ)r8)r)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr!r"r&r(r*r,r/r1r3r4r7rrrrrJs      rc@sBeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dS)rz#Convert an stream into an iterator.rc#s|tdkrtdƒ‚tj|ƒ‰tjƒˆ_gˆ_dˆ_|ˆ_dˆ_ ‡‡fdd„}tj|ˆjƒ}x|j ƒ}|sjdS|dVqZWdS)NzIterI requires greenlet supportFrcsˆˆƒˆjƒdS)N)Úcloser)ÚfuncÚstreamrrÚrun¸szIterI.__new__..run) ÚgreenletÚ RuntimeErrorÚobjectrZ getcurrentÚ_parentÚ_bufferrr r Úswitch)rr>r r@ÚgÚrvr)r>r?rr®s  z IterI.__new__cCs|jsd|_|jƒdS)NT)rÚ _flush_impl)rrrrr=Ãsz IterI.closecCs4|jrtdƒ‚|r0|jt|ƒ7_|jj|ƒdS)NzI/O operation on closed file)rrr ÚlenrEÚappend)rr)rrrr*Ès z IterI.writecCsx|D]}|j|ƒqWdS)N)r*)rr+Úitemrrrr,Ïs zIterI.writelinescCs|jrtdƒ‚|jƒdS)NzI/O operation on closed file)rrrI)rrrrr4Ósz IterI.flushcCs>t|j|jƒ}g|_| r,|jr,|jjƒn|jj|fƒdS)N)rrEr rrDrF)rÚdatarrrrIØs   zIterI._flush_implN)r) r9r:r;r<rr=r*r,r4rIrrrrr«s rc@sZeZdZdZddd„Zdd„Zdd„Zd d „Zdd d „Zddd„Z ddd„Z ddd„Z dS)rzCIter output. Wrap an iterator and give it a stream like interface.rcCs,tj|ƒ}||_d|_||_d|_d|_|S)NFr)rCrÚ_genÚ_bufr rr )rÚgenr rrrrräs z IterO.__new__cCs|S)Nr)rrrrríszIterO.__iter__cCs |js||_n|j|7_dS)z[Replace string directly without appending to an empty string, avoiding type issues.N)rO)rÚstringrrrÚ _buf_appendðszIterO._buf_appendcCs&|js"d|_t|jdƒr"|jjƒdS)NTr=)rÚhasattrrNr=)rrrrr=øs z IterO.closerc CsÚ|jrtdƒ‚|dkr"||j7}n8|dkrJ|jƒt|j|j|ƒ|_dS|dkrZtdƒ‚g}y@t|jpjdƒ}x,||kršt|j ƒ}|t|ƒ7}|j |ƒqpWWnt k r²YnX|rÊ|j t ||jƒƒtd|ƒ|_dS)NzI/O operation on closed filer-rrzInvalid argumentr)rrr r/Úminr$rJrOrrNrKr5rRrr Úmax)rr r%ÚbufÚ tmp_end_posrLrrrr&þs,    z IterO.seekr-cCs(|jrtdƒ‚|dkrP|jt|j|jƒƒ|j|jd…}|jt|ƒ7_|S|j|}g}yZ|jdkrndnt|jƒ}x<||ks’|jdkr´| r´t |jƒ}|t|ƒ7}|j |ƒqzWWnt k rÌYnX|rä|jt||jƒƒ|jdkrô|jSt d|ƒ}z|j|j|…St |t|jƒƒ|_XdS)NzI/O operation on closed filer)rrrRrrNr rOr rJrrKr5rUrT)rr.ÚresultÚnew_posrVrWrLrrrr/s2     z IterO.readNcCsF|jrtdƒ‚d}|jr.|jjt|jƒ|jƒ}g}yn|jdkrF|j}n t|jƒ}xL|dkrœt|jƒ}|jt|ƒƒ}|j |ƒ|dkrŽ||}P|t|ƒ7}qRWWnt k r´YnX|rÌ|j t ||j ƒƒ|jdkrÜ|j S|dkrðt|jƒ}n|d}|dk r|j||kr|j|}z|j|j|…St|t|jƒƒ|_XdS)NzI/O operation on closed filer-rr8)rrrOÚfindrr rJrrNrKr5rRrr rT)rr2Únl_posrVr rLZ local_posrYrrrr34s@        zIterO.readlinecCsVd}g}|jƒ}x@|rP|j|ƒ|t|ƒ7}d|ko>|knrFP|jƒ}qW|S)Nr)r3rKrJ)rr0ÚtotalÚlinesr6rrrr1\s   zIterO.readlines)r)rr8)r8)N)r) r9r:r;r<rrrRr=r&r/r3r1rrrrrás    (r)r<ÚwarningsÚ_compatrrAÚ ImportErrorÚwarnÚDeprecationWarningrrrCrrrrrrrÚ)s      `6