f]UdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZddlmZd d gZed e e d f Ze ddZdZded<edZGddZef ddZy )z& Implementation for async generators. ) annotations)get_running_loop)asynccontextmanager)EmptyFullQueue)AnyAsyncGeneratorCallableIterableTypeVar)run_in_executor_with_contextaclosinggenerator_to_async_generator _T_GeneratorN)boundcK ||jd{y7#|jd{7wxYww)z1Similar to `contextlib.aclosing`, in Python 3.10.N)aclose)things c/var/www/cvtools/html/venv/lib/python3.12/site-packages/prompt_toolkit/eventloop/async_generator.pyrrs.  llnellns(A$A"A?;?AiintDEFAULT_BUFFER_SIZE_Tc eZdZy)_DoneN)__name__ __module__ __qualname__rrr1sr!rcfKdt|t}dfd }t|} j}t|trn|( d|d{y#t$r(|j dj d{7}YUwxYw79#d|d{7wxYww)a] Turn a generator or iterable into an async generator. This works by running the generator in a background thread. :param get_iterable: Function that returns a generator or iterable when called. :param buffer_size: Size of the queue between the async consumer and the synchronous generator that produces items. F)maxsizec  D]8}r jtdy j|d: jtdy#t$rrYyYcwxYw#t$r8r3Y jtdy#t$rrYyY-wxYwYwxYw#t$rrYyYwxYw# jtdw#t$r rYYyY/wxYwxYw)z~ Consume the generator in background thread. When items are received, they'll be pushed to the queue. r)timeoutN)putrr)item get_iterableqquittings rrunnerz,generator_to_async_generator..runnerHsF  $ EE%'1E- 'dA.  "EE%'1E-   !#" EE%'1E-  !!EE%'1E-  s CA$CA7CB;$ A43A47 B8CB"" B21B25C7B88C; C  C DC-,D- C?8D<D>C??DTN)returnNone) rrr get_nowaitrrun_in_executorget isinstancer)r( buffer_sizeloopr+runner_fr'r)r*s` @@rrr5sH 5A  DD,F3H ?||~$&    ?!11$>>> ?  si,B1BA)BB1#B$B1)(BBBBBBB1B.'B*(B..B1)rrr,z"AsyncGenerator[_T_Generator, None])r(zCallable[[], Iterable[_T]]r2rr,zAsyncGenerator[_T, None])__doc__ __future__rasyncior contextlibrqueuerrrtypingr r r r r utilsr__all__rrr__annotations__rrrr r!rr>s#$*$$CC/"  ~^CI-FG  '* S T]  +H,HHHr!