fume-manage-python.git

U  
M±dQã@s¤dZddlmZddlmZddlmZGdddeZGdd    d    eZ    Gd
ddeZ
Gdd d eZGdddZGdddZ GdddZGdddZdS)a·PooledDB - pooling for DB-API 2 connections.
 
Implements a pool of steady, thread-safe cached connections
to a database which are transparently reused,
using an arbitrary DB-API 2 compliant database interface module.
 
This should result in a speedup for persistent applications such as the
application server of "Webware for Python," without loss of robustness.
 
Robustness is provided by using "hardened" SteadyDB connections.
Even if the underlying database is restarted and all connections
are lost, they will be automatically and transparently reopened.
However, since you don't want this to happen in the middle of a database
transaction, you must explicitly start transactions with the begin()
method so that SteadyDB knows that the underlying connection shall not
be replaced and errors passed on until the transaction is completed.
 
Measures are taken to make the pool of connections thread-safe.
If the underlying DB-API module is thread-safe at the connection level,
the requested connections may be shared with other threads by default,
but you can also request dedicated connections in case you need them.
 
For the Python DB-API 2 specification, see:
    https://www.python.org/dev/peps/pep-0249/
For information on Webware for Python, see:
    https://webwareforpython.github.io/w4py/
 
 
Usage:
 
First you need to set up the database connection pool by creating
an instance of PooledDB, passing the following parameters:
 
    creator: either an arbitrary function returning new DB-API 2
        connection objects or a DB-API 2 compliant database module
    mincached: the initial number of idle connections in the pool
        (the default of 0 means no connections are made at startup)
    maxcached: the maximum number of idle connections in the pool
        (the default value of 0 or None means unlimited pool size)
    maxshared: maximum number of shared connections allowed
        (the default value of 0 or None means all connections are dedicated)
        When this maximum number is reached, connections are
        shared if they have been requested as shareable.
    maxconnections: maximum number of connections generally allowed
        (the default value of 0 or None means any number of connections)
    blocking: determines behavior when exceeding the maximum
        (if this is set to true, block and wait until the number of
        connections decreases, but by default an error will be reported)
    maxusage: maximum number of reuses of a single connection
        (the default of 0 or None means unlimited reuse)
        When this maximum usage number of the connection is reached,
        the connection is automatically reset (closed and reopened).
    setsession: an optional list of SQL commands that may serve to
        prepare the session, e.g. ["set datestyle to german", ...]
    reset: how connections should be reset when returned to the pool
        (False or None to rollback transcations started with begin(),
        the default value True always issues a rollback for safety's sake)
    failures: an optional exception class or a tuple of exception classes
        for which the connection failover mechanism shall be applied,
        if the default (OperationalError, InternalError) is not adequate
    ping: an optional flag controlling when connections are checked
        with the ping() method if such a method is available
        (0 = None = never, 1 = default = whenever fetched from the pool,
        2 = when a cursor is created, 4 = when a query is executed,
        7 = always, and all other bit combinations of these values)
 
    The creator function or the connect function of the DB-API 2 compliant
    database module specified as the creator will receive any additional
    parameters such as the host, database, user, password etc.  You may
    choose some or all of these parameters in your own creator function,
    allowing for sophisticated failover and load-balancing mechanisms.
 
For instance, if you are using pgdb as your DB-API 2 database module and
want a pool of at least five connections to your local database 'mydb':
 
    import pgdb  # import used DB-API 2 module
    from dbutils.pooled_db import PooledDB
    pool = PooledDB(pgdb, 5, database='mydb')
 
Once you have set up the connection pool you can request
database connections from that pool:
 
    db = pool.connection()
 
You can use these connections just as if they were ordinary
DB-API 2 connections.  Actually what you get is the hardened
SteadyDB version of the underlying DB-API 2 connection.
 
Please note that the connection may be shared with other threads
by default if you set a non-zero maxshared parameter and the DB-API 2
module allows this.  If you want to have a dedicated connection, use:
 
    db = pool.connection(shareable=False)
 
You can also use this to get a dedicated connection:
 
    db = pool.dedicated_connection()
 
If you don't need it any more, you should immediately return it to the
pool with db.close().  You can get another connection in the same way.
 
Warning: In a threaded environment, never do the following:
 
    pool.connection().cursor().execute(...)
 
This would release the connection too early for reuse which may be
fatal if the connections are not thread-safe.  Make sure that the
connection object stays alive as long as you are using it, like that:
 
    db = pool.connection()
    cur = db.cursor()
    cur.execute(...)
    res = cur.fetchone()
    cur.close()  # or del cur
    db.close()  # or del db
 
Note that you need to explicitly start transactions by calling the
begin() method.  This ensures that the connection will not be shared
with other threads, that the transparent reopening will be suspended
until the end of the transaction, and that the connection will be rolled
back before being given back to the connection pool.
 
 
Ideas for improvement:
 
* Add a thread for monitoring, restarting (or closing) bad or expired
  connections (similar to DBConnectionPool/ResourcePool by Warren Smith).
* Optionally log usage, bad connections and exceeding of limits.
 
 
Copyright, credits and license:
 
* Contributed as supplement for Webware for Python and PyGreSQL
  by Christoph Zwerschke in September 2005
* Based on the code of DBPool, contributed to Webware for Python
  by Dan Green in December 2000
 
Licensed under the MIT license.
é)Ú    Conditioné)Ú__version__)Úconnectc@seZdZdZdS)Ú PooledDBErrorzGeneral PooledDB error.N©Ú__name__Ú
__module__Ú__qualname__Ú__doc__©rrúHd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\dbutils/pooled_db.pyrsrc@seZdZdZdS)ÚInvalidConnectionzDatabase connection is invalid.Nrrrrr rsrc@seZdZdZdS)ÚNotSupportedErrorz(DB-API module not supported by PooledDB.Nrrrrr rsrc@seZdZdZdS)ÚTooManyConnectionsz*Too many database connections were opened.Nrrrrr rsrc
@s`eZdZdZeZdddZd    d
ZdddZd dZ    ddZ
ddZddZddZ ddZdS)ÚPooledDBzPool for DB-API 2 connections.
 
    After you have created the connection pool, you can use
    connection() to get pooled, steady DB-API 2 connections.
    rFNTrcsjz
|j}WnDtk
rNzt|js*tWntk
rDd}YnXd}YnX|s\td|_|| __|_|_    |_
|    _|
_|_ |dkr d}|dkr¬d}|dkr¸d}|rÐ||krÈ|}|_nd_|dkrð|rð|_g_nd_|r ||kr
|}||kr|}|_nd_g_t_d_fddt|D}|rf| ¡ ¡qPdS)a    Set up the DB-API 2 connection pool.
 
        creator: either an arbitrary function returning new DB-API 2
            connection objects or a DB-API 2 compliant database module
        mincached: initial number of idle connections in the pool
            (0 means no connections are made at startup)
        maxcached: maximum number of idle connections in the pool
            (0 or None means unlimited pool size)
        maxshared: maximum number of shared connections
            (0 or None means all connections are dedicated)
            When this maximum number is reached, connections are
            shared if they have been requested as shareable.
        maxconnections: maximum number of connections generally allowed
            (0 or None means an arbitrary number of connections)
        blocking: determines behavior when exceeding the maximum
            (if this is set to true, block and wait until the number of
            connections decreases, otherwise an error will be reported)
        maxusage: maximum number of reuses of a single connection
            (0 or None means unlimited reuse)
            When this maximum usage number of the connection is reached,
            the connection is automatically reset (closed and reopened).
        setsession: optional list of SQL commands that may serve to prepare
            the session, e.g. ["set datestyle to ...", "set time zone ..."]
        reset: how connections should be reset when returned to the pool
            (False or None to rollback transcations started with begin(),
            True to always issue a rollback for safety's sake)
        failures: an optional exception class or a tuple of exception classes
            for which the connection failover mechanism shall be applied,
            if the default (OperationalError, InternalError) is not adequate
        ping: determines when the connection should be checked with ping()
            (0 = None = never, 1 = default = whenever fetched from the pool,
            2 = when a cursor is created, 4 = when a query is executed,
            7 = always, and all other bit combinations of these values)
        args, kwargs: the parameters that shall be passed to the creator
            function or the connection constructor of the DB-API 2 module
        érú#Database module is not thread-safe.Nrcsg|]} ¡qSr)Údedicated_connection)Ú.0Úi©Úselfrr Ú
<listcomp>sz%PooledDB.__init__.<locals>.<listcomp>)ÚthreadsafetyÚAttributeErrorÚcallablerrÚ_creatorÚ_argsÚ_kwargsÚ    _blockingÚ    _maxusageÚ_setsessionÚ_resetÚ    _failuresÚ_pingÚ
_maxcachedÚ
_maxsharedÚ _shared_cacheÚ_maxconnectionsÚ_idle_cacherÚ_lockÚ_connectionsÚrangeÚpopÚclose)rZcreatorZ    mincachedZ    maxcachedZ    maxsharedZmaxconnectionsÚblockingZmaxusageZ
setsessionÚresetZfailuresZpingÚargsÚkwargsrZidlerrr Ú__init__¬s^*
 
 
 
 
 
zPooledDB.__init__cCs(t|j|j|j|j|jdf|j|jS)z+Get a steady, unpooled DB-API 2 connection.T)rrr!r"r$r%rrrrrr Ústeady_connectionsþþþzPooledDB.steady_connectioncCs¼|r*|jr*|j ¡zø|js<|jr<|j|jkr<| ¡qt|j|jkrz|j     
d¡}Wntk
rx| ¡}Yn
X|  ¡t|}|jd7_n^|j ¡|j 
d¡}|jjræ|j d|¡| ¡|j ¡|j 
d¡}q°|j  ¡| ¡|j |¡|j ¡W5|j ¡Xt||}n|j ¡zv|jrX|j|jkrX| ¡q6z|j     
d¡}Wntk
r| ¡}Yn
X|  ¡t||}|jd7_W5|j ¡X|S)zÂGet a steady, cached DB-API 2 connection from the pool.
 
        If shareable is set and the underlying DB-API 2 allows it,
        then the connection may be shared with other threads.
        rr)r'r+ÚacquireÚreleaser(r)r,Ú
_wait_lockÚlenr*r.Ú
IndexErrorr5Z_ping_checkÚSharedDBConnectionÚsortÚconÚ_transactionÚinsertÚshareÚappendÚnotifyÚPooledSharedDBConnectionÚPooledDedicatedDBConnection)rZ    shareabler=rrr Ú
connectionsV
 
ÿ
 
 
 
 
 
ÿ
zPooledDB.connectioncCs
| d¡S)z&Alias for connection(shareable=False).F)rErrrr rKszPooledDB.dedicated_connectioncCsh|j ¡z<| ¡|j}|sDz|j |¡Wntk
rBYnXW5|j ¡X|sd| |j    ¡dS)z7Decrease the share of a connection in the shared cache.N)
r+r6r7ÚunshareÚsharedr(ÚremoveÚ
ValueErrorÚcacher=)rr=rGrrr rFOs
 
zPooledDB.unsharecCsr|j ¡zV|jr"t|j|jkr>|j|jd|j |¡n| ¡|j    d8_    |j 
¡W5|j ¡XdS)z4Put a dedicated connection back into the idle cache.)ÚforcerN)r+r6r7r&r9r*r#rAr/r,rB©rr=rrr rJ_s
zPooledDB.cachecCs¬|j ¡z|jrB|j d¡}z| ¡Wqtk
r>YqXq|jr|jr|j d¡j    }z| ¡Wntk
r|YnX|j
d8_
qH|j ¡W5|j ¡XdS)z"Close all connections in the pool.rrN)r+r6r7r*r.r/Ú    Exceptionr'r(r=r,Ú    notifyAllrLrrr r/ns$
zPooledDB.closecCs&z| ¡Wntk
r YnXdS)zDelete the pool.N©r/rMrrrr Ú__del__szPooledDB.__del__cCs|js
t|j ¡dS)z'Wait until notified or report an error.N)r rr+Úwaitrrrr r8szPooledDB._wait_lock)
rrrrFNNTNr)T)rr    r
rrÚversionr4r5rErrFrJr/rPr8rrrr r£s*ü
_
:rc@s0eZdZdZddZddZddZdd    Zd
S)rDz7Auxiliary proxy class for pooled dedicated connections.cCs&d|_| ¡std||_||_dS)zCreate a pooled dedicated connection.
 
        pool: the corresponding PooledDB instance
        con: the underlying SteadyDB connection
        Nr)Ú_conrrÚ_pool)rÚpoolr=rrr r4s
z$PooledDedicatedDBConnection.__init__cCs|jr|j |j¡d|_dS)z&Close the pooled dedicated connection.N)rSrTrJrrrr r/¥sz!PooledDedicatedDBConnection.closecCs|jrt|j|StdS©zProxy all members of the class.N©rSÚgetattrr©rÚnamerrr Ú__getattr__sz'PooledDedicatedDBConnection.__getattr__cCs&z| ¡Wntk
r YnXdS©zDelete the pooled connection.NrOrrrr rP´sz#PooledDedicatedDBConnection.__del__N©rr    r
rr4r/r[rPrrrr rDs
rDc@sXeZdZdZddZddZddZdd    Zd
dZdd Z    ddZ
ddZddZdS)r;z'Auxiliary class for shared connections.cCs||_d|_dS)zUCreate a shared connection.
 
        con: the underlying SteadyDB connection
        rN)r=rGrLrrr r4¿szSharedDBConnection.__init__cCs*|jj|jjkr|j|jkS|jjSdS©N©r=r>rG©rÚotherrrr Ú__lt__ÇszSharedDBConnection.__lt__cCs*|jj|jjkr|j|jkS|jjSdSr^r_r`rrr Ú__le__ÍszSharedDBConnection.__le__cCs|jj|jjko|j|jkSr^r_r`rrr Ú__eq__Ós
ÿzSharedDBConnection.__eq__cCs| |¡Sr^)rdr`rrr Ú__ne__×szSharedDBConnection.__ne__cCs
| |¡Sr^)rbr`rrr Ú__gt__ÚszSharedDBConnection.__gt__cCs
| |¡Sr^)rcr`rrr Ú__ge__ÝszSharedDBConnection.__ge__cCs|jd7_dS)z&Increase the share of this connection.rN©rGrrrr r@àszSharedDBConnection.sharecCs|jd8_dS)z&Decrease the share of this connection.rNrhrrrr rFäszSharedDBConnection.unshareN) rr    r
rr4rbrcrdrerfrgr@rFrrrr r;¼sr;c@s0eZdZdZddZddZddZdd    Zd
S)rCz4Auxiliary proxy class for pooled shared connections.cCs6d|_|j}| ¡dks td||_||_||_dS)zCreate a pooled shared connection.
 
        pool: the corresponding PooledDB instance
        con: the underlying SharedDBConnection
        Nrz'Database connection is not thread-safe.)rSr=rrrTÚ_shared_con)rrUZ
shared_conr=rrr r4ìsz!PooledSharedDBConnection.__init__cCs$|jr |j |j¡d|_|_dS)z#Close the pooled shared connection.N)rSrTrFrirrrr r/üszPooledSharedDBConnection.closecCs|jrt|j|StdSrVrWrYrrr r[sz$PooledSharedDBConnection.__getattr__cCs&z| ¡Wntk
r YnXdSr\rOrrrr rPsz PooledSharedDBConnection.__del__Nr]rrrr rCés
rCN)rÚ    threadingrÚrZ    steady_dbrrMrrrrrrDr;rCrrrr Ú<module>s r(-