[h6dZddlmZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZededefZGddeZGddeZGddZdS)z9State tracking utilities used by :class:`_orm.Session`. ) annotationsN)Enum)Any)Callable)cast)Iterator)NoReturn)Optional)Tuple)TypeVar)Union)exc)util)Literal_F.)boundceZdZdS)_StateChangeStateN)__name__ __module__ __qualname__k/var/www/api.easyaligner.net/htdocs/venv_linux/lib/python3.11/site-packages/sqlalchemy/orm/state_changes.pyrr!sDrrceZdZdZdZdZdS)_StateChangeStatesrN)rrrANY NO_CHANGECHANGE_IN_PROGRESSrrrrr%s CIrrceZdZUdZejZded<ejZ ded<dZ ded<dd Z e ddZ ejddZdS) _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_stateNzOptional[Callable[..., Any]] _current_fnoperation_namestrstatereturnr c:tjd|d|d)NzCan't run operation 'z()' when Session is in state iscecode)sa_excIllegalStateChangeError)selfr(r*s r_raise_for_prerequisite_statez*_StateChange._raise_for_prerequisite_state9s?, %N % %  % %    rprerequisite_statesEUnion[Literal[_StateChangeStates.ANY], Tuple[_StateChangeState, ...]]moves_toCallable[[_F], _F]c|s Jd|tjutd|tjutjd fd }|S) aMethod 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 sentzTuple[_StateChangeState, ...]fnrr2rargkwr+c&|j} r| vr||j||j}|j} r n|}|t juro rm||uri|rC|t jt jfvr)tj d|jd|jd dtj d|jd d|d||_t j|_ ||g|Ri|}|j|ur|||_||_S|j|ur"tj d|jd d d|r.tj d |jd |jd |jdtj d|jd |jd#xYw#||_||_wxYw)NzMethod 'z"()' can't be called here; method 'zN()' is already in progress and this would cause an unexpected state change to r-r.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&r3rr%r'rr r!r"r0r1) r9r2r:r; current_state next_state existing_fn expect_state ret_valueexpect_state_changehas_prerequisite_statesr6prerequisite_state_collections r_goz(_StateChange.declare_states.._go_s KM( O!)FFF222; NNN)J*K':M88 L"4"888'9l22:&0&92$$!882;88#.#788,488$ !84r{44.644%/44$  "D 1DD ! /Bt0c000R00 ;,..$4$. #.  3;-// 872;77&777$  ! 8;)=;;#%;;;,0; ;; $ !8;2;;;+/;;;#/ :$. #. ....s% E: F9BF:E==FF) r9rr2rr:rr;rr+r)rr rr!r decorator)clsr4r6rErBrCrDs ` @@@rdeclare_statesz_StateChange.declare_statesBs&#??$???? '9'= = )- +-@) ) %'.@.JJ K /K /K /K /K /K /K /K / K /Z rexpected Iterator[Any]c#K|jtjus Jd||_ dV|j|urt jd|jdn#xYw tj|_dS#tj|_wxYw)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%rr"r&r0r1)r2rIs r _expect_statez_StateChange._expect_states#5#HHHH $IHH $ E EEE{(**4A$+AA+  * 2DD   1DD  D D D DsA(A.AA..B)r(r)r*rr+r )r4r5r6rr+r7)rIrr+rJ)rrr__doc__rr r%__annotations__r!r&r'r3 classmethodrH contextlibcontextmanagerrLrrrr$r$+s&8%;K;;;; 2 rWs#"""""!!!!!! WT#s(+,,,        * [E[E[E[E[E[E[E[E[E[Er