File: //usr/lib/python3/dist-packages/twisted/enterprise/__pycache__/adbapi.cpython-38.pyc
U
��W[�B � @ st d Z ddlZddlmZ ddlmZmZmZ G dd� de�Z G dd� de
�Z G d d
� d
�Z G d d � d �Z
d
d gZdS )
zg
An asynchronous mapping to U{DB-API
2.0<http://www.python.org/topics/database/DatabaseAPI-2.0.html>}.
� N)�threads)�reflect�log�compatc @ s e Zd ZdZdS )�ConnectionLostzb
This exception means that a db connection has been lost. Client code may
try again.
N)�__name__�
__module__� __qualname__�__doc__� r r �;/usr/lib/python3/dist-packages/twisted/enterprise/adbapi.pyr s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�
Connectiona
A wrapper for a DB-API connection instance.
The wrapper passes almost everything to the wrapped connection and so has
the same API. However, the L{Connection} knows about its pool and also
handle reconnecting should when the real connection dies.
c C s || _ d | _| �� d S �N)�_pool� _connection� reconnect)�self�poolr r r �__init__! s zConnection.__init__c C s d S r r �r r r r �close' s zConnection.closec C s� | j js| j�� d S z:| j�� | j�� }|�| j j� |�� | j�� W d S t �
d d� Y nX | j � | j� | j j r�t �
d� t� �d S )N�Rollback failedzConnection lost.)r r r �rollback�cursor�execute�good_sqlr �commitr �err�
disconnect�noisy�msgr )r Zcursr r r r 2 s
zConnection.rollbackc C s( | j d k r| j�| j � | j�� | _ d S r )r r r �connectr r r r r I s
zConnection.reconnectc C s t | j|�S r )�getattrr �r �namer r r � __getattr__O s zConnection.__getattr__N) r r r r
r r r r r% r r r r r
s r
c @ s< e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z dS )
� Transactiona
A lightweight wrapper for a DB-API 'cursor' object.
Relays attribute access to the DB cursor. That is, you can call
C{execute()}, C{fetchall()}, etc., and they will be called on the
underlying DB-API cursor object. Attributes will also be retrieved from
there.
Nc C s || _ || _| �� d S r )r r �reopen)r r Z
connectionr r r r _ s zTransaction.__init__c C s | j }d | _ |�� d S r )�_cursorr )r r( r r r r e s zTransaction.closec C st | j d k r| �� z| j�� | _ W d S | jjs8� n t�d d� Y nX | jjr\t� d� | �� | j�� | _ d S )NzCursor creation failedz Connection lost, reconnecting)
r( r r r r r r r r r r r r r r' k s
zTransaction.reopenc C s | j �� d | _d S r )r r r( r r r r r s
zTransaction.reconnectc C s t | j|�S r )r" r( r# r r r r% � s zTransaction.__getattr__)
r r r r
r( r r r' r r% r r r r r& T s r& c @ s� e Zd ZdZd�� ZdZdZdZdZ dZ
dZ dZ dZ
eZeZdZdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z d d � Z d d � Z d d!� Z d"d#� Z d$d%� Z!d&d'� Z"d(d)� Z#d*d+� Z$dS ),�ConnectionPoola�
Represent a pool of connections to a DB-API 2.0 compliant database.
@ivar connectionFactory: factory for connections, default to L{Connection}.
@type connectionFactory: any callable.
@ivar transactionFactory: factory for transactions, default to
L{Transaction}.
@type transactionFactory: any callable
@ivar shutdownID: L{None} or a handle on the shutdown event trigger which
will be used to stop the connection pool workers when the reactor
stops.
@ivar _reactor: The reactor which will be used to schedule startup and
shutdown events.
@type _reactor: L{IReactorCore} provider
z-min max name noisy openfun reconnect good_sqlF� � Nzselect 1c O s || _ t�|�| _t| jdd�dkr.t�d� t| jdd�dk rJt�d� |�d d�}|dkrjdd
lm } || _
|| _ || _ | j
D ],}d |f }||kr�t| ||| � ||= q�t| j| j�| _t| j| j�| _i | _dd lm} dd
lm} |j| _|�| j| j�| _| j
�| j�| _dS )a�
Create a new L{ConnectionPool}.
Any positional or keyword arguments other than those documented here
are passed to the DB-API object when connecting. Use these arguments to
pass database names, usernames, passwords, etc.
@param dbapiName: an import string to use to obtain a DB-API compatible
module (e.g. C{'pyPgSQL.PgSQL'})
@param cp_min: the minimum number of connections in pool (default 3)
@param cp_max: the maximum number of connections in pool (default 5)
@param cp_noisy: generate informational log messages during operation
(default C{False})
@param cp_openfun: a callback invoked after every C{connect()} on the
underlying DB-API object. The callback is passed a new DB-API
connection object. This callback can setup per-connection state
such as charset, timezone, etc.
@param cp_reconnect: detect connections which have failed and reconnect
(default C{False}). Failed connections may result in
L{ConnectionLost} exceptions, which indicate the query may need to
be re-sent.
@param cp_good_sql: an sql query which should always succeed and change
no state (default C{'select 1'})
@param cp_reactor: use this reactor instead of the global reactor
(added in Twisted 10.2).
@type cp_reactor: L{IReactorCore} provider
ZapilevelNz2.0z'DB API module not DB API 2.0 compliant.Z threadsafetyr � z+DB API module not sufficiently thread-safe.Z
cp_reactor)�reactorzcp_%s)�
threadpool)�
threadable)� dbapiNamer Z namedModule�dbapir" r r �pop�twisted.internetr- �_reactor�connargs�connkw�CP_ARGS�setattr�min�max� connections�twisted.pythonr. r/ Z getThreadID�threadIDZ
ThreadPoolZcallWhenRunning�_start�startID) r r0 r5 r6 r- �argZcpArgr. r/ r r r r � s2 #
zConnectionPool.__init__c C s d | _ | �� S r )r? �startr r r r r> � s zConnectionPool._startc C s. | j s*| j�� | j�dd| j�| _d| _ dS )z�
Start the connection pool.
If you are using the reactor normally, this function does *not*
need to be called.
ZduringZshutdownTN)�runningr. rA r4 ZaddSystemEventTrigger�
finalClose�
shutdownIDr r r r rA � s
�zConnectionPool.startc O s t j| j| j| j|f|�|�S )a
Execute a function with a database connection and return the result.
@param func: A callable object of one argument which will be executed
in a thread with a connection from the pool. It will be passed as
its first argument a L{Connection} instance (whose interface is
mostly identical to that of a connection object for your DB-API
module of choice), and its results will be returned as a
L{Deferred}. If the method raises an exception the transaction will
be rolled back. Otherwise, the transaction will be committed.
B{Note} that this function is B{not} run in the main thread: it
must be threadsafe.
@param *args: positional arguments to be passed to func
@param **kw: keyword arguments to be passed to func
@return: a L{Deferred} which will fire the return value of
C{func(Transaction(...), *args, **kw)}, or a
L{twisted.python.failure.Failure}.
)r �deferToThreadPoolr4 r. �_runWithConnection)r �func�args�kwr r r �runWithConnection s ���z ConnectionPool.runWithConnectionc
O sz | � | �}z ||f|�|�}|�� |W S t�� \}}}z |�� W n t�d d� Y nX t�||� Y nX d S �Nr ) �connectionFactoryr �sys�exc_infor r r r �reraise) r rG rH rI �conn�result�excType�excValue� excTracebackr r r rF & s
z!ConnectionPool._runWithConnectionc O s t j| j| j| j|f|�|�S )a�
Interact with the database and return the result.
The 'interaction' is a callable object which will be executed in a
thread using a pooled connection. It will be passed an L{Transaction}
object as an argument (whose interface is identical to that of the
database cursor for your DB-API module of choice), and its results will
be returned as a L{Deferred}. If running the method raises an
exception, the transaction will be rolled back. If the method returns a
value, the transaction will be committed.
NOTE that the function you pass is *not* run in the main thread: you
may have to worry about thread-safety in the function you pass to this
if it tries to use non-local objects.
@param interaction: a callable object whose first argument is an
L{adbapi.Transaction}.
@param *args: additional positional arguments to be passed to
interaction
@param **kw: keyword arguments to be passed to interaction
@return: a Deferred which will fire the return value of
C{interaction(Transaction(...), *args, **kw)}, or a
L{twisted.python.failure.Failure}.
)r rE r4 r. �_runInteraction)r � interactionrH rI r r r �runInteraction5 s ���z ConnectionPool.runInteractionc O s | j | jf|�|�S )aY
Execute an SQL query and return the result.
A DB-API cursor which will be invoked with C{cursor.execute(*args,
**kw)}. The exact nature of the arguments will depend on the specific
flavor of DB-API being used, but the first argument in C{*args} be an
SQL statement. The result of a subsequent C{cursor.fetchall()} will be
fired to the L{Deferred} which is returned. If either the 'execute' or
'fetchall' methods raise an exception, the transaction will be rolled
back and a L{twisted.python.failure.Failure} returned.
The C{*args} and C{**kw} arguments will be passed to the DB-API
cursor's 'execute' method.
@return: a L{Deferred} which will fire the return value of a DB-API
cursor's 'fetchall' method, or a L{twisted.python.failure.Failure}.
)rW � _runQuery�r rH rI r r r �runQueryV s zConnectionPool.runQueryc O s | j | jf|�|�S )aK
Execute an SQL query and return L{None}.
A DB-API cursor which will be invoked with C{cursor.execute(*args,
**kw)}. The exact nature of the arguments will depend on the specific
flavor of DB-API being used, but the first argument in C{*args} will be
an SQL statement. This method will not attempt to fetch any results
from the query and is thus suitable for C{INSERT}, C{DELETE}, and other
SQL statements which do not return values. If the 'execute' method
raises an exception, the transaction will be rolled back and a
L{Failure} returned.
The C{*args} and C{*kw} arguments will be passed to the DB-API cursor's
'execute' method.
@return: a L{Deferred} which will fire with L{None} or a
L{twisted.python.failure.Failure}.
)rW �
_runOperationrY r r r � runOperationk s zConnectionPool.runOperationc C s@ | j r| j�| j � d| _ | jr4| j�| j� d| _| �� dS )zC
Close all pool connections and shutdown the pool.
N)rD r4 ZremoveSystemEventTriggerr? rC r r r r r � s zConnectionPool.closec C s> d| _ | j�� d| _| j�� D ]}| �|� q | j�� dS )zE
This should only be called by the shutdown trigger.
NF)rD r. �stoprB r; �values�_close�clear�r rP r r r rC � s
zConnectionPool.finalClosec C sx | � � }| j�|�}|dkrt| jrDt�d| j| jp4d| jp<df � | j j
| j| j�}| j dkrj| � |� || j|< |S )a�
Return a database connection when one becomes available.
This method blocks and should be run in a thread from the internal
threadpool. Don't call this method directly from non-threaded code.
Using this method outside the external threadpool may exceed the
maximum number of connections in the pool.
@return: a database connection from the pool.
Nzadbapi connecting: %s %s%s� ) r= r; �getr r r r0 r5 r6 r1 r! �openfun)r �tidrP r r r r! � s
�
zConnectionPool.connectc C s> | � � }|| j�|�k r td��|dk r:| �|� | j|= dS )a
Disconnect a database connection associated with this pool.
Note: This function should only be used by the same thread which called
L{ConnectionPool.connect}. As with C{connect}, this function is not
used in normal non-threaded Twisted code.
zwrong connection for threadN)r= r; rc � Exceptionr_ )r rP re r r r r � s
zConnectionPool.disconnectc C sB | j rt�d| jf � z |�� W n t�d d� Y nX d S )Nzadbapi closing: %szConnection close failed)r r r r0 r r ra r r r r_ � s zConnectionPool._closec
O s� | � | �}| �| |�}z&||f|�|�}|�� |�� |W S t�� \}}} z |�� W n t�d d� Y nX t �
|| � Y nX d S rK ) rL �transactionFactoryr r rM rN r r r r rO )
r rV rH rI rP �transrQ rR rS rT r r r rU � s
z ConnectionPool._runInteractionc O s |j ||� |�� S r )r Zfetchall�r rh rH rI r r r rX � s zConnectionPool._runQueryc O s |j ||� d S r )r ri r r r r[ � s z ConnectionPool._runOperationc C s& | j | j| j| j| j| j| j| jd�S )N�r0 r9 r: r r r r5 r6 rj r r r r � __getstate__� s �zConnectionPool.__getstate__c C s" || _ | j| jf| j�| j� d S r )�__dict__r r0 r5 r6 )r �stater r r � __setstate__� s zConnectionPool.__setstate__)%r r r r
�splitr7 r r9 r: r$ rd r r rB r
rL r& rg rD r r> rA rJ rF rW rZ r\ r rC r! r r_ rU rX r[ rk rn r r r r r) � s> I!
r) )r
rM r3 r r< r r r rf r �objectr
r& r) �__all__r r r r �<module> s <5 q