o g@sdZddlmZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZededefdZGdddeZGdddeZGdddZdS)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 eZdZdS)_StateChangeStateN)__name__ __module__ __qualname__rrX/var/www/html/flaskapp/venv/lib/python3.10/site-packages/sqlalchemy/orm/state_changes.pyr!src@seZdZdZdZdZdS)_StateChangeStatesr N)rrrANY NO_CHANGECHANGE_IN_PROGRESSrrrrr%src@sbeZdZUdZejZded<ejZ ded<dZ ded<dd dZ e dddZ ejdddZdS) _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_namestrstatereturnrcCstjd|d|dd)NzCan't run operation 'z()' when Session is in state iscecode)sa_excIllegalStateChangeError)selfr#r%rrr_raise_for_prerequisite_state9s z*_StateChange._raise_for_prerequisite_stateprerequisite_statesEUnion[Literal[_StateChangeStates.ANY], Tuple[_StateChangeState, ...]]moves_toCallable[[_F], _F]csH|sJd|tjutd|tjutjd fd d }|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, ...]fnrr,rargkwr&c sv|j}r|vr||j||j}|j}rn|}|tjurTrT||urT|rC|tjtjfvrCt j d|jd|jdddt j d|jdd|dd||_tj|_zXz ||g|Ri|}Wn|j|ur||W||_||_S|j|urt j d|jd d dd|rt j d |jd |jd |jddt j d|jd |jdd||_||_w)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!r-rr r"rrrrr*r+) r2r,r3r4 current_state next_state existing_fn expect_state ret_valueexpect_state_changehas_prerequisite_statesr0prerequisite_state_collectionrr_go_s         z(_StateChange.declare_states.._goN) r2rr,rr3rr4rr&r)rrrrr decorator)clsr.r0r>rr:rdeclare_statesBs  Mz_StateChange.declare_statesexpected Iterator[Any]ccsj|jtjus Jd||_z!zdVWn|j|ur)tjd|jddWtj|_dStj|_w)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 rrr!r*r+)r,rBrrr _expect_states   z_StateChange._expect_state)r#r$r%rr&r)r.r/r0rr&r1)rBrr&rC)rrr__doc__rrr __annotations__rr!r"r- classmethodrA contextlibcontextmanagerrDrrrrr+s     lr)rE __future__rrHenumrtypingrrrrrr r r r rr*r util.typingrrrrrrrrrs(