e!ddlmZddlZddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z d d lmZd dlmZd dlmZd dlmZe dZej rGddeZd=dZn ddlmZddlmZeedZd>dZGddeZe dd !Ze rd?d%ZnejZd@d(ZdAd)ZdBd+Z dBd,Z!d-d.dCd5Z"Gd6d7Z#dDd9Z$dDd:Z%dEd<Z&dS)F) annotationsN)Context)Any) Awaitable)Callable) Coroutine)Optional) TYPE_CHECKING)TypeVar)memoized_property)exc)Protocol) TypeGuard_Tc<eZdZUded<ded<ddZdd ZddZdS)greenletbooldeadzOptional[Context] gr_contextfnCallable[..., Any]drivercdSNselfrrs Alib/python3.11/site-packages/sqlalchemy/util/_concurrency_py3k.py__init__zgreenlet.__init__$s CargrreturncdSrr)rr#s r throwzgreenlet.throw'4r"valuecdSrr)rr(s r switchzgreenlet.switch*r'r"Nrrrr)r#rr$r)r(rr$r)__name__ __module__ __qualname____annotations__r!r&r*rr"r rr sg %%%%              r"rr$cdSrrrr"r getcurrentr1-s r")r1)rre BaseExceptionrczt|t p%t|tjtjfSr) isinstance Exceptionasyncio TimeoutErrorCancelledError)r2s r is_exit_exceptionr:;s:!Y'' ' : G '"8 9,,r"c"eZdZUded<d dZdS) _AsyncIoGreenletrrrrrrcjtj|||||_tr|j|_dSdSr)rr!r_has_gr_contextrrs r r!z_AsyncIoGreenlet.__init__Ks>$F+++  0$/DOOO 0 0r"Nr+)r,r-r.r/r!rr"r r<r<Hs3JJJ000000r"r<_T_coT) covariant awaitableAwaitable[_T_co]%TypeGuard[Coroutine[Any, Any, _T_co]]cdSrrrAs r iscoroutinerFVs  r"Awaitable[Any]NonecPt|r|dSdSr)rFcloserEs r _safe_cancel_awaitablerK_s49r"cHt}t|tSr)r1r5r<)currents r in_greenletrNfsllG g/ 0 00r" Awaitable[_T]ct}t|ts#t|t jd|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?)r1r5r<rKrMissingGreenletrr*)rArMs r await_onlyrRks`llG g/ 0 0 y)))! =    >  + ++r"c0t}t|tsZt}|r#t |t jd||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?) r1r5r<get_event_loop is_runningrKrrQrun_until_completerr*)rArMloops r await_fallbackrXs llG g/ 0 0 2 ??    "9 - - -%;  &&y111 >  + ++r"F)_require_awaitrCallable[..., _T]argsrrYkwargscZKt|t}d} |j|i|}|jsSd} |d{V}||}n+#t$r|jt j}YnwxYw|jS|`n#|`wxYw|r|stj 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.) r<r1r*rr3r&sysexc_inforr AwaitRequired)rrYr[r\contextswitch_occurredresultr(s r greenlet_spawnrds"r:<<00G O000, /"O /% !.. ! 8 8 8' 7  8 , / NNGN o  G   Ms.B AB %A?<B >A?? B Bc6eZdZed dZd dZdd Zd S)AsyncAdaptedLockr$ asyncio.Lockc(tjSr)r7Lockrs r mutexzAsyncAdaptedLock.mutexs|~~r"rcNt|jSr)rXrkacquirerjs r __enter__zAsyncAdaptedLock.__enter__s dj0022333r"r#rkwrHc8|jdSr)rkrelease)rr#ros r __exit__zAsyncAdaptedLock.__exit__s r"N)r$rgr$r)r#rrorr$rH)r,r-r.r rkrnrrrr"r rfrfs[ 4444 r"rf'Callable[..., Coroutine[Any, Any, Any]]ct}|rtd|||i|S)for test suite/ util onlyz]for async run coroutine we expect that no greenlet or event loop is running when we start out)rTrUr6rVrr[r\rWs r "_util_async_run_coroutine_functionrxs]   D   0     " "22t#6v#6#6 7 77r"ct}|s#|t|g|Ri|St t t sJ||i|S)rv)rTrUrVrdr5r1r<rws r _util_async_runrzs   D ??  #&&~b'J4'J'J'J6'J'JKKK*,,(899999r4"6"""r"asyncio.AbstractEventLoopc tjS#t$rYnwxYwtjS)z}vendor asyncio.get_event_loop() for python 3.7 and above. Python 3.10 deprecates get_event_loop() as a standalone. )r7get_running_loop RuntimeErrorget_event_loop_policyrTrr"r rTrTsS  ')))        ( * * 9 9 ; ;;s  "")r$r)r2r3r$r)rArBr$rC)rArGr$rHrs)rArOr$r) rrZr[rrYrr\rr$r)rrtr[rr\rr$r)r$r{)' __future__rr7 contextvarsrr^typingrrrrr r r langhelpersr r util.typingrrrrr1hasattrr>r:r<r?rFrKrNrRrXrdrfrxrzrTrr"r rs#"""""  ******""""""###### WT]] "     8        $#####!!!!!! '**,, 5500000x000 4(((&     %K1111 ,,,,4,,,,F!//////d         8 8 8 8 # # # # < < < < <