OwlCyberSecurity - MANAGER
Edit File: state_changes.cpython-312.pyc
� E��g� � � � 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 �� Z G d� de� Z G d� de� Z G d� d� Zy)z9State tracking utilities used by :class:`_orm.Session`. � )� annotationsN)�Enum)�Any)�Callable)�cast)�Iterator)�NoReturn)�Optional)�Tuple)�TypeVar)�Union� )�exc)�util)�Literal�_F.)�boundc � � e Zd Zy)�_StateChangeStateN)�__name__� __module__� __qualname__� � �M/opt/hc_python/lib64/python3.12/site-packages/sqlalchemy/orm/state_changes.pyr r ! s � �r r c � � e Zd ZdZdZdZy)�_StateChangeStates� r � N)r r r �ANY� NO_CHANGE�CHANGE_IN_PROGRESSr r r r r % s � � �C��I��r r c � � e Zd ZU dZej Zded< ej Z ded< dZ ded< d d�Z e d d �� Z ej d d �� Zy)� _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_fnc �< � t j d|� d|��d�� �)NzCan't run operation 'z ()' when Session is in state �isce��code)�sa_exc�IllegalStateChangeError)�self�operation_name�states r � _raise_for_prerequisite_statez*_StateChange._raise_for_prerequisite_state9 s1 � � �,�,�#�N�#3� 4� �)� %�� � r c � ����� |sJ d� �|t j u�t d|� ��t j u�t j d����fd� � }|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, ...]c �� �� |j }� r |� vr |j | j |� |j }|j }� r� n|}|t j ur�� r�||ur�|rV|t j t j fv r4t j d| j � d|j � d� ��d�� �t j d| j � d� �d|��d�� �| |_ t j |_ | |g|��i |��}|j |u r|||_ ||_ S |j |u r(t j d| j � d � �d �d�� �|r>t j d |j � d | j � d |j ��d�� �t j d| j � d |j ��d�� �# � xY w# ||_ ||_ w xY 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& r1 r r% r' r r r! r" r, r- ) �fnr. �arg�kw� current_state� next_state� existing_fn� expect_state� ret_value�expect_state_change�has_prerequisite_states�moves_to� prerequisite_state_collections ����r �_goz(_StateChange.declare_states.<locals>._go_ sB �� � �K�K�M� (�!�)F�F��2�2�2�;�;� �N� �)�)�J� �*�*�K�':�8� �L� �"4�"8�"8�8�'� �l�2� �:�&�0�0�&�9�9�2� $� !�8�8�"�2�;�;�-� 0#�#.�#7�#7�"8� 9+�+3�,�8� $� � � !�8�8�.�r�{�{�m� <.�.6�\� :%�%/�N�4� $� � � "�D� � 1� D� D�D� �! /� �t� 0�c� 0�R� 0� � �;�;�,�.�$�4 $.�� �#.�� �3 �;�;�-�/� �8�8�"�2�;�;�-� 0 � &�\� �7� $� � � !� �8�8�(��)=�)=�(>� ?#�#%�;�;�-� 0+�+/�;�;�/� ;� $� � � !�8�8�"�2�;�;�-� 0+�+/�;�;�/�;� #�� ��/ ���: $.�� �#.�� �s% � G �G �/B'G �G�G � G,) r4 r r. r r5 r r6 r �returnr )r r r r! r � decorator)�cls�prerequisite_statesr> r@ r< r= r? s ` @@@r �declare_statesz_StateChange.declare_statesB so �� �& #�?�$?�?�"� �'9�'=�'=� =� � )-� +�-@�) �%� '�.@�.J�.J� J�� � � �K /� �K /�Z � r c # �: K � | j t j u sJ d� �|| _ d�� | j |ur$t j d| j ��d�� � t j | _ y# � xY w# t j | _ w xY w�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. �expecteds r � _expect_statez_StateChange._expect_state� s� � �� � � � �#5�#H�#H�H� � $� �H� $��� E� � �{�{�(�*� �4�4�1�$�+�+� �A��� � +� 2� D� D�D� �� � �� 2� D� D�D� �s- �+B�A; �2B �%B�;A>�>B �B�B)r/ �strr0 r rA r )rD zEUnion[Literal[_StateChangeStates.ANY], Tuple[_StateChangeState, ...]]r> r rA zCallable[[_F], _F])rG r rA z Iterator[Any])r r r �__doc__r r r% �__annotations__r! r& r' r1 � classmethodrE � contextlib�contextmanagerrH r r r r$ r$ + s� � �� &8�%;�%;�K�"�;� 2� <� <�F� �<�04�K�-�4� � !� �*;� � � � �j� �j� $� j� � j� �j�X � � �E� �Er r$ )rJ � __future__r rM �enumr �typingr r r r r r r r r � r r, r � util.typingr r r r r$ r r r �<module>rT su ��� #� � � � � � � � � � � � � � !� �T��#�s�(�+�,�� �� ��*� � [E� [Er