Edit File: threadlocal.cpython-37.pyc
B ��4]� � @ sL d Z ddlZddlmZ ddlmZ G dd� dej�ZG d d � d ej�ZdS ) a8 Provides a thread-local transactional wrapper around the root Engine class. The ``threadlocal`` module is invoked when using the ``strategy="threadlocal"`` flag with :func:`~sqlalchemy.engine.create_engine`. This module is semi-private and is invoked automatically when the threadlocal engine strategy is used. � N� )�base� )�utilc s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) � TLConnectionc s t t| �j||� d| _d S )Nr )�superr �__init__�_TLConnection__opencount)�self�arg�kw)� __class__� �P/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/engine/threadlocal.pyr s zTLConnection.__init__c C s | j d7 _ | S )Nr )r )r r r r �_increment_connect s z TLConnection._increment_connectc C s( | j dkrtj�| � | j d8 _ d S )Nr )r r � Connection�close)r r r r r s zTLConnection.closec C s d| _ tj�| � d S )Nr )r r r r )r r r r � _force_close$ s zTLConnection._force_close)�__name__� __module__� __qualname__r r r r � __classcell__r r )r r r s r c s� e Zd ZdZeZe�dd�� fdd��Zdd� Z dd � Z d#d d �Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Z� fdd �Zed d � �Zd d � Zd!d"� Z� ZS )$�TLEnginezPAn Engine that includes support for thread-local managed transactions. z1.3a The 'threadlocal' engine strategy is deprecated, and will be removed in a future release. The strategy is no longer relevant to modern usage patterns (including that of the ORM :class:`.Session` object) which make use of a :class:`.Connection` object in order to invoke statements.c s" t t| �j||� tj�� | _d S )N)r r r r � threading�local� _connections)r �args�kwargs)r r r r 1 s zTLEngine.__init__c K s | j f |�S )N)�_contextual_connect)r r r r r �contextual_connect= s zTLEngine.contextual_connectc K s\ t | jd�sd }n | j�� }|d ks*|jrT| j| | �| jj|�f|�}t� |�| j_|� � S )N�conn) �hasattrr r �closed�_tl_connection_clsZ_wrap_pool_connect�pool�connect�weakref�refr )r r � connectionr r r r @ s z TLEngine._contextual_connectNc C s2 t | jd�sg | j_| jj�| �� j|d�� | S )N�trans)�xid)r! r r) �appendr �begin_twophase)r r* r r r r, R s zTLEngine.begin_twophasec C s. t | jd�sg | j_| jj�| �� �� � | S )Nr) )r! r r) r+ r � begin_nested)r r r r r- Z s zTLEngine.begin_nestedc C s. t | jd�sg | j_| jj�| �� �� � | S )Nr) )r! r r) r+ r �begin)r r r r r. b s zTLEngine.beginc C s | S )Nr )r r r r � __enter__h s zTLEngine.__enter__c C s |d kr| � � n| �� d S )N)�commit�rollback)r �type_�value� tracebackr r r �__exit__k s zTLEngine.__exit__c C s, t | jd�r| jjsd S | jjd �� d S )Nr) ���)r! r r) �prepare)r r r r r7 q s zTLEngine.preparec C s2 t | jd�r| jjsd S | jj�d�}|�� d S )Nr) r6 )r! r r) �popr0 )r r) r r r r0 y s zTLEngine.commitc C s2 t | jd�r| jjsd S | jj�d�}|�� d S )Nr) r6 )r! r r) r8 r1 )r r) r r r r1 � s zTLEngine.rollbackc s t j�� | _tt| ��� d S )N)r r r r r r �dispose)r )r r r r9 � s zTLEngine.disposec C s( t | jd� p&| j�� d kp&| j�� jS )Nr )r! r r r" )r r r r r" � s zTLEngine.closedc C s6 | j s2| �� �� | j�� }|�� | j`g | j_d S )N)r" r r r r r r) )r r( r r r r � s zTLEngine.closec C s d| j S )Nz TLEngine(%r))�url)r r r r �__repr__� s zTLEngine.__repr__)N)r r r �__doc__r r# r Z deprecatedr r r r, r- r. r/ r5 r7 r0 r1 r9 �propertyr" r r; r r r )r r r ) s& r ) r< r& � r r r r ZEnginer r r r r �<module> s