a xdc@sUdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZmZddlmZd d gZed e e d fd Zed dddd ZdZded<edZGdddZefdddddd Zd S)z& Implementation for async generators. ) annotations)get_running_loop)asynccontextmanager)EmptyFullQueue)AnyAsyncGeneratorCallableIterableTypeVarUnion)run_in_executor_with_contextaclosinggenerator_to_async_generator _T_GeneratorN)boundz"AsyncGenerator[_T_Generator, None])thingreturncCs.z|VW|IdHn|IdH0dS)z1Similar to `contextlib.aclosing`, in Python 3.10.N)aclose)rr`/var/www/html/Ranjet/env/lib/python3.9/site-packages/prompt_toolkit/eventloop/async_generator.pyrsiintDEFAULT_BUFFER_SIZE_Tc@s eZdZdS)_DoneN)__name__ __module__ __qualname__rrrrr0srzCallable[[], Iterable[_T]]zAsyncGenerator[_T, None]) get_iterable buffer_sizercsdt|dt}ddfdd }t|}z\z }Wn&tyf|djIdH}Yn0t|trvq~q4|Vq4Wd|IdHnd|IdH0dS) 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)maxsizeNone)rcs`z D]}rTWzjtddWqPtyJrBYdSYqYq0qPqdSzj|ddWqtyrYWzjtddWqtyrYdSYq~Yq~0qq~dSYqTYqT0q qTq WzjtddWn"tyrYdSYqYq0q\qnNzjtddWn.tyPrFYYdSYqYn0qZq0dS)z~ Consume the generator in background thread. When items are received, they'll be pushed to the queue. r)timeoutN)putrr)itemr qZquittingrrrunnerGsR         z,generator_to_async_generator..runnerNT) rrr get_nowaitrZrun_in_executorget isinstancer)r r!Zloopr)Zrunner_fr&rr'rr4s" "     )__doc__ __future__rZasyncior contextlibrqueuerrrtypingrr r r r r utilsr__all__rrr__annotations__rrrrrrrs"