o g#@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]drivercCdSNselfrrrr]/var/www/html/flaskapp/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<dZd dd Zd S) _AsyncIoGreenletrrTrrrrcCs"t|||tr|j|_dSdSr)rr#_has_gr_contextrr rrr"r#Os z_AsyncIoGreenlet.__init__Nr+)r,r-r.r/ __sqlalchemy_greenlet_provider__r#rrrr"r9Js r9_T_coT) covariant awaitableAwaitable[_T_co]%TypeGuard[Coroutine[Any, Any, _T_co]]cCrrrr>rrr" iscoroutineYsrBAwaitable[Any]NonecCst|r |dSdSr)rBcloserArrr"_safe_cancel_awaitableas rFcCst}t|ddS)Nr;F)r0getattr)currentrrr" in_greenleths rI Awaitable[_T]cCs0t}t|ddst|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. r;Fzjgreenlet_spawn has not been called; can't call await_only() here. Was IO attempted in an unexpected place?)r0rGrFrMissingGreenletparentr*)r>rHrrr" await_onlyms  rMcCsHt}t|ddst}|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`. r;Fzgreenlet_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?) r0rGget_event_loop is_runningrFrrKrun_until_completerLr*)r>rHlooprrr"await_fallbacks   rRF)_require_awaitrCallable[..., _T]argsrrSkwargscst|t}d}|j|i|}|js6d}z|IdH}Wnty-|jt}Ynw||}|jr|r?|s?t 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_infor AwaitRequired)rrSrUrVcontextswitch_occurredresultr)rrr"greenlet_spawns$   r]c@s.eZdZedddZdddZdd d ZdS)AsyncAdaptedLockr& asyncio.LockcCstSr)r5Lockr!rrr"mutexszAsyncAdaptedLock.mutexrcCst|jSr)rRrbacquirerarrr" __enter__szAsyncAdaptedLock.__enter__r%rkwrDcOs|jdSr)rbrelease)r!r%rerrr"__exit__szAsyncAdaptedLock.__exit__N)r&r_r&r)r%rrerr&rD)r,r-r.r rbrdrgrrrr"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_policyrNrrrr"rNs    rNc@s`eZdZUdZded<dddZdd d Zd ddZdddZd!ddZ d"ddZ dddZ dS)#_Runnerz#Runner implementation for test onlyz6Union[None, asyncio.AbstractEventLoop, Literal[False]]_loopr&rDcCs d|_dSr)rnrarrr"r#s z_Runner.__init__rcCs ||Sr) _lazy_initrarrr"rdsz_Runner.__enter__exc_typerexc_valexc_tbcCs |dSr)rE)r!rprqrrrrr"rgs z_Runner.__exit__cCsF|jr!z|j|jW|jd|_dS|jd|_wdS)NF)rnrPshutdown_asyncgensrErarrr"rEs   z _Runner.closericCs||js J|jS)zReturn embedded event loop.)rornrarrr"get_loops z_Runner.get_loopcoroCoroutine[Any, Any, _T]rcCs||js J|j|Sr)rornrP)r!rurrr"runs  z _Runner.runcCs.|jdur td|jdurt|_dSdS)NFzRunner is closed)rnrkr5new_event_looprarrr"ros  z_Runner._lazy_initN)r&rD)r&r)rprrqrrrrr&rDr&ri)rurvr&r) r,r-r.__doc__r/r#rdrgrErtrwrorrrr"rms      rm)r&r)r1r2r&r)r>r?r&r@)r>rCr&rDrh)r>rJr&r) rrTrUrrSrrVrr&rry), __future__rr5 contextvarsrrWtypingrrrrrr r r langhelpersr rutilr util.typingrrrrrrr0hasattrr:r8r9r<rBrFrIrMrRr]r^rNRunnerrmrrrr"sX                           #/