U ou f\@sUdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZddlmZd d gZed e e d fd Zed dddd ZdZded<edZGdddZefdddddd Zd S)z& Implementation for async generators. ) annotations)get_running_loop)asynccontextmanager)EmptyFullQueue)AnyAsyncGeneratorCallableIterableTypeVar)run_in_executor_with_contextaclosinggenerator_to_async_generator _T_GeneratorN)boundz"AsyncGenerator[_T_Generator, None])thingreturncCs z |VW5|IdHXdS)z1Similar to `contextlib.aclosing`, in Python 3.10.N)aclose)rrL/tmp/pip-unpacked-wheel-8milen2s/prompt_toolkit/eventloop/async_generator.pyrs iintDEFAULT_BUFFER_SIZE_Tc@s eZdZdS)_DoneN)__name__ __module__ __qualname__rrrrr0srzCallable[[], Iterable[_T]]zAsyncGenerator[_T, None]) get_iterable buffer_sizercsdt|dt}ddfdd }t|}zPz }Wn(tk rh|djIdH}YnXt|trxqq4|Vq4W5d|IdHXdS) 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)rcsdzbD]V}rWNdSzj|ddWq`tk rZrRYWdSYqYqXq qq W5zjtddWqtk rrYdSYqfYqfXqqfXdS)z~ Consume the generator in background thread. When items are received, they'll be pushed to the queue. r )timeoutN)putrr)itemrqZquittingrrrunnerGs$      z,generator_to_async_generator..runnerTN) rrr get_nowaitrZrun_in_executorget isinstancer)rr Zloopr(Zrunner_fr%rr&rr4s "   )__doc__ __future__rZasyncior contextlibrqueuerrrtypingrr r r r utilsr__all__rrr__annotations__rrrrrrrs"