o Df#@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]drivercCdSNselfrrrrZ/home/ubuntu/webapp/venv/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 TimeoutErrorCancelledError)r1rrr"is_exit_exception=s r8c@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"r9Js r9_T_coT) covariant awaitableAwaitable[_T_co]%TypeGuard[Coroutine[Any, Any, _T_co]]cCrrrr=rrr" iscoroutineXsrAAwaitable[Any]NonecCst|r |dSdSr)rAcloser@rrr"_safe_cancel_awaitable`s rEcCst}t|tSr)r0r3r9)currentrrr" in_greenletgs rG 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?)r0r3r9rErMissingGreenletrr*)r=rFrrr" await_onlyls  rJcCsFt}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?) r0r3r9get_event_loop is_runningrErrIrun_until_completerr*)r=rFlooprrr"await_fallbacks   rOF)_require_awaitrCallable[..., _T]argsrrPkwargscst|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.) r9r0r*rr2r'sysexc_inforr AwaitRequired)rrPrRrScontextswitch_occurredresultr)rrr"greenlet_spawns*    rZc@s.eZdZedddZdddZdd d ZdS)AsyncAdaptedLockr& asyncio.LockcCstSr)r5Lockr!rrr"mutexszAsyncAdaptedLock.mutexrcCst|jSr)rOr_acquirer^rrr" __enter__szAsyncAdaptedLock.__enter__r%rkwrCcOs|jdSr)r_release)r!r%rbrrr"__exit__szAsyncAdaptedLock.__exit__N)r&r\r&r)r%rrbrr&rC)r,r-r.r r_rardrrrr"r[s   r[asyncio.AbstractEventLoopcCs*ztWStyYnwtS)z}vendor asyncio.get_event_loop() for python 3.7 and above. Python 3.10 deprecates get_event_loop() as a standalone. )r5get_running_loop RuntimeErrorget_event_loop_policyrKrrrr"rKs    rKc@s`eZdZUdZded<dddZdd d Zd ddZdddZd!ddZ d"ddZ dddZ dS)#_Runnerz#Runner implementation for test onlyz6Union[None, asyncio.AbstractEventLoop, Literal[False]]_loopr&rCcCs d|_dSr)rkr^rrr"r#s z_Runner.__init__rcCs ||Sr) _lazy_initr^rrr"rasz_Runner.__enter__exc_typerexc_valexc_tbcCs |dSr)rD)r!rmrnrorrr"rds z_Runner.__exit__cCsF|jr!z|j|jW|jd|_dS|jd|_wdS)NF)rkrMshutdown_asyncgensrDr^rrr"rD s   z _Runner.closerfcCs||js J|jS)zReturn embedded event loop.)rlrkr^rrr"get_loops z_Runner.get_loopcoroCoroutine[Any, Any, _T]rcCs||js J|j|Sr)rlrkrM)r!rrrrr"runs  z _Runner.runcCs.|jdur td|jdurt|_dSdS)NFzRunner is closed)rkrhr5new_event_loopr^rrr"rls  z_Runner._lazy_initN)r&rC)r&r)rmrrnrrorr&rCr&rf)rrrsr&r) r,r-r.__doc__r/r#rardrDrqrtrlrrrr"rjs      rj)r&r)r1r2r&r)r=r>r&r?)r=rBr&rCre)r=rHr&r) rrQrRrrPrrSrr&rrv), __future__rr5 contextvarsrrTtypingrrrrrr r r langhelpersr rutilr util.typingrrrrrrr0hasattrr:r8r9r;rArErGrJrOrZr[rKRunnerrjrrrr"sX                           #2