Edit File: threadlocal.cpython-311.pyc
� u�d� � �v � d Z ddlZddlmZ ddlmZ G d� dej � � Z G 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 �0 � � e Zd Z� fd�Zd� Zd� Zd� Z� xZS )� TLConnectionc �V �� t t | � � j |i |�� d| _ d S �Nr )�superr �__init__�_TLConnection__opencount)�self�arg�kw� __class__s ��T/opt/cloudlinux/venv/lib64/python3.11/site-packages/sqlalchemy/engine/threadlocal.pyr zTLConnection.__init__ s3 �� �*��l�D�!�!�*�C�6�2�6�6�6� ����� c �& � | xj dz c_ | S �Nr )r �r s r �_increment_connectz TLConnection._increment_connect s � � ���A� ���� r c �z � | j dk r t j � | � � | xj dz c_ d S r �r r � Connection�closer s r r zTLConnection.close s@ � � � �q� � � �O� !� !�$� '� '� '� ���A� ����r c �R � d| _ t j � | � � d S r r r s r � _force_closezTLConnection._force_close$ s& � � ��� �� � �d�#�#�#�#�#r )�__name__� __module__� __qualname__r r r r � __classcell__�r s @r r r se �� � � � � � � � � � �� � � � � � $� $� $� $� $� $� $r r c � � � e Zd ZdZeZ ej dd� � � fd�� � Zd� Z d� Z dd�Z d � Z d � Z d � Zd � Zd � Zd� Zd� Z� fd�Zed� � � Zd� Zd� Z� xZS )�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:`_engine.Connection` object in order to invoke statements.c � �� t t | � � j |i |�� t j � � � | _ d S �N)r r$ r r � threading�local� _connections)r �args�kwargsr s �r r zTLEngine.__init__1 sC �� � '��h�� � �&��7��7�7�7� �N� 0� 0� 2� 2����r c � � | j di |��S )N� )�_contextual_connect)r r s r �contextual_connectzTLEngine.contextual_connect> s � �'�t�'�-�-�"�-�-�-r c �4 � t | j d� � sd }n| j � � � }|�|j rK | j | | � | j j |� � fi |��}t j |� � | j _ |� � � S �N�conn) �hasattrr) r2 �closed�_tl_connection_cls�_wrap_pool_connect�pool�connect�weakref�refr )r r � connections r r. z TLEngine._contextual_connectA s� � ��t�(�&�1�1� 2� �J�J� �*�/�/�1�1�J� � ��!2� � 1��0���'�'�� �(9�:�F�F�� � �� �J� &-�[��%<�%<�D� � "��,�,�.�.�.r Nc �� � t | j d� � s g | j _ | j j � | � � � � |�� � � � | S )N�trans)�xid)r3 r) r= �appendr. �begin_twophase)r r> s r r@ zTLEngine.begin_twophaseS sg � ��t�(�'�2�2� )�&(�D� � #� �� �&�&� � $� $� &� &� 5� 5�#� 5� >� >� � � � � r c �� � t | j d� � s g | j _ | j j � | � � � � � � � � | S �Nr= )r3 r) r= r? r. � begin_nestedr s r rC zTLEngine.begin_nested[ sb � ��t�(�'�2�2� )�&(�D� � #� �� �&�&� � $� $� &� &� 3� 3� 5� 5� � � � � r c �� � t | j d� � s g | j _ | j j � | � � � � � � � � | S rB )r3 r) r= r? r. �beginr s r rE zTLEngine.beginc sY � ��t�(�'�2�2� )�&(�D� � #� �� �&�&�t�'?�'?�'A�'A�'G�'G�'I�'I�J�J�J�� r c � � | S r& r- r s r � __enter__zTLEngine.__enter__i s � �� r c �^ � |�| � � � d S | � � � d S r& )�commit�rollback)r �type_�value� tracebacks r �__exit__zTLEngine.__exit__l s) � � �=� �K�K�M�M�M�M�M� �M�M�O�O�O�O�Or c � � t | j d� � r | j j sd S | j j d � � � d S �Nr= ���)r3 r) r= �preparer s r rR zTLEngine.preparer sO � ���)�7�3�3� ��$�*� � �F� �� ��#�+�+�-�-�-�-�-r c � � t | j d� � r | j j sd S | j j � d� � }|� � � d S rP )r3 r) r= �poprI �r r= s r rI zTLEngine.commitz sV � ���)�7�3�3� ��$�*� � �F��!�'�+�+�B�/�/�� � � �����r c � � t | j d� � r | j j sd S | j j � d� � }|� � � d S rP )r3 r) r= rT rJ rU s r rJ zTLEngine.rollback� s[ � ���)�7�3�3� ��$�*� � �F��!�'�+�+�B�/�/�� �������r c � �� t j � � � | _ t t | � � � � � d S r& )r r' r( r) r r$ �dispose)r r s �r rX zTLEngine.dispose� s; �� � �N� 0� 0� 2� 2��� �h�� � �%�%�'�'�'�'�'r c � � t | j d� � p8| j � � � d u p | j � � � j S r1 )r3 r) r2 r4 r s r r4 zTLEngine.closed� sR � � ��)�6�2�2� 2� /�� �%�%�'�'�4�/� /�� �%�%�'�'�.� r c �� � | j sh| � � � � � � | j � � � }|� � � | j `g | j _ d S d S r& )r4 r. r r) r2 r r= )r r; s r r zTLEngine.close� st � ��{� )� � $� $� &� &� ,� ,� .� .� .� �*�/�/�1�1�J� � #� #� %� %� %��!�&�&(�D� � #� #� #� )� )r c � � d| j z S )Nz TLEngine(%r))�urlr s r �__repr__zTLEngine.__repr__� s � � ���(�(r r& )r r r �__doc__r r5 r � deprecatedr r/ r. r@ rC rE rG rN rR rI rJ rX �propertyr4 r r] r! r" s @r r$ r$ ) sZ �� � � � � �� � &���T�_� � 0�� �3� 3� 3� 3�� �3�.� .� .�/� /� /�$� � � �� � �� � � � � � � � � .� .� .�� � �� � �(� (� (� (� (� � � � �X� �)� )� )�)� )� )� )� )� )� )r r$ ) r^ r9 � r r r r �Enginer$ r- r r �<module>rc s� ��� � ���� � � � � � � � � � � � �$� $� $� $� $�4�?� $� $� $�&x)� x)� x)� x)� x)�t�{� x)� x)� x)� x)� x)r