관리-도구
편집 파일: util.cpython-38.pyc
U -?�f2 � @ s 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 ejs�e s�d dlmZ d dlmZ nd dlmZ d dlmZ edeg ef d�Zddd�dd�ZG dd� de �ZG dd� d�ZdS )� )� annotationsN)�Any)�Callable)�Optional)�TypeVar� )�exc)�util)�HAS_CYEXTENSION)�Protocol)�Self� )�_distill_params_20)�_distill_raw_params�_C)�bound�strzCallable[[_C], _C])�key�returnc s t j� fdd��}|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 sF |� � }z |j� W S tk r@ | ||� |j� < }| Y S X d S �N)�connect�info�KeyError)�fn�self� connection�val�r � �F/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/engine/util.py� decorated+ s z%connection_memoize.<locals>.decorated)r � decorator)r r r r r �connection_memoize$ s r" c @ s e Zd ZU ded< dS )� _TConsSubjectz Optional[TransactionalContext]�_trans_context_managerN)�__name__� __module__� __qualname__�__annotations__r r r r r# 7 s r# c @ s� e Zd ZU dZdZded< dd�dd�Zdd�d d �Zdd�d d �Zd d�dd�Z dd�dd�Z dd�dd�Z dd�dd�Z e d dd�dd��Zdd�dd �Zd d d dd �d d �Zd!S )"�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]r+ �bool)r c C s t � �d S r ��NotImplementedError�r r r r �_transaction_is_activeG s z+TransactionalContext._transaction_is_activec C s t � �d S r r. r0 r r r �_transaction_is_closedJ s z+TransactionalContext._transaction_is_closedc C s t � �dS )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 Nr. r0 r r r �_rollback_can_be_calledM s z,TransactionalContext._rollback_can_be_calledr# c C s t � �d S r r. r0 r r r � _get_subject^ s z!TransactionalContext._get_subject�Nonec C s t � �d S r r. r0 r r r �commita s zTransactionalContext.commitc C s t � �d S r r. r0 r r r �rollbackd s z TransactionalContext.rollbackc C s t � �d S r r. r0 r r r �closeg s zTransactionalContext.close)�subjectr c C s |j }|r |�� s t�d��d S )Nz�Can't operate on closed transaction inside context manager. Please complete the context manager before emitting further commands.)r$ r1 r ZInvalidRequestError)�clsr9 � trans_contextr r r �_trans_ctx_checkj s �z%TransactionalContext._trans_ctx_checkr c C s$ | � � }|j}|| _|| _| |_| S r )r4 r$ r* r+ )r r9 r; r r r � __enter__u s z TransactionalContext.__enter__r )�type_�value� tracebackr c C s� t | dd �}|d kp |j| k }|d kr�| �� r�zBz | �� W n0 t�� � | � � r^| � � W 5 Q R X Y nX W 5 |s�|d k s�t�| j|_d | _| _X nVz.| �� s�| � � s�| � � n| � � r�| � � W 5 |s�|d k s�t�| j|_d | _| _X d S )Nr+ ) �getattrr$ r1 �AssertionErrorr* r+ r6 r Z safe_reraiser3 r7 r2 r8 )r r> r? r@ r9 Zout_of_band_exitr r r �__exit__� s0 � z TransactionalContext.__exit__N)r% r& r' �__doc__� __slots__r( r1 r2 r3 r4 r6 r7 r8 � classmethodr< r= rC r r r r r) ; s r) )� __future__r �typingr r r r � r r Z util._has_cyr Z util.typingr r � TYPE_CHECKINGZ_py_utilr r Zsqlalchemy.cyextension.utilr r" r# r) r r r r �<module> s&