j]UdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZddlmZd d gZed e e d f ZeddZdZded<edZGddZefddZd S)z& Implementation for async generators. ) annotations)get_running_loop)asynccontextmanager)EmptyFullQueue)AnyAsyncGeneratorCallableIterableTypeVar)run_in_executor_with_contextaclosinggenerator_to_async_generator _T_GeneratorN)boundthingreturn"AsyncGenerator[_T_Generator, None]cK |WV|d{VdS#|d{VwxYw)z1Similar to `contextlib.aclosing`, in Python 3.10.N)aclose)rs u/var/www/finuniver-perm.ru/html/student/venv/lib/python3.11/site-packages/prompt_toolkit/eventloop/async_generator.pyrrsb  llnnellnns %AiintDEFAULT_BUFFER_SIZE_TceZdZdS)_DoneN)__name__ __module__ __qualname__rrr1sDr#r get_iterableCallable[[], Iterable[_T]] buffer_sizeAsyncGenerator[_T, None]c|Kdt|t}dfd }t|} }n1#t$r$|djd{V}YnwxYwt|trn|WVc d|d{VdS#d|d{VwxYw) 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)maxsizerNonecp D]}r= tddS#t$rrYdSY8wxYw |dnP#t$rCr>Y tddS#t$rrYdSY8wxYwYfwxYw tddS#t$rrYdSY8wxYw# tdn#t$r rYYdSY9wxYwwxYw)z~ Consume the generator in background thread. When items are received, they'll be pushed to the queue. Tr)timeoutN)putrr)itemr$qquittings rrunnerz,generator_to_async_generator..runnerHs  $    EE%''1E--- E HdA...   !!!##"F  EE%''1E--- E H!!  " EE%''1E--- E H EE%''1E---  H    sC6#9 A  A C6A('C6( B54C68#B B/.B/2C64B55C6<#C!! C32C36D59#DD5 D1)D5.D50D11D5TN)rr*) rrr get_nowaitrrun_in_executorget isinstancer)r$r&loopr1runner_fr.r/r0s` @@rrr5s1H 555A   DD,F33H  ?||~~ ? ? ?!11$>>>>>>>> ?$&&      s/B/AB/+BB/BB// B;)rrrr)r$r%r&rrr')__doc__ __future__rasyncior contextlibrqueuerrrtypingr r r r r utilsr__all__rrr__annotations__rrrr"r#rrAsm#"""""$$$$$$******$$$$$$$$$$CCCCCCCCCCCCCC//////"  w~^CI-FGGG *  WT]]        +HHHHHHHr#