rj"dZddlmZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z d d lmZddlmZe dZd$dZdZd%dZGddZeZe re ddZ d&dZn ej2Zd'dZd(dZd)dZeZdd d*dZGd d!ZGd"d#Z y)+z&asyncio-related concurrency functions.) annotationsN)Any) Awaitable)Callable) Coroutine)NoReturn) TYPE_CHECKING) TypeGuard)TypeVar)memoized_property)exc_Tc|t|t xs*t|tjtjfSN) isinstance Exceptionasyncio TimeoutErrorCancelledError)es f/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/sqlalchemy/util/concurrency.pyis_exit_exceptionrs8!Y' ' : G '"8"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]'c ttr) ImportError_ERROR_MESSAGE)argkws r_not_implementedr!.s n %%rc0eZdZdZdZddd dZdZd dZy) _concurrency_shim_clszLate import shim for greenlet) _has_greenletgreenlet_AsyncIoGreenlet getcurrentTraise_chdtvryts ddlmaddlmad|_t tdaGdd tat|_t|_t|_y#t$r;}tsdad |_|j|rtt|Yd}~yd}~wwxYw) z(Import greenlet and initialize the classr%Nr)r')r%T gr_contextc$eZdZUded<dZddZy)r&booldeadTcbtj|||tr|j|_yyr)r%__init___has_gr_contextr+)selffndrivers rr0z_AsyncIoGreenlet.__init__`s)%%dB7&*0*;*;'rN)r3zCallable[..., Any]r4r%)__name__ __module__ __qualname____annotations__ __sqlalchemy_greenlet_provider__r0rrr&r&[s 370getattrAttributeError)r2keys r __getattr__z!_concurrency_shim_cls.__getattr__ns2 $..    4% % % %rN)r)r-returnNone)rEstrrGr)r5r6r7__doc__rBr>r=rFr:rrr#r#2s!'I-1+5Z1 &rr#_T_coT) covariantcyrr: awaitables r iscoroutinerP{s03rc<t|r|jyyr)rPcloserNs r_safe_cancel_awaitablerSs9rcDtj}t|ddS)Nr9F)_concurrency_shimr'rC)currents r in_greenletrWs **,G 7> FFrctj}t|dds t|t j d|j sJ|j 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. r9Fzfgreenlet_spawn has not been called; can't call await_() here. Was IO attempted in an unexpected place?)rUr'rCrSrMissingGreenletparentswitch)rOrVs rawait_r\sb **,G 7> Fy)!! =   >>> >>  ++rF)_require_awaitcKtj|tj}d}|j|i|}|js+d} |d{}|j|}|js+|r|stjd|S7=#t $r$|j tj}YVwxYww)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.) rUr&r'r[r. BaseExceptionthrowsysexc_infor AwaitRequired)r3r]argskwargscontextswitch_occurredresultvalues rgreenlet_spawnrjs" 00  ( ( *GO W^^T ,V ,Fll +!LE^^E*Fllo G  M! 4#W]]CLLN3F  4sBA CBBBC6CB*C=C?CCc.eZdZeddZddZddZy)AsyncAdaptedLockc*tjSr)rLockr@s rmutexzAsyncAdaptedLock.mutexs||~rcHt|jjSr)r\roacquirer@s r __enter__zAsyncAdaptedLock.__enter__sdjj((*++rc8|jjyr)rorelease)r2rr s r__exit__zAsyncAdaptedLock.__exit__s rN)rGz asyncio.LockrGr-)rrr rrGrH)r5r6r7r rorrrur:rrrlrls , rrlcPeZdZdZddZ ddZ d dZddZy) _AsyncUtilz&Asyncio util for test suite/ util onlyc6tj|_yr)rRunnerrunnerr@s rr0z_AsyncUtil.__init__snn& rcD|jj||i|S)zRun coroutine on the loop)r{runr2r3rdres rr}z_AsyncUtil.runs!{{r426233rc&tjdtjrd|jj j rt sJ||i|S|jjt|g|i|S||i|S)z3Run sync function in greenlet. Support nested callsFr() rUr>r$r{get_loop is_runningrWr}rjr~s rrun_in_greenletz_AsyncUtil.run_in_greenlets %%U%3  * *{{##%002"}$}4*6**{{~b'J4'J6'JKKt&v& &rc8|jjyr)r{rRr@s rrRz_AsyncUtil.closes rN)rGrH)r3z&Callable[..., Coroutine[Any, Any, _T]]rdrrerrGr)r3Callable[..., _T]rdrrerrGr)r5r6r7rJr0r}rrRr:rrrxrxs_0'4 244 4 4'#',/';>' ' rrx)rr_rGr-)rrr rrGr)rOzAwaitable[_T_co]rGz%TypeGuard[Coroutine[Any, Any, _T_co]])rOzAwaitable[Any]rGrHrv)rOz Awaitable[_T]rGr) r3rrdrr]r-rerrGr)!rJ __future__rrratypingrrrrrr r r langhelpersr rrrrr!r#rUrKrPrSrWr\ await_onlyrjrlrxr:rrrs-"  * T](&A&A&H*+ Gt ,E4#4 .4 %%KG ,6 !// // /  /d   !!r