j"dZddlmZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z d d lmZddlmZe dZd7dZdZd8dZGddZeZe re dd Zd9d$ZnejZd:d'Zd;d(Zdz&asyncio-related concurrency functions.) annotationsN)Any) Awaitable)Callable) Coroutine)NoReturn) TYPE_CHECKING) TypeGuard)TypeVar)memoized_property)exc_Te BaseExceptionreturnboolczt|t p%t|tjtjfSN) isinstance Exceptionasyncio TimeoutErrorCancelledError)rs /home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/sqlalchemy/util/concurrency.pyis_exit_exceptionrs:!Y'' ' : G '"8 9,,zThe SQLAlchemy asyncio module requires that the Python 'greenlet' library is installed. In order to ensure this dependency is available, use the 'sqlalchemy[asyncio]' install target: 'pip install sqlalchemy[asyncio]'argrkwrc*ttr) ImportError_ERROR_MESSAGE)rr s r_not_implementedr$.s n % %%rc2eZdZdZdZdddd Zd ZddZdS)_concurrency_shim_clszLate import shim for greenlet) _has_greenletgreenlet_AsyncIoGreenlet getcurrentTraise_r,rrNonecdtvrdSts ddlmaddlmad|_t tdaGdd tat|_t|_t|_dS#t$rF}tsdad |_| |rtt|Yd}~dSd}~wwxYw) z(Import greenlet and initialize the classr(Nr)r*)r(T gr_contextc&eZdZUded<dZd dZd S) r)rdeadTfnCallable[..., Any]driverr(cht|||tr|j|_dSdSr)r(__init___has_gr_contextr/)selfr2r4s rr6z_AsyncIoGreenlet.__init__`s:%%dB777&<*0*;<rrr&r&2sh''I-1+5+5+5+5+5+5Z111 &&&&&&rr&_T_coT) covariant awaitableAwaitable[_T_co]%TypeGuard[Coroutine[Any, Any, _T_co]]cdSrr>rOs r iscoroutinerT{s03rAwaitable[Any]r-cPt|r|dSdSr)rTcloserSs r_safe_cancel_awaitablerXs49rcVt}t|ddS)Nr=F)_concurrency_shimr*rI)currents r in_greenletr\s&**,,G 7> F FFr Awaitable[_T]ct}t|dds#t|t jd|jsJ|j|S)zAwaits an async function in a sync method. The sync method must be inside a :func:`greenlet_spawn` context. :func:`await_` calls cannot be nested. :param awaitable: The coroutine to call. r=Fzfgreenlet_spawn has not been called; can't call await_() here. Was IO attempted in an unexpected place?)rZr*rIrXrMissingGreenletparentswitch)rOr[s rawait_rbsx **,,G 7> F F y)))! =    > >  + ++rF)_require_awaitr2Callable[..., _T]argsrckwargscpKt|t}d}|j|i|}|jsSd} |d{V}||}n+#t $r|jtj}YnwxYw|jS|r|stj d|S)aCRuns a sync function ``fn`` in a new greenlet. The sync function can then use :func:`await_` 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.) rZr)r*rar1rthrowsysexc_infor AwaitRequired)r2rcrerfcontextswitch_occurredresultvalues rgreenlet_spawnrps" 00  ( ( * *GO W^T ,V , ,Fl + +!LLLLLLE^^E**FF  4 4 4#W]CLNN3FFF  4 l + o  G   MsA,,%BBc6eZdZed dZd dZdd Zd S)AsyncAdaptedLockr asyncio.Lockc(tjSr)rLockrDs rmutexzAsyncAdaptedLock.mutexs|~~rrcNt|jSr)rbrvacquirerDs r __enter__zAsyncAdaptedLock.__enter__s dj((**+++rrrr r-c8|jdSr)rvrelease)r8rr s r__exit__zAsyncAdaptedLock.__exit__s rN)rrsrr)rrr rrr-)r9r:r;r rvryr|r>rrrrrrs[ ,,,, rrrc2eZdZdZddZdd Zdd ZddZdS) _AsyncUtilz&Asyncio util for test suite/ util onlyrr-c6tj|_dSr)rRunnerrunnerrDs rr6z_AsyncUtil.__init__sn&& rr2&Callable[..., Coroutine[Any, Any, _T]]rerrfrcB|j||i|S)zRun coroutine on the loop)rrunr8r2rerfs rrz_AsyncUtil.runs'{rr42622333rrdc6tdtjrk|jrt sJ||i|S|jt|g|Ri|S||i|S)z3Run sync function in greenlet. Support nested callsFr+) rZrBr'rget_loop is_runningr\rrprs rrun_in_greenletz_AsyncUtil.run_in_greenlets %%U%333  * '{##%%0022 L"}}$$$r4*6***{~b'J4'J'J'J6'J'JKKK2t&v&& &rc8|jdSr)rrWrDs rrWz_AsyncUtil.closes rN)rr-)r2rrerrfrrr)r2rdrerrfrrr)r9r:r;rLr6rrrWr>rrrrsj00''''4444'''' rr)rrrr)rrr rrr)rOrPrrQ)rOrUrr-r})rOr]rr) r2rdrerrcrrfrrr)!rL __future__rrritypingrrrrrr r r langhelpersr rrrr#r$r&rZrMrTrXr\rb await_onlyrprrrr>rrrsj-,""""""  ****** WT]](&&&&A&A&A&A&A&A&A&A&H*)++& GGt , , ,E44444 %KGGGG ,,,,6 !//////d         !!!!!!!!!!r