관리-도구
편집 파일: state_changes.cpython-38.pyc
U -?�f� � @ 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 �L/opt/hc_python/lib64/python3.8/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 ddd�dd��Z ejddd�dd��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�strr )�operation_name�state�returnc C s t jd|� d|��dd��d S )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_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 s~ |j }�r |�kr |�| j|� |j}|j}� r4�n|}|tjk r�� r�||k r�|r�|tjtjfkr�t j d| j� d|j� d���dd��n"t j d| j� d��d|��dd��| |_tj|_z�z| |f|�|�}W n � Y n�X |j |kr�|W �~S |j |k�r t j d| j� d ��d �dd��nL|�rJt j d |j� d | j� d |j ��dd��n t j d| j� d |j ��dd��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 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+ ) r0 r, r1 r2 Z current_stateZ next_stateZ existing_fnZ expect_stateZ ret_value�Zexpect_state_changeZhas_prerequisite_statesr/ Z prerequisite_state_collectionr r �_go_ sf �� ������ �� � z(_StateChange.declare_states.<locals>._go)�AssertionErrorr r r r r � decorator)�clsr. r/ r4 r r3 r �declare_statesB s � � "Mz_StateChange.declare_statesz Iterator[Any])�expectedr& c c sh | j tjkstd��|| _ z>z dV W n � Y n"X | j|k rVtjd| j��dd��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 r r r5 r! r* r+ )r, r9 r r r � _expect_state� s � � z_StateChange._expect_state)r r r �__doc__r r r �__annotations__r r! r" r- � classmethodr8 � contextlib�contextmanagerr: r r r r r + s lr )r; � __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&