o jx@sUdZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z mZddlmZmZdd lmZd d gZed eed fdZe ddd ZdZded<edZGdddZefddd Zd S) z& Implementation for async generators. ) annotations)get_running_loop)AsyncGeneratorCallableIterable)asynccontextmanager)EmptyFullQueue)AnyTypeVar)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)rr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/eventloop/async_generator.pyrs"iintDEFAULT_BUFFER_SIZE_Tc@s eZdZdS)_DoneN)__name__ __module__ __qualname__rrrrr2sr 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 qquittingrrrunnerIsf       z,generator_to_async_generator..runnerTN)rr%) r rr get_nowaitrrun_in_executorget isinstancer)r r"loopr,runner_fr(rr)rr6s, "    )rrrr)r r!r"rrr#)__doc__ __future__rasynciorcollections.abcrrr contextlibrqueuerr r typingr r utilsr__all__rrr__annotations__rrrrrrrs&