o

���g��@
s�UdZd
dl
mZ
d
dlmZ
d
dlZd
dlmZ
d
dlZd
dl	Z	d
dl
Z
d
dl
mZ
d
dl
mZ
d
dl
m
Z

d
d	l
mZ
d
d
l
mZ
d
dl
mZ
d
dl
mZ
d
d
l
mZ
d
dl
mZ
d
dl
mZ
d
dlZddlmZ
ddlmZ
ddlmZ
ddlmZ
ddlmZ
ddlmZ
er�ddlmZ
ddlm Z 
ddlm!Z!
ddlm"Z"
ddlm#Z#
ddlm$Z$
ddl%m&Z&
ej'dd�
Gd d!�d!��
Z(Gd"d#�d#e�Z)ee)ed$fZ*e+e)�
\Z,Z-Z.Gd%d&�d&�Z/Gd'd(�d(e/�Z0Gd)d*�d*e�Z1Gd+d,�d,e�Z2Gd-d.�d.ej3ej4�Z5Gd/d0�d0�Z6Gd1d2�d2e6�Z7Gd3d4�d4e7�Z8	5	dOdPdEdF�Z9iZ:dGe;dH<GdIdJ�dJe6�Z<GdKdL�dLe<�Z=GdMdN�dNe<�Z>dS)Qz'Base constructs for connection pools.

�)
�annotations)
�dequeN)
�Enum)
�Any)
�Callable)
�cast)
�Deque)
�Dict)
�List)
�Optional)
�Tuple)
�
TYPE_CHECKING)
�Union�)
�event)
�exc)
�log)
�util)
�Literal)
�Protocol)
�DBAPIConnection)
�DBAPICursor)
�Dialect)
�_DispatchCommon)
�_ListenerFnType)
�
dispatcher)
�	_InfoTypeT)
�frozenc@
s2eZ
dZUd
ZdZded<	ded<	ded<dS)�PoolResetStatez�describes the state of a DBAPI connection as it is being passed to
    the :meth:`.PoolEvents.reset` connection pool event.

    .. versionadded:: 2.0.0b3

    ��transaction_was_reset�terminate_only�asyncio_safe�boolr r!r"N)�__name__�
__module__�__qualname__�__doc__�	__slots__�__annotations__�r*r*�G/home/arjun/.local/lib/python3.10/site-packages/sqlalchemy/pool/base.pyr2s



rc
@
seZ
dZd
ZdZdZdZdS)�
ResetStylez1Describe options for "reset on return" behaviors.r
�
rN)r$r%r&r'�reset_rollback�reset_commit�
reset_noner*r*r*r+r,_s



r,)TNF�commit�rollbackc@
sTeZ
dZd
ZdZdZddd�Zdd	d
�Zddd
�Zddd�Z	ddd�Z
ddd�ZdS)�_ConnDialectz�partial implementation of :class:`.Dialect`
    which provides DBAPI connection methods.

    When a :class:`_pool.Pool` is combined with an :class:`_engine.Engine`,
    the :class:`_engine.Engine` replaces this with its own
    :class:`.Dialect`.

    F�dbapi_connection�PoolProxiedConnection�return�NonecC
�|
��
dS�
N)
r2��selfr4r*r*r+�do_rollback{�
z_ConnDialect.do_rollbackcC
r8r9)
r1r:r*r*r+�	do_commit~r=z_ConnDialect.do_commitrcC
r8r9�
�closer:r*r*r+�do_terminate�r=z_ConnDialect.do_terminatecC
r8r9r?r:r*r*r+�do_close�r=z_ConnDialect.do_closer#cC
std
�
�
)NzJThe ping feature requires that a dialect is passed to the connection pool.�
�NotImplementedErrorr:r*r*r+�_do_ping_w_event�s

�z_ConnDialect._do_ping_w_event�
connectionrc
C
s|
Sr9r*)r;rFr*r*r+�get_driver_connection�s
z"_ConnDialect.get_driver_connectionN)r4r5r6r7)r4rr6r7)r4rr6r#)rFrr6r)r$r%r&r'�is_async�
has_terminater<r>rArBrErGr*r*r*r+r3ns
	





r3c
@
seZ
dZd
ZdS)�_AsyncConnDialectTN)r$r%r&rHr*r*r*r+rJ�s
rJc@
seZ
dZddd�ZdS)�_CreatorFnTyper6rc


C
�dSr9r*�
r;r*r*r+�__call__��z_CreatorFnType.__call__N�r6r�r$r%r&rNr*r*r*r+rK��
rKc@
seZ
dZddd�ZdS)	�_CreatorWRecFnType�rec�ConnectionPoolEntryr6rc
C
rLr9r*�r;rTr*r*r+rN�rOz_CreatorWRecFnType.__call__N)rTrUr6rrQr*r*r*r+rS�rRrSc@
s$
eZ
dZUd
Zded<ded<ded<e�Zded	<d
ed<ded
<ded<								dMdNd"d#�Zej	dOd%d&��
Z
edPd'd(��
Zej
dQd*d(��
ZejdRd+d(��
ZdSd,d-�Zdd.�
dTd2d3�ZdUd5d6�Z		dVdWd;d<�ZdXd=d>�ZdRd?d@�ZdYdAdB�ZdZdDdE�ZdUdFdG�ZdZdHdI�Zd[dKdL�ZdS)\�Poolz)Abstract base class for connection pools.zdispatcher[Pool]�dispatch�log._EchoFlagType�echo�
Optional[str]�_orig_logging_namezUnion[_ConnDialect, Dialect]�_dialect�)Union[_CreatorFnType, _CreatorWRecFnType]�_creator_argrS�_invoke_creator�float�_invalidate_time���NTF�creator�recycle�int�logging_name�reset_on_return�_ResetStyleArgType�events�+Optional[List[Tuple[_ListenerFnType, str]]]�dialect�&Optional[Union[_ConnDialect, Dialect]]�pre_pingr#�	_dispatch�Optional[_DispatchCommon[Pool]]c
	C
s�|r	||_|_
nd
|_
tj||d�
|
|_||_d|_||_t�	|t
jddgt
jgd�
t
j
dg
id�|_||_|	rB|jj|	d	d
�
|rG||_|rW|D]
\}
}t�|||
�
qKd
Sd
S)a4
        Construct a Pool.

        :param creator: a callable function that returns a DB-API
          connection object.  The function will be called with
          parameters.

        :param recycle: If set to a value other than -1, number of
          seconds between connection recycling, which means upon
          checkout, if this timeout is surpassed the connection will be
          closed and replaced with a newly opened connection. Defaults to -1.

        :param logging_name:  String identifier which will be used within
          the "name" field of logging records generated within the
          "sqlalchemy.pool" logger. Defaults to a hexstring of the object's
          id.

        :param echo: if True, the connection pool will log
         informational output such as when connections are invalidated
         as well as when connections are recycled to the default log handler,
         which defaults to ``sys.stdout`` for output..   If set to the string
         ``"debug"``, the logging will include pool checkouts and checkins.

         The :paramref:`_pool.Pool.echo` parameter can also be set from the
         :func:`_sa.create_engine` call by using the
         :paramref:`_sa.create_engine.echo_pool` parameter.

         .. seealso::

             :ref:`dbengine_logging` - further detail on how to configure
             logging.

        :param reset_on_return: Determine steps to take on
         connections as they are returned to the pool, which were
         not otherwise handled by a :class:`_engine.Connection`.
         Available from :func:`_sa.create_engine` via the
         :paramref:`_sa.create_engine.pool_reset_on_return` parameter.

         :paramref:`_pool.Pool.reset_on_return` can have any of these values:

         * ``"rollback"`` - call rollback() on the connection,
           to release locks and transaction resources.
           This is the default value.  The vast majority
           of use cases should leave this value set.
         * ``"commit"`` - call commit() on the connection,
           to release locks and transaction resources.
           A commit here may be desirable for databases that
           cache query plans if a commit is emitted,
           such as Microsoft SQL Server.  However, this
           value is more dangerous than 'rollback' because
           any data changes present on the transaction
           are committed unconditionally.
         * ``None`` - don't do anything on the connection.
           This setting may be appropriate if the database / DBAPI
           works in pure "autocommit" mode at all times, or if
           a custom reset handler is established using the
           :meth:`.PoolEvents.reset` event handler.

         * ``True`` - same as 'rollback', this is here for
           backwards compatibility.
         * ``False`` - same as None, this is here for
           backwards compatibility.

         For further customization of reset on return, the
         :meth:`.PoolEvents.reset` event hook may be used which can perform
         any connection activity desired on reset.

         .. seealso::

            :ref:`pool_reset_on_return`

            :meth:`.PoolEvents.reset`

        :param events: a list of 2-tuples, each of the form
         ``(callable, target)`` which will be passed to :func:`.event.listen`
         upon construction.   Provided here so that event listeners
         can be assigned via :func:`_sa.create_engine` before dialect-level
         listeners are applied.

        :param dialect: a :class:`.Dialect` that will handle the job
         of calling rollback(), close(), or commit() on DBAPI connections.
         If omitted, a built-in "stub" dialect is used.   Applications that
         make use of :func:`_sa.create_engine` should not use this parameter
         as it is handled by the engine creation strategy.

        :param pre_ping: if True, the pool will emit a "ping" (typically
         "SELECT 1", but is dialect-specific) on the connection
         upon checkout, to test if the connection is alive or not.   If not,
         the connection is transparently re-connected and upon success, all
         other pooled connections established prior to that timestamp are
         invalidated.     Requires that a dialect is passed as well to
         interpret the disconnection error.

         .. versionadded:: 1.2

        N)
�echoflagr
r2T)�noneNFr1rhF)
�only_propagate)rgr\r�instance_logger�_creator�_recyclerb�	_pre_pingr�parse_user_argument_for_enumr,r.r0r/�_reset_on_returnrZrX�_updater]r�listen)r;rdrerZrgrhrjrlrnro�fn�targetr*r*r+�__init__�s4l










��






�
z
Pool.__init__r6c


C
�|jj
Sr9)r]rHrMr*r*r+�_is_asyncio3
�zPool._is_asyncioc


C
�|jSr9)
r_rMr*r*r+ru7
�z
Pool._creatorr7cC
s|
|_|�
|
�
|_dSr9)r_�_should_wrap_creatorr`)r;rdr*r*r+ru;
sc


C
s|`|`
dSr9)r_r`rMr*r*r+ruE
s
c
s�z
tj
|jd
d�}Wnty


tt|
���f
dd�YSw|jdu
r*t|j�
}nd}t|d�
|}|d|dfdg
d	fkrFtt|
�S|d
krOtt|
�Stt|
���f
dd�S)zlDetect if creator accepts a single argument, or is sent
        as a legacy style no-arg function.

        T)
�no_selfc



���Sr9r*�
rT�
�
creator_fnr*r+�<lambda>W
�z+Pool._should_wrap_creator.<locals>.<lambda>Nr
��connection_recordr9r-c



r�r9r*r�r�r*r+r�j
r�)	r�get_callable_argspecru�	TypeErrorrrK�defaults�lenrS)r;rd�argspec�	defaulted�positionalsr*r�r+r�K
s 




�








zPool._should_wrap_creator�
�	terminaterFrr�c

C
s�|j�
d
|rdn
d|
�
z|r|j�|
�

WdS|j�|
�

WdStyJ
}
z|jjd|r0dn
d�d�|
dd	�
t|t�s?�WYd}~dSd}~w
w)
Nz%s connection %rzHard-closing�Closingz
Exception �terminating�closingz connection %rT�
�exc_info)	�logger�debugr]rArB�
BaseException�error�
isinstance�	Exception)r;rFr��
er*r*r+�_close_connectionl
s(



�





�

���zPool._close_connectionrUc

C
st|�
S)
z6Called by subclasses to create a new ConnectionRecord.)
�_ConnectionRecordrMr*r*r+�_create_connection�
szPool._create_connectionr5�	exception�Optional[BaseException]�_checkincC
sLt|
d
d�}|r|j
|jkrt��|_
|r"t|
dd�r$|
�|�

dSdSdS)a�
Mark all connections established within the generation
        of the given connection as invalidated.

        If this pool's last invalidate time is before when the given
        connection was created, update the timestamp til now.  Otherwise,
        no action is performed.

        Connections with a start time prior to this pool's invalidation
        time will be recycled upon next checkout.
        �_connection_recordN�is_validF)�getattrrb�	starttime�time�
invalidate)r;rFr�r�rTr*r*r+�_invalidate�
s




�zPool._invalidatec


C
�t��
)
a"
Return a new :class:`_pool.Pool`, of the same class as this one
        and configured with identical creation arguments.

        This method is used in conjunction with :meth:`dispose`
        to close out an entire :class:`_pool.Pool` and create a new one in
        its place.

        rCrMr*r*r+�recreate�
s
z
Pool.recreatec


C
r�)
z�Dispose of this pool.

        This method leaves the possibility of checked-out connections
        remaining open, as it only affects connections that are
        idle in the pool.

        .. seealso::

            :meth:`Pool.recreate`

        rCrMr*r*r+�dispose�
�
zPool.disposec

C
s
t�
|�
S)
z�Return a DBAPI connection from the pool.

        The connection is instrumented such that when its
        ``close()`` method is called, the connection will be returned to
        the pool.

        )�_ConnectionFairy�	_checkoutrMr*r*r+�connect�
s
zPool.connect�recordcC
s|�|
�

d
S)z�Given a _ConnectionRecord, return it to the :class:`_pool.Pool`.

        This method is called when an instrumented DBAPI connection
        has its ``close()`` method called.

        N)
�_do_return_conn�r;r�r*r*r+�_return_conn�
szPool._return_connc


C
r�)
z7Implementation for :meth:`get`, supplied by subclasses.rCrMr*r*r+�_do_get�
�zPool._do_getc
C
r�)
z?Implementation for :meth:`return_conn`, supplied by subclasses.rCr�r*r*r+r��
r�zPool._do_return_conn�strc


C
r�r9rCrMr*r*r+�status�
s
zPool.status)rcNNTNNFN)rdr^rerfrZrYrgr[rhrirjrkrlrmrnr#rorp�r6r#)r6r^)rdr^r6r7�r6r7)rdr^r6rS)rFrr�r#r6r7)r6rU�NT)rFr5r�r�r�r#r6r7)r6rW)r6r5)r�rUr6r7)r6r�)r$r%r&r'r)r3r]r~r�hybridpropertyr��propertyru�setter�deleterr�r�r�r�r�r�r�r�r�r�r�r*r*r*r+rW�sR













�


	

"�

�





	
rWc@
sXeZ
dZUd
ZdZded<	ded<	ejdd	d
��
Zejddd
��
Z		dddd�Z
dS)�ManagesConnectiona3
Common base for the two connection-management interfaces
    :class:`.PoolProxiedConnection` and :class:`.ConnectionPoolEntry`.

    These two objects are typically exposed in the public facing API
    via the connection pool event hooks, documented at :class:`.PoolEvents`.

    .. versionadded:: 2.0

    r*�Optional[DBAPIConnection]r4�
Optional[Any]�driver_connectionr6rc


C
r�)
a�Info dictionary associated with the underlying DBAPI connection
        referred to by this :class:`.ManagesConnection` instance, allowing
        user-defined data to be associated with the connection.

        The data in this dictionary is persistent for the lifespan
        of the DBAPI connection itself, including across pool checkins
        and checkouts.  When the connection is invalidated
        and replaced with a new one, this dictionary is cleared.

        For a :class:`.PoolProxiedConnection` instance that's not associated
        with a :class:`.ConnectionPoolEntry`, such as if it were detached, the
        attribute returns a dictionary that is local to that
        :class:`.ConnectionPoolEntry`. Therefore the
        :attr:`.ManagesConnection.info` attribute will always provide a Python
        dictionary.

        .. seealso::

            :attr:`.ManagesConnection.record_info`


        rCrMr*r*r+�infoszManagesConnection.info�Optional[_InfoType]c


C
r�)
a�Persistent info dictionary associated with this
        :class:`.ManagesConnection`.

        Unlike the :attr:`.ManagesConnection.info` dictionary, the lifespan
        of this dictionary is that of the :class:`.ConnectionPoolEntry`
        which owns it; therefore this dictionary will persist across
        reconnects and connection invalidation for a particular entry
        in the connection pool.

        For a :class:`.PoolProxiedConnection` instance that's not associated
        with a :class:`.ConnectionPoolEntry`, such as if it were detached, the
        attribute returns None. Contrast to the :attr:`.ManagesConnection.info`
        dictionary which is never None.


        .. seealso::

            :attr:`.ManagesConnection.info`

        rCrMr*r*r+�record_info+szManagesConnection.record_infoNFr�r��softr#r7c
C
r�)
aO
Mark the managed connection as invalidated.

        :param e: an exception object indicating a reason for the invalidation.

        :param soft: if True, the connection isn't closed; instead, this
         connection will be recycled on next checkout.

        .. seealso::

            :ref:`pool_connection_invalidation`


        rC�r;r�r�r*r*r+r�CszManagesConnection.invalidate�r6r�r6r��NF�r�r�r�r#r6r7)r$r%r&r'r(r)r�ro_memoized_propertyr�r�r�r*r*r*r+r��
s






�r�c@
s,eZ
dZd
ZdZeddd��
Zddd	�Zd
S)
rUaInterface for the object that maintains an individual database
    connection on behalf of a :class:`_pool.Pool` instance.

    The :class:`.ConnectionPoolEntry` object represents the long term
    maintainance of a particular connection for a pool, including expiring or
    invalidating that connection to have it replaced with a new one, which will
    continue to be maintained by that same :class:`.ConnectionPoolEntry`
    instance. Compared to :class:`.PoolProxiedConnection`, which is the
    short-term, per-checkout connection manager, this object lasts for the
    lifespan of a particular "slot" within a connection pool.

    The :class:`.ConnectionPoolEntry` object is mostly visible to public-facing
    API code when it is delivered to connection pool event hooks, such as
    :meth:`_events.PoolEvents.connect` and :meth:`_events.PoolEvents.checkout`.

    .. versionadded:: 2.0  :class:`.ConnectionPoolEntry` provides the public
       facing interface for the :class:`._ConnectionRecord` internal class.

    r*r6r#c


C
r�)
z3Return True the connection is currently checked outrCrMr*r*r+�in_usemszConnectionPoolEntry.in_user7c


C
r�)
zAClose the DBAPI connection managed by this connection pool entry.rCrMr*r*r+r@sr�zConnectionPoolEntry.closeNr�r�)r$r%r&r'r(r�r�r@r*r*r*r+rUVs

rUc@
s(
eZ
dZUd
ZdZded<ded<ded<d	ed
<dFdGdd�Zded<edHdd��
Zee	�
dd�dIdd��
�
ZdZd	ed<e	j
dJdd��
Ze	j
dKd!d"��
ZedLd$d%��
Z	dFdMd*d+�ZdFdNd,d-�ZedOd.d/��
ZedPd0d1��
ZdQd2d3�Z	5dRdSd9d:�ZdTd<d=�ZdOd>d?�Zd5d@�
dUdBdC�ZdQdDdE�Zd4S)Vr�aMaintains a position in a connection pool which references a pooled
    connection.

    This is an internal object used by the :class:`_pool.Pool` implementation
    to provide context management to a DBAPI connection maintained by
    that :class:`_pool.Pool`.   The public facing interface for this class
    is described by the :class:`.ConnectionPoolEntry` class.  See that
    class for public API details.

    .. seealso::

        :class:`.ConnectionPoolEntry`

        :class:`.PoolProxiedConnection`

    )�__pool�	fairy_ref�finalize_callback�freshr�r4�__weakref__�__dict__z(Deque[Callable[[DBAPIConnection], None]]r�r#r��'Optional[weakref.ref[_ConnectionFairy]]r�rar�T�poolrWr�cC
s6d
|_d|_
d|_d|_|
|_|r|��
t�|_dS)NFr
)r�r�r�r4�_ConnectionRecord__pool�_ConnectionRecord__connectrr�)r;r�r�r*r*r+r~�s






z_ConnectionRecord.__init__r�r4r6r�c

C
s|jdurdS|j
j�|j�
Sr9)r4r�r]rGrMr*r*r+r��s



�z#_ConnectionRecord.driver_connection�2.0zXThe _ConnectionRecord.connection attribute is deprecated; please use 'driver_connection'c


C
r�r9�
r4rMr*r*r+rF��z_ConnectionRecord.connectionr
�_soft_invalidate_timerc


C
�iSr9r*rMr*r*r+r���z_ConnectionRecord.infor�c


C
r�r9r*rMr*r*r+r��r�z_ConnectionRecord.record_infor�c
s�tr
t
t�
����n�
���z���}Wn(ty<
}
zt���
�j|d
d�
Wd�
�1s2w



Y
�d}~w
w�
�	��t
�
|���}t�|��
�fdd���_
}�t|<�rd�
j�d|�
|S)NF�
�_fairy_was_createdc


s tdu
rtd��
|�d
d�SdS)NF�
r )
�_finalize_fairy)
�ref�rZr�rTr*r+r��s
�
�z,_ConnectionRecord.checkout.<locals>.<lambda>z#Connection %r checked out from pool)r
rr�r��get_connectionr�r�safe_reraise�_checkin_failed�_should_log_debugr��weakrefr�r��_strong_ref_connection_recordsr�r�)�clsr�r4�err�fairyr�r*r�r+�checkout�s4






����



�



�z_ConnectionRecord.checkoutr�r�r�r7cC
s|j|
d
�

|j
|d�

dS)N�
r�r�)r��checkin)r;r�r�r*r*r+r��s


�z!_ConnectionRecord._checkin_failedcC
s||jdur|
rt
�d
|�

dSd|_|j}|j}|jr,|j��}|du
r)||�

|js|jjr7|j�||�
|�	|�

dS)NzDouble checkin attempted on %s)
r�r�warnr4r�r��poprXr�r�)r;r�rFr��	finalizerr*r*r+r��s









�
z_ConnectionRecord.checkinc

C
�
|jdu
Sr9)
r�rMr*r*r+r��
z_ConnectionRecord.in_usec


C
r�r9)
r�rMr*r*r+�last_connect_time
r�z#_ConnectionRecord.last_connect_timec

C
s|jdu
r|�
�
dSdSr9)r4�_ConnectionRecord__closerMr*r*r+r@s


�z_ConnectionRecord.closeNFr�r�r�cC
s�|jdurdS|r|j
j�|j||
�
n
|j
j�|j||
�
|
du
r5|j
j�d
|r+dn
d|j|
jj|
�
n|j
j�d|r>dn
d|j�
|rLt	�	�|_
dS|jdd�

d|_dS)Nz*%sInvalidate connection %r (reason: %s:%s)zSoft �z%sInvalidate connection %rTr�)r4r�rX�soft_invalidater�r�r��	__class__r$r�r�r�r�r*r*r+r�s0




�







�



�


z_ConnectionRecord.invalidaterc
C
s�d
}
|jdur|j
��
|��
nA|jjdkr.t��|j|jjkr.|jj�
d|j�
d}
n$|jj	|jkrA|jj�
d|j�
d}
n|j
|jkrR|jj�
d|j�
d}
|
rc|jdd�

|j
��
|��
|jdu
sjJ�
|jS)NFrcz)Connection %r exceeded timeout; recyclingTz=Connection %r invalidated due to pool invalidation; recyclingzCConnection %r invalidated due to local soft invalidation; recyclingr�)r4r��clearr�r�rvr�r�r�rbr�r�)r;rer*r*r+r�3s:








�


�


�



z _ConnectionRecord.get_connectionc

C
s$|jdup|j
j|jkp|j|jkSr9)r4r�rbr�r�rMr*r*r+�_is_hard_or_soft_invalidatedgs


�
�z._ConnectionRecord._is_hard_or_soft_invalidatedr�r�c

C
sP|j�
�
|jjjr|jj�|j|�
|jdu
sJ�
|jj|j|
d
�
d|_dS)Nr�)r�r

r�rXr@r4r�)r;r�r*r*r+�__closens







�
z_ConnectionRecord.__closec
C
s�|j}
d|_
zt��|_|
�|�
|_
}|
j�d
|�
d|_Wn5tyT
}
z)t	�
��
|
j�d|�
Wd�
n1s=w



Y
WYd}~dSWYd}~dSd}~w
w|
jjrf|
jj�
|
j�
�|j
|�
|
jj�
|
j�
�|j
|�
dS)NzCreated new connection %rTzError on connect(): %s)r�r4r�r�r`r�r�r�r�rr�rX�
first_connect�
for_modify�exec_once_unless_exceptionr��_exec_w_sync_on_first_run)r;r�rFr�r*r*r+�	__connectxs,











8���

�
�z_ConnectionRecord.__connect)
T)r�rWr�r#�r6r�)r6r�r�r�)r�rWr6r�)r�r�r�r#r6r7)r�r#r6r7r�)r6rar�r�r�rP)r�r#r6r7)r$r%r&r'r(r)r~r�r�r�
deprecatedrFr�r�r�r��classmethodr�r�r�r�r�r@r�r�r
r�r�r*r*r*r+r�xsL








�


$�


�
!
4
r�Fr4r�r��Optional[_ConnectionRecord]r�r�r�rZ�Optional[log._EchoFlagType]r r#r��Optional[_ConnectionFairy]r6r7cC
sv|d
u
}|r)|d
u
sJ�
t�
|d
�
|
d
u
sJ�
|
j|u
rd
S|d
us%J�
|
j}n|r4t�
t�|�
d
�
|jj}|rU|
d
up?|}	|}
|jjpK|jj}|jjoS|jj}n
|
d
u}	d}
}d}|d
u
�
r|
ro|ro|j	�
d|�
z�z?|s�|
d
u
syJ�
t|||
|�}|j|us�J�
|j|||	|
d�
|	r�|
r�||_
|��
|r�|jjr�|j�|�

|j||d�
Wn(ty�
}

z|j	jddd�
|
r�|
j|
d	�

t|
t�s͂WYd
}
~
nd
}
~
w
wW|	r�|r�|r�d
|�d|s�dn
d
�d�}|j	�|�

t�|�

n&|	�
r|�
r|�
rd
|�d|�
s
dn
d
�d�}|j	�|�

t�|�

wwww|
�
r+|
jd
u
�
r+|
��
|d
u
�
r6d
|_d
|_~~
~d
S)ac
Cleanup for a :class:`._ConnectionFairy` whether or not it's already
    been garbage collected.

    When using an async dialect no IO can happen here (without using
    a dedicated thread), since this is called outside the greenlet
    context and with an already running loop. In this case function
    will only log a message and raise a warning.
    NTFz$Connection %r being returned to poolrr�z!Exception during reset or similarr�r�zFThe garbage collector is trying to clean up non-checked-in connection z, which will be z*dropped, as it cannot be safely terminated�
terminatedz�.  Please ensure that SQLAlchemy pooled connections are returned to the pool explicitly, either by calling ``close()`` or by using appropriate context managers to manage their lifecycle.)r�r�r�r4r�r�r]rHrIr�r�r��_reset�_pool�detachrX�close_detachedr�r�r�r�r�r�rr�r�r�)r4r�r�r�rZr r��
is_gc_cleanup�dont_restore_gcedr
�can_manipulate_connection�!can_close_or_terminate_connection�requires_terminate_for_closer��messager*r*r+r��s�










��





�






�



�






��

�



����	
����

��
����
�






r�z6Dict[weakref.ref[_ConnectionFairy], _ConnectionRecord]r�c@
sheZ
dZd
ZdZejrddd�Zddd	�Zdd
d�Z	e
dd
d��
Ze
ddd��
Zddd�Z
ddd�ZdS)r5aA connection-like adapter for a :pep:`249` DBAPI connection, which
    includes additional methods specific to the :class:`.Pool` implementation.

    :class:`.PoolProxiedConnection` is the public-facing interface for the
    internal :class:`._ConnectionFairy` implementation object; users familiar
    with :class:`._ConnectionFairy` can consider this object to be equivalent.

    .. versionadded:: 2.0  :class:`.PoolProxiedConnection` provides the public-
       facing interface for the :class:`._ConnectionFairy` internal class.

    r*r6r7c


C
rLr9r*rMr*r*r+r13rOzPoolProxiedConnection.commitrc


C
rLr9r*rMr*r*r+�cursor5rOzPoolProxiedConnection.cursorc


C
rLr9r*rMr*r*r+r27rOzPoolProxiedConnection.rollbackr#c


C
r�)
zgReturn True if this :class:`.PoolProxiedConnection` still refers
        to an active DBAPI connection.rCrMr*r*r+r�9�zPoolProxiedConnection.is_validc


C
r�)
zVReturn True if this :class:`.PoolProxiedConnection` is detached
        from its pool.rCrMr*r*r+�is_detached@r
z!PoolProxiedConnection.is_detachedc


C
r�)
aSeparate this connection from its Pool.

        This means that the connection will no longer be returned to the
        pool when closed, and will instead be literally closed.  The
        associated :class:`.ConnectionPoolEntry` is de-associated from this
        DBAPI connection.

        Note that any overall connection limiting constraints imposed by a
        Pool implementation may be violated after a detach, as the detached
        connection is removed from the pool's knowledge and control.

        rCrMr*r*r+r
GszPoolProxiedConnection.detachc


C
r�)
a$Release this connection back to the pool.

        The :meth:`.PoolProxiedConnection.close` method shadows the
        :pep:`249` ``.close()`` method, altering its behavior to instead
        :term:`release` the proxied connection back to the connection pool.

        Upon release to the pool, whether the connection stays "opened" and
        pooled in the Python process, versus actually closed out and removed
        from the Python process, is based on the pool implementation in use and
        its configuration and current state.

        rCrMr*r*r+r@Wr�zPoolProxiedConnection.closeNr�)r6rr�)r$r%r&r'r(�typingr
r1r
r2r�r�r
r
r@r*r*r*r+r5"s






r5c@
s�eZ
dZUd
ZdZded<ded<d&dd	�Zed'dd
��
Zed(dd��
Z	ed)dd��
Z
	d*d+dd�Zej
d,dd��
Zd-d!d"�Zd.d$d%�ZdS)/�_AdhocProxiedConnectionad
provides the :class:`.PoolProxiedConnection` interface for cases where
    the DBAPI connection is not actually proxied.

    This is used by the engine internals to pass a consistent
    :class:`.PoolProxiedConnection` object to consuming dialects in response to
    pool events that may not always have the :class:`._ConnectionFairy`
    available.

    �r4r��	_is_validrr4rUr�r�cC
s|
|_||_
d
|_dSr�r
)r;r4r�r*r*r+r~ws


z _AdhocProxiedConnection.__init__r6rc


C
rr9�r�r�rMr*r*r+r��r�z)_AdhocProxiedConnection.driver_connectionc


C
r�r9r�rMr*r*r+rF�r�z"_AdhocProxiedConnection.connectionr#c


C
r�)
z�Implement is_valid state attribute.

        for the adhoc proxied connection it's assumed the connection is valid
        as there is no "invalidate" routine.

        �
r 
rMr*r*r+r��sz _AdhocProxiedConnection.is_validNFr�r�r�r7cC
s
d
|_dSr�r"
r�r*r*r+r��s
z"_AdhocProxiedConnection.invalidater�c


C
rr9�r�r�rMr*r*r+r��r�z#_AdhocProxiedConnection.record_info�args�kwargsrc
O
s|jj
|
i|�
�
Sr9�r4r
�r;r$
r%
r*r*r+r
�s
z_AdhocProxiedConnection.cursor�keycC
�t|j
|
�Sr9�r�r4�r;r(
r*r*r+�__getattr__�r=z#_AdhocProxiedConnection.__getattr__)r4rr�rU)r6rrPr�r�r�r��r$
rr%
rr6r)r(
rr6r)r$r%r&r'r(r)r~r�r�rFr�r�r�ro_non_memoized_propertyr�r
r,
r*r*r*r+r
gs$




	



�

r
c@
s2
eZ
dZUd
ZdZded<ded<ded<dPdd
�Zded<edQdd��
Zee	�
dd�dRdd��
�
Ze		dSdTdd��
Z
dUdd �ZdVdWd%d&�ZdXd'd(�ZdYd+d,�ZedZd.d/��
Zed[d0d1��
Zed[d2d3��
Ze	jd\d5d6��
Ze	jd]d8d9��
Z	!d^d_d=d>�Zd`dCdD�ZdadGdH�ZdXdIdJ�ZdXdKdL�ZdVdbdNdO�ZdS)cr�a&Proxies a DBAPI connection and provides return-on-dereference
    support.

    This is an internal object used by the :class:`_pool.Pool` implementation
    to provide context management to a DBAPI connection delivered by
    that :class:`_pool.Pool`.   The public facing interface for this class
    is described by the :class:`.PoolProxiedConnection` class.  See that
    class for public API details.

    The name "fairy" is inspired by the fact that the
    :class:`._ConnectionFairy` object's lifespan is transitory, as it lasts
    only for the length of a specific DBAPI connection being checked out from
    the pool, and additionally that as a transparent proxy, it is mostly
    invisible.

    .. seealso::

        :class:`.PoolProxiedConnection`

        :class:`.ConnectionPoolEntry`


    )r4r��_echor
�_counterr�r�rWr�rr4rYr/
r�r�rZcC
s"|
|_d
|_
||_||_||_dS)Nr
)r
r0
r4r�r/
)r;r�r4r�rZr*r*r+r~�s





z_ConnectionFairy.__init__r
r�r6r�c

C
�|jdurdS|jj
Sr9r!
rMr*r*r+r��s


z"_ConnectionFairy.driver_connectionr�zWThe _ConnectionFairy.connection attribute is deprecated; please use 'driver_connection'c


C
r�r9r�rMr*r*r+rF�r�z_ConnectionFairy.connectionN�threadconns�Optional[threading.local]r�r
cC
s�|st�
|
�
}|du
rt�|�
|_|jdu
sJd
�
�
|jdu
s#Jd�
�
|jd7_|
jj
s1|
j	r6|jdkr8|Sd}|dk�
r1|jj
}d|j_
zA|
j	r||sq|jrX|
j�
d|j�
|
j�|j�
}|sp|jrl|
j�
d|j�
t���
n|jr||
j�
d	|j�
|
j�
|j|j|�
|WStjy�
}
ze|jr�|
j�d
|�
|j�|�

|
j||dd�
n|
j�d|j|�
|j�|�

z|j��|_Wn*ty�
}
zt���
|jj|d
d�
~Wd�
�1s�w



Y
�d}~w
w|d8}WYd}~n7d}~w
t�
y,
}	
z&t���
|j}
|
du
�
r|
j|	d
d�
~Wd�
�1�
s"w



Y
�d}	~	w
w|dks?|
j�d�

|��
t�d�
�
)Nz,can't 'checkout' a detached connection fairyz0can't 'checkout' an invalidated connection fairyr-rr
FzPool pre-ping on connection %sz;Pool pre-ping on connection %s failed, will invalidate poolz)Connection %s is fresh, skipping pre-pingzoDisconnection detected on checkout, invalidating all pooled connections prior to current timestamp (reason: %r)�
r�zVDisconnection detected on checkout, invalidating individual connection %s (reason: %r)Tr�z+Reconnection attempts exhausted on checkoutzThis connection is closed)r�r�r�r��currentr�r4r0
rXrwr�r/
r�r�r]rEr�InvalidatePoolError�DisconnectionError�invalidate_poolr�r�r�r�r�rr�r��InvalidRequestError)r�r�r2
r��attempts�connection_is_fresh�resultr�r��be_outerrTr*r*r+r��s�



�
��
�











�
�


��


�

�



�


�
�




�

�����








�
�
�
���X


z_ConnectionFairy._checkoutc

C
stj
|j|d
�S)N)
r�)r�r�r
rMr*r*r+�_checkout_existinges
z#_ConnectionFairy._checkout_existingFr r#r7c	C
s"t|j
|j|jd|j|
|d
�
dS)N)r r�)r�r4r�r
r/
)r;r r*r*r+r�hs








�z_ConnectionFairy._checkinc

C
s|��
dSr9r4
rMr*r*r+�_closesr=z_ConnectionFairy._closer!r"c	C
s�|
jj
r|
j�
|j|jt|||d
��
|sdS|
jtur@|r-|jr+|
j�	d|j�
dSdS|jr8|
j�	d|j�
|
j
�|�

dS|
jturX|jrP|
j�	d|j�
|
j
�
|�

dSdS)Nrz.Connection %s reset, transaction already resetz Connection %s rollback-on-returnzConnection %s commit-on-return)rX�resetr4r�rryr.r/
r�r�r]r<r/r>)r;r�r r!r"r*r*r+r
vsD






��







��


�





��z_ConnectionFairy._reset�log._IdentifiedLoggerTypec


C
rr9)r
r�rMr*r*r+�_logger�r�z_ConnectionFairy._loggerc

C
r�r9r�rMr*r*r+r��r�z_ConnectionFairy.is_validc

C
s
|jduSr9)
r�rMr*r*r+r
�r�z_ConnectionFairy.is_detachedrc

C
s|jduriS|jj
Sr9)r�r�rMr*r*r+r���

z_ConnectionFairy.infor�c

C
r1
r9r#
rMr*r*r+r��rC
z_ConnectionFairy.record_infor�r�r�cC
sH|jdurt
�d
�

dS|jr|jj|
|d�
|s"d|_|��
dSdS)Nz.Can't invalidate an already-closed connection.)r�r�)r4rr�r�r�r�r�r*r*r+r��s






�z_ConnectionFairy.invalidater$
rr%
rc
O
s |jdu
sJ�
|jj
|
i|�
�
Sr9r&
r'
r*r*r+r
�s

z_ConnectionFairy.cursorr(
r�cC
r)
r9r*
r+
r*r*r+r,
�r=z_ConnectionFairy.__getattr__c
C
sd|jdu
r.|j}
d|
_
d|
_|j�|j�

|j��|_d|_|jjjr0|jj�|j|
�
dSdSdSr9)	r�r�r4r
r�r��copyrXr
rVr*r*r+r
�s






�
z_ConnectionFairy.detachc

C
s(|jd
8_|jdkr|�
�
dSdS)Nr-r
�r0
r�rMr*r*r+r@�s



�z_ConnectionFairy.close�transaction_resetcC
s,|jd
8_|jdkr|j
|
d�

dSdS)Nr-r
r�rE
)r;rF
r*r*r+�_close_special�s



�z_ConnectionFairy._close_special)r�rWr4rr�r�rZrYr	
rP)NN)r�rWr2
r3
r�r
r6r�)r6r�)
F)r r#r6r7r�)
r�rWr r#r!r#r"r#r6r7)r6rA
r�r�r�r�r�r-
)r(
r�r6r)rF
r#r6r7)r$r%r&r'r(r)r~r�r�rr

rFr
r�r>
r�r?
r
rB
r�r
r�r�r.
r�r�r
r,
r
r@rG
r*r*r*r+r��sR










�
�
}

+




�



r�)FN)r4r�r�r
r�rWr�r�rZr

r r#r�r
r6r7)?r'�
__future__r�collectionsr�dataclasses�enumr�	threadingr�r
rrrrr	r
rrr
rr�r�rrrr�util.typingrr�engine.interfacesrrrrrr�sql._typingr�	dataclassrr,ri�listr.r/r0r3rJrKrS�
Identified�EventTargetrWr�rUr�r�r�r)r5r
r�r*r*r*r+�<module>
s~





























,

��#?|"'
�
�E;