U
|
¸ý°d ! ã @ sä d dl mZ d dlZd dlmZ d dlZd dlZd dlmZ d dlmZ d dlm Z d dlm
|
Z
|
d dlm Z d d lm Z d d
|
lm Z d d lmZ d dlmZ d dlmZ d dlmZ e dZej râG dd deZddddZnd dlmZ d dlmZ ee dZddd d d ZG d d d eZe d!d"d#Z e rJd$d%d&d'd(Z nej Z d)d*d&d+d,Z d-dd&d.d/Z d-dd&d0d1Z d2d3d4d5dd5dd6d7d8Z!G d9d: d:Z"d;d5d5d5d<d=d>Z#d;d5d5d5d<d?d@Z$dAddBdCZ%dS )Dé )Ú annotationsN)ÚContext)ÚAny)Ú Awaitable)ÚCallable)Ú Coroutine)ÚOptional)Ú TYPE_CHECKING)ÚTypeVaré )Úmemoized_propertyé )Úexc)ÚProtocol)Ú TypeGuardÚ_Tc @ sN e Zd ZU ded< ded< dd dddZd d d
|
d d Zd d d ddZdS )ÚgreenletÚboolÚdeadzOptional[Context]Ú
|
gr_contextúCallable[..., Any]©ÚfnÚdriverc C s d S ©N© ©Úselfr r r r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/util/_concurrency_py3k.pyÚ__init__% s zgreenlet.__init__r )ÚargÚreturnc G s d S r r )r r r r r Úthrow( s zgreenlet.throw)Úvaluer! c C s d S r r )r r# r r r Úswitch+ s zgreenlet.switchN)Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__r r" r$ r r r r r s
|
|
r ©r! c C s d S r r r r r r Ú
|
getcurrent. s r* )r* )r r Ú BaseExceptionr )Úer! c C s t | t p t | tjtjfS r )Ú
|
isinstanceÚ ExceptionÚasyncioÚ TimeoutErrorZCancelledError)r, r r r Úis_exit_exception< s
|
ÿr1 c @ s& e Zd ZU ded< dddddZdS ) Ú_AsyncIoGreenletr r r r r c C s$ t | ||¡ || _tr |j| _d S r )r r r Ú_has_gr_contextr r r r r r L s z_AsyncIoGreenlet.__init__N)r% r& r' r( r r r r r r2 I s
|
r2 Ú_T_coT)Ú covariantzAwaitable[_T_co]z%TypeGuard[Coroutine[Any, Any, _T_co]])Ú awaitabler! c C s d S r r ©r6 r r r Ú iscoroutineW s r8 zAwaitable[Any]ÚNonec C s t | r| ¡ d S r )r8 Úcloser7 r r r Ú_safe_cancel_awaitable` s r; z Awaitable[_T]c C s. t }t|ts"t| t d¡|j | ¡S )zÓAwaits 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?)r* r- r2 r; r ÚMissingGreenletr r$ )r6 Úcurrentr r r Ú
|
await_onlyg s
|
|
ÿ r> c C sF t }t|ts:t }| ¡ r0t| t d¡| | ¡S |j
|
| ¡S )z×Awaits 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.
|
|
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?) r* r- r2 Úget_event_loopÚ
|
is_runningr; r r< Úrun_until_completer r$ )r6 r= Úloopr r r Úawait_fallback s
|
ÿ
|
rC F)Ú_require_awaitzCallable[..., _T]r )r ÚargsrD Úkwargsr! c Ï s t | t }d}zX|j||}|jsfd}z|I dH }W n" tk
|
rX |jt ¡ }Y q X | |¡}q W 5 |`X |r|st
|
d¡|S )aG Runs 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.
|
FTNz¶The 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.) r2 r* r r$ r r+ r" ÚsysÚexc_infor Z AwaitRequired)r rD rE rF ÚcontextZswitch_occurredÚresultr# r r r Úgreenlet_spawn s" ÿrK c @ s> e Zd ZeddddZddddZddd d
|
d d Zd S )ÚAsyncAdaptedLockz asyncio.Lockr) c C s t ¡ S r )r/ ÚLock©r r r r ÚmutexÐ s zAsyncAdaptedLock.mutexr c C s t | j ¡ S r )rC rO ÚacquirerN r r r Ú __enter__Ö s zAsyncAdaptedLock.__enter__r r9 )r Úkwr! c O s | j ¡ d S r )rO Úrelease)r r rR r r r Ú__exit__Û s zAsyncAdaptedLock.__exit__N)r% r& r' r rO rQ rT r r r r rL Ï s rL z'Callable[..., Coroutine[Any, Any, Any]])r rE rF r! c O s& t }| ¡ rtd| | ||¡S )úfor test suite/ util onlyz]for async run coroutine we expect that no greenlet or event loop is running when we start out)r? r@ r. rA ©r rE rF rB r r r Ú"_util_async_run_coroutine_functionß s ÿrW c O sB t }| ¡ s$| t| f||¡S tt ts4t| ||S dS )rU N)r? r@ rA rK r- r* r2 ÚAssertionErrorrV r r r Ú_util_async_runí s
|
rY zasyncio.AbstractEventLoopc C s0 z
|
t ¡ W S tk
|
r* t ¡ ¡ Y S X dS )z}vendor asyncio.get_event_loop() for python 3.7 and above.
|
|
Python 3.10 deprecates get_event_loop() as a standalone.
|
|
N)r/ Zget_running_loopÚ RuntimeErrorZget_event_loop_policyr? r r r r r? ü s
|
r? )&Ú
|
__future__r r/ Z contextvarsr rG Útypingr r r r r r r
|
Z langhelpersr Ú r Z util.typingr r r r r* Úhasattrr3 r1 r2 r4 r8 r; r> rC rK rL rW rY r? r r r r Ú<module> sJ
|
ý2
|
