U cc@sddlZddlZddlmZddlmZddlmZddlZddlmZddl m Z dd lm Z e e d Zd d ZGd ddejZeedddZeedddZddeedddZGdddZddZddZdd ZdS)!N)Any)Callable) Coroutine)compat)memoized_property)exc gr_contextcCst|t pt|tjtjfSN) isinstance Exceptionasyncio TimeoutErrorZCancelledError)erE/tmp/pip-unpacked-wheel-8u86ls_i/sqlalchemy/util/_concurrency_py3k.pyis_exit_exceptions rc@seZdZddZdS)_AsyncIoGreenletcCs&tj|||||_tr"|j|_dSr )greenlet__init__driver_has_gr_contextr )selffnrrrrr(sz_AsyncIoGreenlet.__init__N)__name__ __module__ __qualname__rrrrrr'sr) awaitablereturncCs(t}t|tstd|j|S)zAwaits an async function in a sync method. The sync method must be inside a :func:`greenlet_spawn` context. :func:`await_only` calls cannot be nested. :param awaitable: The coroutine to call. zjgreenlet_spawn has not been called; can't call await_only() here. Was IO attempted in an unexpected place?)r getcurrentr rr MissingGreenletrswitch)rcurrentrrr await_only/s   r$cCs@t}t|ts4t}|r*td||S|j |S)zAwaits an async function in a sync method. The sync method must be inside a :func:`greenlet_spawn` context. :func:`await_fallback` calls cannot be nested. :param awaitable: The coroutine to call. zgreenlet_spawn has not been called and asyncio event loop is already running; can't call await_fallback() here. Was IO attempted in an unexpected place?) rr r rget_event_loop is_runningr r!run_until_completerr")rr#looprrrawait_fallbackGs   r)F)_require_await)rrcst|t}d}zX|j||}|jshd}z|IdH}Wn"tk rZ|jt }Yq X||}q W5|`X|r|st d|S)aGRuns a sync function ``fn`` in a new greenlet. The sync function can then use :func:`await_only` to wait for async functions. :param fn: The sync callable to call. :param \*args: Positional arguments to pass to the ``fn`` callable. :param \*\*kwargs: Keyword arguments to pass to the ``fn`` callable. FTNzThe current operation required an async execution but none was detected. This will usually happen when using a non compatible DBAPI driver. Please ensure that an async DBAPI is used.) rrr rr"Zdead BaseExceptionthrowsysexc_infor Z AwaitRequired)rr*argskwargscontextZswitch_occurredresultvaluerrrgreenlet_spawn_s"  r4c@s(eZdZeddZddZddZdS)AsyncAdaptedLockcCstSr )rLockrrrrmutexszAsyncAdaptedLock.mutexcCst|j|Sr )r)r8acquirer7rrr __enter__szAsyncAdaptedLock.__enter__cOs|jdSr )r8release)rargkwrrr__exit__szAsyncAdaptedLock.__exit__N)rrrrr8r:r>rrrrr5s r5cOs&t}|rtd||||S)for test suite/ util onlyz]for async run coroutine we expect that no greenlet or event loop is running when we start out)r%r&r r'rr/r0r(rrr"_util_async_run_coroutine_functions rAcOsDt}|s$|t|f||Sttts6t|||SdS)r?N) r%r&r'r4r rr rAssertionErrorr@rrr_util_async_runs rCcCs@tjr4z tWStk r0tYSXntSdS)z}vendor asyncio.get_event_loop() for python 3.7 and above. Python 3.10 deprecates get_event_loop() as a standalone. N)rZpy37rZget_running_loop RuntimeErrorZget_event_loop_policyr%rrrrr%s  r%)rr-typingrrrrrZ langhelpersrr hasattrr rrrr$r)r4r5rArCr%rrrrs,        .