U ¯ªh"ã@s:ddlZddlZddlZddlmZGdd„dejƒZdS)éNé)Úhelpersc@sfeZdZdZdddœdd„Zdd„Zd d „Zd d „Zd d„Zdd„Z dd„Z e j dd„ƒZ dd„ZdS)Ú RequestIterah Helper class to deal with requests that need offsets to iterate. It has some facilities, such as automatically sleeping a desired amount of time between requests if needed (but not more). Can be used synchronously if the event loop is not running and as an asynchronous iterator otherwise. `limit` is the total amount of items that the iterator should return. This is handled on this base class, and will be always ``>= 0``. `left` will be reset every time the iterator is used and will indicate the amount of items that should be emitted left, so that subclasses can be more efficient and fetch only as many items as they need. Iterators may be used with ``reversed``, and their `reverse` flag will be set to `True` if that's the case. Note that if this flag is set, `buffer` should be filled in reverse too. FN)ÚreverseÚ wait_timecKsX||_||_||_||_t|dkr*tdƒn|dƒ|_|j|_d|_d|_ d|_ d|_ dS)NÚinfr) ÚclientrrÚkwargsÚmaxÚfloatÚlimitÚleftÚbufferÚindexÚtotalÚ last_load)Úselfrr rrr ©rú8/tmp/pip-unpacked-wheel-c81u5j2r/telethon/requestiter.pyÚ__init__szRequestIter.__init__cËsdS)aA Called when asynchronous initialization is necessary. All keyword arguments passed to `__init__` will be forwarded here, and it's preferable to use named arguments in the subclasses without defaults to avoid forgetting or misspelling any of them. This method may ``raise StopAsyncIteration`` if it cannot continue. This method may actually fill the initial buffer if it needs to, and similarly to `_load_next_chunk`, ``return True`` to indicate that this is the last iteration (just the initial load). Nr)rr rrrÚ_init)szRequestIter._initcÃsÚ|jdkr0g|_|jf|jŽIdHr0t|jƒ|_|jdkr>t‚|jt|jƒkr¤|jr~t  |jt   ¡|j ¡IdHt   ¡|_ d|_g|_|  ¡IdHr¤t|jƒ|_|js®t‚|j|j}|jd8_|jd7_|S)Nrr) rrr Úlenr ÚStopAsyncIterationrrÚasyncioÚsleepÚtimerÚ_load_next_chunk)rÚresultrrrÚ __anext__7s,   ÿ    zRequestIter.__anext__cCs2z|jj | ¡¡WStk r,t‚YnXdS©N)rÚloopZrun_until_completerrÚ StopIteration©rrrrÚ__next__UszRequestIter.__next__cCsd|_d|_d|_|j|_|S)Nr)rrrr r r"rrrÚ __aiter__[s zRequestIter.__aiter__cCs|jj ¡rtdƒ‚| ¡S)NzZYou must use "async for" if the event loop is running (i.e. you are inside an "async def"))rr Z is_runningÚ RuntimeErrorr$r"rrrÚ__iter__bs  ÿzRequestIter.__iter__cÃs2t ¡}|2z3dHW}| |¡q 6|j|_|S)z Create a `self` iterator and collect it into a `TotalList` (a normal list with a `.total` attribute). N)rZ TotalListÚappendr)rrÚmessagerrrÚcollectks zRequestIter.collectcÃst‚dS)a  Called when the next chunk is necessary. It should extend the `buffer` with new items. It should return `True` if it's the last chunk, after which moment the method won't be called again during the same iteration. N)ÚNotImplementedErrorr"rrrrws zRequestIter._load_next_chunkcCs|j |_|Sr)rr"rrrÚ __reversed__„s zRequestIter.__reversed__)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr#r$r&r)ÚabcÚabstractmethodrr+rrrrrs    r)r0rrÚrÚABCrrrrrÚs