File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/__pycache__/state_changes.cpython-310.pyc
o
���g� � @ 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 d�ZG dd� de�ZG dd� de�ZG dd� d�ZdS )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 �G/usr/local/lib/python3.10/dist-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 @ sb e Zd ZU dZejZded< ejZ ded< dZ
ded< dd
d�Z e ddd��Z
ejddd��ZdS ) � _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�operation_name�str�state�returnr c C s t jd|� d|��dd��)NzCan't run operation 'z ()' when Session is in state �isce��code)�sa_exc�IllegalStateChangeError)�selfr# r% r r r � _raise_for_prerequisite_state9 s ��z*_StateChange._raise_for_prerequisite_state�prerequisite_states�EUnion[Literal[_StateChangeStates.ANY], Tuple[_StateChangeState, ...]]�moves_to�Callable[[_F], _F]c sH |sJ d��|t ju�td|���t ju� tjd
� ���fd
d � �}|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, ...]�fnr r, r �arg�kwr& c sv |j }�r|�vr|�| j|� |j}|j}� r�n|}|tjurT� rT||urT|rC|tjtjfv rCt j
d| j� d|j� d���dd��t j
d| j� d��d|��dd��| |_tj|_zXz
| |g|�R i |��}W n � |j |u r||W ||_||_S |j |u r�t j
d| j� d ��d
�dd��|r�t j
d |j� d | j� d
|j ��dd��t j
d| j� d
|j ��dd��||_||_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- r r r" r r r r r* r+ ) r2 r, r3 r4 �
current_state�
next_state� existing_fn� expect_state� ret_value��expect_state_change�has_prerequisite_statesr0 � prerequisite_state_collectionr r �_go_ s� �
��
���
���
�
��
���
��z(_StateChange.declare_states.<locals>._goN)
r2 r r, r r3 r r4 r r& r )r r r r r � decorator)�clsr. r0 r> r r: r �declare_statesB s ��
Mz_StateChange.declare_states�expected�
Iterator[Any]c c sj � | j tju s J d��|| _ z!zdV W n � | j|ur)tjd| j��dd��W tj| _ dS tj| _ 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 r r r! r* r+ )r, rB r r r �
_expect_state� s ��
��z_StateChange._expect_state)r# r$ r% r r& r )r. r/ r0 r r& r1 )rB r r& rC )r r r �__doc__r r r �__annotations__r r! r" r- � classmethodrA �
contextlib�contextmanagerrD r r r r r + s
lr )rE �
__future__r rH �enumr �typingr r r r r r r
r r � r r* r � util.typingr r r r r r r r r �<module> s(