�
g\�g2 � �2 � d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ ddlmZ dd lm Z dd
l
m Z dd l m
Z
dd l mZ ej se s
d
dlmZ d
dlmZ n d dlmZ d dlmZ edeg ef �� � Zdd�Z G d� de
� � Z G d� d� � ZdS ) � )� annotationsN)�Any)�Callable)�Optional)�TypeVar� )�exc)�util)�HAS_CYEXTENSION)�Protocol)�Self� )�_distill_params_20)�_distill_raw_params�_C)�bound�key�str�return�Callable[[_C], _C]c �6 � � t j � fd�� � }|S )z�Decorator, memoize a function in a connection.info stash.
Only applicable to functions which take no arguments other than a
connection. The memo will be stored in ``connection.info[key]``.
c � �� |� � � } |j � S # t $ r | ||� � x|j �< }|cY S w xY w�N)�connect�info�KeyError)�fn�self�
connection�valr s ��f/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/sqlalchemy/engine/util.py� decoratedz%connection_memoize.<locals>.decorated+ sg �� � �'�'�)�)�
� � �?�3�'� '��� � � �)+��D�*�)=�)=� =�J�O�C� �3��J�J�J� ���s � $ �"A �A )r
� decorator)r r" s ` r! �connection_memoizer$ $ s2 �� �
�^�� � � � �^�� �� c � � e Zd ZU ded< dS )�
_TConsSubjectz Optional[TransactionalContext]�_trans_context_managerN)�__name__�
__module__� __qualname__�__annotations__� r% r! r' r' 7 s � � � � � � �:�:�:�:�:�:r% r' c � � e Zd ZU dZdZded< dd�Zdd�Zdd �Zdd �Z d d
�Z
d d�Z d d�Z e
d d�� � Zd d�Zd d�ZdS ) �TransactionalContextz�Apply Python context manager behavior to transaction objects.
Performs validation to ensure the subject of the transaction is not
used if the transaction were ended prematurely.
)�_outer_trans_ctx�_trans_subject� __weakref__zOptional[_TConsSubject]r1 r �boolc � � t � � �r ��NotImplementedError�r s r! �_transaction_is_activez+TransactionalContext._transaction_is_activeG � � �!�#�#�#r% c � � t � � �r r5 r7 s r! �_transaction_is_closedz+TransactionalContext._transaction_is_closedJ r9 r% c � � t � � �)a indicates the object is in a state that is known to be acceptable
for rollback() to be called.
This does not necessarily mean rollback() will succeed or not raise
an error, just that there is currently no state detected that indicates
rollback() would fail or emit warnings.
It also does not mean that there's a transaction in progress, as
it is usually safe to call rollback() even if no transaction is
present.
.. versionadded:: 1.4.28
r5 r7 s r! �_rollback_can_be_calledz,TransactionalContext._rollback_can_be_calledM s � � "�#�#�#r% r' c � � t � � �r r5 r7 s r! � _get_subjectz!TransactionalContext._get_subject^ r9 r% �Nonec � � t � � �r r5 r7 s r! �commitzTransactionalContext.commita r9 r% c � � t � � �r r5 r7 s r! �rollbackz TransactionalContext.rollbackd r9 r% c � � t � � �r r5 r7 s r! �closezTransactionalContext.closeg r9 r% �subjectc �l � |j }|r(|� � � st j d� � �d S d S )Nz�Can't operate on closed transaction inside context manager. Please complete the context manager before emitting further commands.)r( r8 r �InvalidRequestError)�clsrG �
trans_contexts r! �_trans_ctx_checkz%TransactionalContext._trans_ctx_checkj sT � � �6�
� � � �7�7�9�9�
��-�8�� � � � �
�
r% r
c �f � | � � � }|j }|| _ || _ | |_ | S r )r? r( r0 r1 )r rG rK s r! � __enter__z TransactionalContext.__enter__u s: � ��#�#�%�%�� �6�
� -�� � %���)-��&�� r% �type_r �value� tracebackc �2 � t | dd � � }|d u p|j | u}|��| � � � r� | � � � nZ# t j � � 5 | � � � r| � � � d d d � � n # 1 swxY w Y Y nxY w|s|�J �| j |_ d x| _ | _ d S # |s|�J �| j |_ d x| _ | _ w xY w | � � � s)| �
� � s| � � � n(| � � � r| � � � |s|�J �| j |_ d x| _ | _ d S # |s|�J �| j |_ d x| _ | _ w xY w)Nr1 ) �getattrr( r8 rB r
� safe_reraiser= rD r0 r1 r; rF )r rO rP rQ rG �out_of_band_exits r! �__exit__z TransactionalContext.__exit__� s4 � ��$� 0�$�7�7��
�t�O� I�w� =�T� I� � �=�T� 8� 8� :� :�=�
C�� � �
�
�
�
��
(��&�(�(� (� (��3�3�5�5� (� �
�
����(� (� (� (� (� (� (� (� (� (� (���� (� (� (� (������ (� K�"�.�.�.�59�5J�G�2�>B�B��#�d�&;�&;�&;�� (� K�"�.�.�.�59�5J�G�2�>B�B��#�d�&;�B�B�B�B�
C��2�2�4�4� (� �6�6�8�8� %� �
�
� � � ���3�3�5�5� (� �
�
����'� K�"�.�.�.�59�5J�G�2�>B�B��#�d�&;�&;�&;�� (� K�"�.�.�.�59�5J�G�2�>B�B��#�d�&;�B�B�B�BsN �A �
C � B"� )B� B"�B �B"� B � B"� C �"C)�-A%E4 �4"FN)r r3 )r r' )r r@ )rG r' r r@ )r r
)rO r rP r rQ r r r@ )r) r* r+ �__doc__� __slots__r, r8 r; r= r? rB rD rF � classmethodrL rN rV r- r% r! r/ r/ ; s � � � � � � �� � F�I�+�+�+�+�$� $� $� $�$� $� $� $�$� $� $� $�"$� $� $� $�$� $� $� $�$� $� $� $�$� $� $� $� �� � � �[��
�
�
�
�&C� &C� &C� &C� &C� &Cr% r/ )r r r r )�
__future__r �typingr r r r � r r
� util._has_cyr � util.typingr r
�
TYPE_CHECKING�_py_utilr r �sqlalchemy.cyextension.utilr r$ r' r/ r- r% r! �<module>rb s� �� #� "� "� "� "� "�
�
�
�
� � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � *� *� *� *� *� *� "� "� "� "� "� "� � � � � � � �� �� �B�B�B�B�B�B�D�D�D�D�D�D�D�� � � � � �� � � � � �
�W�T��"�c�'�*�+�+�+��� � � �&;� ;� ;� ;� ;�H� ;� ;� ;�lC� lC� lC� lC� lC� lC� lC� lC� lC� lCr%