U
|
¸ý°d| ã @ sú d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
|
Z
|
dd lm Z dd
|
lm Z dd lm Z dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ededef dZG dd deZG dd deZG dd dZdS )z9State tracking utilities used by :class:`_orm.Session`.
|
|
é )Ú annotationsN)ÚEnum)ÚAny)ÚCallable)Úcast)ÚIterator)ÚNoReturn)ÚOptional)ÚTuple)ÚTypeVar)ÚUnioné )Úexc)Úutil)ÚLiteralÚ_F.)Úboundc @ s e Zd ZdS )Ú_StateChangeStateN)Ú__name__Ú
|
__module__Ú __qualname__© r r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/orm/state_changes.pyr ! s r c @ s e Zd ZdZdZdZdS )Ú_StateChangeStatesé r é N)r r r ÚANYÚ NO_CHANGEÚCHANGE_IN_PROGRESSr r r r r % s r c @ sx e Zd ZU dZejZded< ejZ ded< dZ
|
ded< ddd d
|
d d Z e d dddddZ ejdddddZdS )Ú _StateChangea! Supplies state assertion decorators.
|
|
The current use case is for the :class:`_orm.SessionTransaction` class. The
|
:class:`_StateChange` class itself is agnostic of the
|
:class:`_orm.SessionTransaction` class so could in theory be generalized
|
for other systems as well.
|
|
r Ú _next_stateÚ_stateNz Optional[Callable[..., Any]]Ú _current_fnÚstrr )Úoperation_nameÚstateÚreturnc C s t d| d|¡d S )NzCan't run operation 'z ()' when Session is in state )Úsa_excÚIllegalStateChangeError)Úselfr$ r% r r r Ú _raise_for_prerequisite_state9 s ÿz*_StateChange._raise_for_prerequisite_statezEUnion[Literal[_StateChangeStates.ANY], Tuple[_StateChangeState, ...]]zCallable[[_F], _F])Úprerequisite_statesÚmoves_tor& c sT |s t d|tjk td|tjk tjdddddd fdd }|S )a¼ Method decorator declaring valid states.
|
|
:param prerequisite_states: sequence of acceptable prerequisite
|
states. Can be the single constant _State.ANY to indicate no
|
prerequisite state
|
|
:param moves_to: the expected state at the end of the method, assuming
|
no exceptions raised. Can be the constant _State.NO_CHANGE to
|
indicate state should not change at the end of the method.
|
|
zno prequisite states sentz Tuple[_StateChangeState, ...]r r )Úfnr) ÚargÚkwr& c sj |j }r |kr | | j|¡ |j}|j} r4n|}|tjk r¢ r¢||k r¢|r|tjtjfkrt
|
d| j d|j d¡n t
|
d| j dd|¡| |_tj|_z¦z| |f||}W n Y nX |j |krè|W ¢rS |j |krt
|
d| j dd¡nD|r:t
|
d |j d
|
| j d |j ¡nt
|
d| j d |j ¡W 5 ||_||_X d S ) NzMethod 'z"()' can't be called here; method 'zN()' is already in progress and this would cause an unexpected state change to zCant run operation 'z ()' here; will move to state z where we are expecting z ()' failed to change state to z as expectedzWhile method 'z()' was running, method 'z)()' caused an unexpected state change to ) r! r* r r r" r r r r r' r( ) r- r) r. r/ Z current_stateÚ
|
next_stateZ existing_fnZ expect_stateZ ret_value©Zexpect_state_changeZhas_prerequisite_statesr, Z prerequisite_state_collectionr r Ú_go^ s\ ÿþ þüúþÿÿ
|
ÿÿÿz(_StateChange.declare_states.<locals>._go)ÚAssertionErrorr r r r r Ú decorator)Úclsr+ r, r2 r r1 r Údeclare_statesA s ÿ ÿ
|
"Iz_StateChange.declare_statesz Iterator[Any])Úexpectedr& c c sd | j tjkstd|| _ z:z
|
dV W n Y n X | j|k rRt d| j¡W 5 tj| _ X dS )zxcalled within a method that changes states.
|
|
method must also use the ``@declare_states()`` decorator.
|
|
zAUnexpected call to _expect_state outside of state-changing methodNzUnexpected state change to )r r r r3 r! r' r( )r) r7 r r r Ú _expect_stateª s ÿ
|
|
|
ÿz_StateChange._expect_state)r r r Ú__doc__r r r Ú__annotations__r r! r" r* Ú classmethodr6 Ú
|
contextlibÚcontextmanagerr8 r r r r r + s
|
hr )r9 Ú
|
__future__r r< Úenumr Útypingr r r r r r r
|
r r Ú r r' r Z util.typingr r r r r r r r r Ú<module> s&
|
