o Df]@sUdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZddlmZd d gZed e e d fd Zeddd ZdZded<edZGdddZefddd Zd 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]cCs.z |VW|IdHdS|IdHw)z1Similar to `contextlib.aclosing`, in Python 3.10.N)aclose)rra/home/ubuntu/webapp/venv/lib/python3.10/site-packages/prompt_toolkit/eventloop/async_generator.pyrs"iintDEFAULT_BUFFER_SIZE_Tc@s eZdZdS)_DoneN)__name__ __module__ __qualname__rrrrr1sr get_iterableCallable[[], Iterable[_T]] buffer_sizeAsyncGenerator[_T, None]csdt|dt}d fdd }t|}z/ z}Wnty3|djIdH}Ynwt|tr:n|VqWd|IdHdSd|IdHw) 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)maxsizerNonecs,zvD]V}r%W z jtddWdSty$r"YdSYq w z j|ddWn)tyXrVYW z jtddWdStyUrSYdSYq;wYq%w qW z jtddWdStyvrtYdSYq\w z jtddW wtyrYYdSYqww)z~ Consume the generator in background thread. When items are received, they'll be pushed to the queue. Tr )timeoutN)putrr)itemr qquittingrrrunnerHsf       z,generator_to_async_generator..runnerTN)rr%) rrr get_nowaitrrun_in_executorget isinstancer)r r"loopr,runner_fr(rr)rr5s, "    )rrrr)r r!r"rrr#)__doc__ __future__rasyncior contextlibrqueuerrrtypingrr r r r utilsr__all__rrr__annotations__rrrrrrrs$