? f (ddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZedZd ed e ed eefd Zd e ed ed eeefd Zdeed e ed eeeeeffdZdS)N) AsyncIterableListTupleTypeVar) maybe_await)iter)islice) AnyIterable PredicateTniterablereturncpK|dkrtddt||2d{VS)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"Kg|3d{V}| 6SN).0items j/home/panchajanya/Development/work/volatility3/lib/python3.11/site-packages/aioitertools/more_itertools.py ztake..s. 7 7 7 7 7 7 7 74D 7 7 7 7sN) ValueErrorr )rrs rtakersP 1uuCDDD 7 76(A#6#6 7 7 7 7 7 7 7 7 77cKt|}t||d{V}|gkr#|WVt||d{V}|gk!dSdS)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)r r)rritchunks rchunkedr"sx hBq"++      E 2++ 1bkk!!!!!! 2++++++r predicatecKt|tjfd}fd}||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. cK23d{V}t|d{Vr|WV+|dS6tdSr)r set_result set_exceptionStopAsyncIteration)elemrr transitions r true_iteratorz'before_and_after..true_iteratorKs       $ 411111111  %%d+++    !344444sA chK d{VWVn#t$rYdSwxYw23d{V}|WV 6dSr)r%)elmrr's rremainder_iteratorz,before_and_after..remainder_iteratorUs ####### $ $ $ $ $!    FF        #IIIIIs  1)r asyncioget_event_loop create_future)r rr(r+rr's` @@rbefore_and_afterr/5s$ hB'))7799J5555555 =??..00 00r)r,typingrrrraioitertools.helpersrbuiltinsr itertoolsr typesr r r intrrr/rrrr6sJ666666666666,,,,,,)))))))) GCLL8#8Q8DG8888""KN"s"}T!W7M""""&)1|)1'21~)1 = ]1- -.)1)1)1)1)1)1r