U
|
¸ý°dÙ ã @ sö d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ ddlmZ dd lm Z dd
|
l
|
m Z dd l m Z ejs~e sd d lmZ d dlmZ nd d lmZ d dlmZ edeg ef dZdddddZG dd de ZG dd dZdS )é )Ú annotationsN)ÚAny)ÚCallable)ÚOptional)ÚTypeVaré )Úexc)Úutil)ÚHAS_CYEXTENSION)ÚProtocolé )Ú_distill_params_20)Ú_distill_raw_paramsÚ_C)ÚboundÚstrzCallable[[_C], _C])ÚkeyÚreturnc s t j fdd}|S )zËDecorator, memoize a function in a connection.info stash.
|
|
Only applicable to functions which take no arguments other than a
|
connection. The memo will be stored in ``connection.info[key]``.
|
c sF | ¡ }z |j W S tk
|
r@ | || |j < }| Y S X d S ©N)ÚconnectÚinfoÚKeyError)ÚfnÚselfÚ
|
connectionÚval©r © úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/engine/util.pyÚ decorated* s z%connection_memoize.<locals>.decorated)r Ú decorator)r r r r r Úconnection_memoize# s r! c @ s e Zd ZU ded< dS )Ú _TConsSubjectz Optional[TransactionalContext]Ú_trans_context_managerN)Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__r r r r r" 6 s
|
r" c @ s¶ e Zd ZU dZdZded< ddddZddd d
|
Zddd d Zd dddZ ddddZ
|
ddddZ ddddZ e d ddddZd dddZd d d dd d d Zd S )!ÚTransactionalContextzÀApply Python context manager behavior to transaction objects.
|
|
Performs validation to ensure the subject of the transaction is not
|
used if the transaction were ended prematurely.
|
|
)Ú_outer_trans_ctxÚ_trans_subjectÚ __weakref__zOptional[_TConsSubject]r* Úbool)r c C s
|
t d S r ©ÚNotImplementedError©r r r r Ú_transaction_is_activeF s z+TransactionalContext._transaction_is_activec C s
|
t d S r r- r/ r r r Ú_transaction_is_closedI s z+TransactionalContext._transaction_is_closedc C s
|
t dS )a indicates the object is in a state that is known to be acceptable
|
for rollback() to be called.
|
|
This does not necessarily mean rollback() will succeed or not raise
|
an error, just that there is currently no state detected that indicates
|
rollback() would fail or emit warnings.
|
|
It also does not mean that there's a transaction in progress, as
|
it is usually safe to call rollback() even if no transaction is
|
present.
|
|
.. versionadded:: 1.4.28
|
|
Nr- r/ r r r Ú_rollback_can_be_calledL s z,TransactionalContext._rollback_can_be_calledr" c C s
|
t d S r r- r/ r r r Ú _get_subject] s z!TransactionalContext._get_subjectÚNonec C s
|
t d S r r- r/ r r r Úcommit` s zTransactionalContext.commitc C s
|
t d S r r- r/ r r r Úrollbackc s z TransactionalContext.rollbackc C s
|
t d S r r- r/ r r r Úclosef s zTransactionalContext.close)Úsubjectr c C s |j }|r | ¡ s t d¡d S )NzCan't operate on closed transaction inside context manager. Please complete the context manager before emitting further commands.)r# r0 r ZInvalidRequestError)Úclsr8 Ú trans_contextr r r Ú_trans_ctx_checki s ÿz%TransactionalContext._trans_ctx_checkc C s$ | ¡ }|j}|| _|| _| |_| S r )r3 r# r) r* )r r8 r: r r r Ú __enter__t s z TransactionalContext.__enter__r )Útype_ÚvalueÚ tracebackr c C sô t | dd }|d kp |j| k }|d kr| ¡ rzBz | ¡ W n0 t ¡ | ¡ r^|
|
¡ W 5 Q R X Y nX W 5 |s|d k st| j|_d | _| _X nVz.| ¡ s¶| ¡ sÆ| ¡ n| ¡ rÆ|
|
¡ W 5 |sâ|d k sÚt| j|_d | _| _X d S )Nr* ) Úgetattrr# r0 ÚAssertionErrorr) r* r5 r Z safe_reraiser2 r6 r1 r7 )r r= r> r? r8 Zout_of_band_exitr r r Ú__exit__ s0
|
ÿ
|
|
z TransactionalContext.__exit__N)r$ r% r& Ú__doc__Ú __slots__r' r0 r1 r2 r3 r5 r6 r7 Ú classmethodr; r< rB r r r r r( : s
|
|
r( )Ú
|
__future__r Útypingr r r r Ú r r Z util._has_cyr
|
Z util.typingr Ú TYPE_CHECKINGZ_py_utilr r Zsqlalchemy.cyextension.utilr r! r" r( r r r r Ú<module> s$
|
|
