U lª/eé ã@sÂddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZedƒZee eeed œd d „Ze eeeeed œd d„Zeee eeeeeefdœdd„ZdS)éN)Ú AsyncIterableÚListÚTupleÚTypeVar)Ú maybe_awaité)Úiter)Úislice)Ú AnyIterableÚ PredicateÚT)ÚnÚiterableÚreturncÃs*|dkrtdƒ‚dd„t||ƒ2ƒIdHS)a  Return the first n items of iterable as a list. If there are too few items in iterable, all of them are returned. n needs to be at least 0. If it is 0, an empty list is returned. Example:: first_two = await take(2, [1, 2, 3, 4, 5]) rz(take's first parameter can't be negativecÓsg|z3dHW}|‘q6S©N©)Ú.0Úitemrrú?/tmp/pip-unpacked-wheel-n_uj7zou/aioitertools/more_itertools.pyÚ sztake..N)Ú ValueErrorr )r rrrrÚtakes r)rr rcCs<t|ƒ}t||ƒIdH}|gkr8|Vt||ƒIdH}qdS)a# Break iterable into chunks of length n. The last chunk will be shorter if the total number of items is not divisible by n. Example:: async for chunk in chunked([1, 2, 3, 4, 5], n=2): ... # first iteration: chunk == [1, 2]; last one: chunk == [5] N)rr)rr ÚitÚchunkrrrÚchunked"s r)Ú predicaterrcƒs>t|ƒ‰t ¡ ¡‰‡‡‡fdd„}‡‡fdd„}|ƒ|ƒfS)aï A variant of :func:`aioitertools.takewhile` that allows complete access to the remainder of the iterator. >>> it = iter('ABCdEfGhI') >>> all_upper, remainder = await before_and_after(str.isupper, it) >>> ''.join([char async for char in all_upper]) 'ABC' >>> ''.join([char async for char in remainder]) 'dEfGhI' Note that the first iterator must be fully consumed before the second iterator can generate valid results. csLˆ2z63dHW}tˆ|ƒƒIdHr*|Vqˆ |¡dSq6ˆ t¡dSr)rZ set_resultZ set_exceptionÚStopAsyncIteration)Úelem©rrÚ transitionrrÚ true_iteratorKs   z'before_and_after..true_iteratorcsFzˆIdHVWntk r&YdSXˆ2z3dHW}|Vq,6dSr)r)Úelm)rrrrÚremainder_iteratorUs z,before_and_after..remainder_iterator)rÚasyncioZget_event_loopZ create_future)rrr r"rrrÚbefore_and_after5s    r$)r#ÚtypingrrrrZaioitertools.helpersrÚbuiltinsrÚ itertoolsr Útypesr r r Úintrrr$rrrrÚs   þ