o Zh#@sddlmZddlZddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZd d lmZddlmZddlmZddlmZddlmZddlmZddlmZe dZej rGdddeZdIddZn ddlmZddlmZeedZdJd!d"ZGd#d$d$eZ e d%d&d'Z!e rdKd+d,Z"nej"Z"dLd/d0Z#dMd1d2Z$dNd4d5Z%dNd6d7Z&d8d9dOd@dAZ'GdBdCdCZ(dPdEdFZ)e serej*Z+dSGdGdHdHZ+dS)Q) annotationsN)Context)Any) Awaitable)Callable) Coroutine)Optional) TYPE_CHECKING)TypeVar)Union)memoized_property)exc)py311)Literal)Protocol)Self) TypeGuard_Tc@s<eZdZUded<ded<ddd Zdd dZdddZdS)greenletbooldeadzOptional[Context] gr_contextfnCallable[..., Any]drivercCdSNselfrrrrX/var/www/html/lang_env/lib/python3.10/site-packages/sqlalchemy/util/_concurrency_py3k.py__init__(zgreenlet.__init__argrreturncGrrr)r!r%rrr"throw*zgreenlet.throwvaluecCrrr)r!r)rrr"switch-r(zgreenlet.switchNrrrr)r%rr&r)r)rr&r)__name__ __module__ __qualname____annotations__r#r'r*rrrr"r$s   rr&cCrrrrrrr" getcurrent0r$r0)r0)rre BaseExceptionrcCst|t pt|tjtjfSr) isinstance Exceptionasyncio TimeoutErrorZCancelledError)r1rrr"is_exit_exception=s r7c@s eZdZUded<d ddZd S) _AsyncIoGreenletrrrrrrcCs(t|||||_tr|j|_dSdSr)rr#r_has_gr_contextrr rrr"r#Ms  z_AsyncIoGreenlet.__init__Nr+)r,r-r.r/r#rrrr"r8Js r8_T_coT) covariant awaitableAwaitable[_T_co]%TypeGuard[Coroutine[Any, Any, _T_co]]cCrrrr<rrr" iscoroutineXsr@Awaitable[Any]NonecCst|r |dSdSr)r@closer?rrr"_safe_cancel_awaitable`s rDcCst}t|tSr)r0r3r8)currentrrr" in_greenletgs rF Awaitable[_T]cCs.t}t|tst|td|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?)r0r3r8rDrMissingGreenletrr*)r<rErrr" await_onlyls  rIcCsFt}t|tst}|rt|td||S|j |S)aAwaits 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. .. deprecated:: 2.0.24 The ``await_fallback()`` function will be removed in SQLAlchemy 2.1. Use :func:`_util.await_only` instead, running the function / program / etc. within a top-level greenlet that is set up using :func:`_util.greenlet_spawn`. 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?) r0r3r8get_event_loopZ is_runningrDrrHrun_until_completerr*)r<rElooprrr"await_fallbacks   rMF)_require_awaitrCallable[..., _T]argsrrNkwargscst|t}d}z1|j|i|}|js7d}z|IdH}Wnty.|jt}Ynw||}|jrW|`n|`w|rG|sGt 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.) r8r0r*rr2r'sysexc_inforrZ AwaitRequired)rrNrPrQcontextZswitch_occurredresultr)rrr"greenlet_spawns*    rVc@s.eZdZedddZdddZdd d ZdS)AsyncAdaptedLockr& asyncio.LockcCstSr)r5Lockr!rrr"mutexszAsyncAdaptedLock.mutexrcCst|jSr)rMr[acquirerZrrr" __enter__szAsyncAdaptedLock.__enter__r%rkwrBcOs|jdSr)r[release)r!r%r^rrr"__exit__szAsyncAdaptedLock.__exit__N)r&rXr&r)r%rr^rr&rB)r,r-r.r r[r]r`rrrr"rWs   rWasyncio.AbstractEventLoopcCs*ztWStyYnwtS)z}vendor asyncio.get_event_loop() for python 3.7 and above. Python 3.10 deprecates get_event_loop() as a standalone. )r5Zget_running_loop RuntimeErrorZget_event_loop_policyrJrrrr"rJs    rJc@s`eZdZUdZded<dddZdd d Zd ddZdddZd!ddZ d"ddZ dddZ dS)#_Runnerz#Runner implementation for test onlyz6Union[None, asyncio.AbstractEventLoop, Literal[False]]_loopr&rBcCs d|_dSr)rerZrrr"r#s z_Runner.__init__rcCs ||Sr) _lazy_initrZrrr"r]sz_Runner.__enter__exc_typerexc_valexc_tbcCs |dSr)rC)r!rgrhrirrr"r`s z_Runner.__exit__cCsF|jr!z|j|jW|jd|_dS|jd|_wdS)NF)rerKZshutdown_asyncgensrCrZrrr"rC s   z _Runner.closerbcCs||js J|jS)zReturn embedded event loop.)rfrerZrrr"get_loops z_Runner.get_loopcoroCoroutine[Any, Any, _T]rcCs||js J|j|Sr)rfrerK)r!rkrrr"runs  z _Runner.runcCs.|jdur td|jdurt|_dSdS)NFzRunner is closed)rercr5Znew_event_looprZrrr"rfs  z_Runner._lazy_initN)r&rB)r&r)rgrrhrrirr&rBr&rb)rkrlr&r) r,r-r.__doc__r/r#r]r`rCrjrmrfrrrr"rds      rd)r&r)r1r2r&r)r<r=r&r>)r<rAr&rBra)r<rGr&r) rrOrPrrNrrQrr&rrn), __future__rr5Z contextvarsrrRtypingrrrrrr r r Z langhelpersr rutilrZ util.typingrrrrrrr0hasattrr9r7r8r:r@rDrFrIrMrVrWrJZRunnerrdrrrr"sX                           #2