tfAdZddlmZddlmZmZGddeZGddZGdd Z Gd d Z d d gZ d S) zg An asynchronous mapping to U{DB-API 2.0}. )threads)logreflectceZdZdZdS)ConnectionLostzb This exception means that a db connection has been lost. Client code may try again. N)__name__ __module__ __qualname____doc__X/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/enterprise/adbapi.pyrrsr rc0eZdZdZdZdZdZdZdZdS) 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. cJ||_d|_|dSN)_pool _connection reconnect)selfpools r__init__zConnection.__init__s'  r cdSrr rs rclosezConnection.close$s  r c6|jjs|jdS |j|j}||jj||jdS#t$rtj ddYnwxYw|j |j|jj rtjdt)NRollback failedzConnection lost.)rrrrollbackcursorexecutegood_sqlrcommit BaseExceptionrerr disconnectnoisymsgr)rcurss rrzConnection.rollback.sz#    % % ' ' ' F -   % % ' ' '#**,,D LL, - - - JJLLL   # # % % % F - - - GD+ , , , , , - d./// :  ( G& ' ' 'sA>B))C  C c|j|j|j|j|_dSr)rrr%connectrs rrzConnection.reconnectDs@   ' J ! !$"2 3 3 3:--//r c,t|j|Sr)getattrrrnames r __getattr__zConnection.__getattr__Ist'...r N) rr r r rrrrr/r r rrrsi    ,000 /////r rc4eZdZdZdZdZdZdZdZdZ 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. NcJ||_||_|dSr)rrreopen)rr connections rrzTransaction.__init__Ys" % r cJ|j}d|_|dSr)_cursorr)rr6s rrzTransaction.close^s!,  r c|j| |j|_dS#t$r%|jjstjddYnwxYw|jj rtj d||j|_dS)NzCursor creation failedzConnection lost, reconnecting) r6rrrr#rrrr$r&r'rs rr3zTransaction.reopencs < # JJLLL 8+2244DL F 8 8 8:' 8677777  8 :  5 G3 4 4 4 '..00 s=,A,+A,cF|jd|_dSr)rrr6rs rrzTransaction.reconnectvs" ""$$$ r c,t|j|Sr)r,r6r-s rr/zTransaction.__getattr__zst|T***r ) rr r r r6rrr3rr/r r rr1r1MspG  111&+++++r r1ceZdZdZdZdZdZdZdZ dZ dZ dZ dZ eZeZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$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_sqlFNzselect 1c||_tj||_t |jdddkrt jdt |jdddkrt jd|d d}|dd lm }||_ ||_ ||_ |j D]%}d |}||vrt||||||=&t|j|j|_t!|j|j|_i|_dd lm}m}|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'}) @keyword cp_min: the minimum number of connections in pool (default 3) @keyword cp_max: the maximum number of connections in pool (default 5) @keyword cp_noisy: generate informational log messages during operation (default C{False}) @keyword 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. @keyword 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. @keyword cp_good_sql: an sql query which should always succeed and change no state (default C{'select 1'}) @keyword cp_reactor: use this reactor instead of the global reactor (added in Twisted 10.2). @type cp_reactor: L{IReactorCore} provider apilevelNz2.0z'DB API module not DB API 2.0 compliant. threadsafetyrz+DB API module not sufficiently thread-safe. cp_reactor)reactorcp_) threadable threadpool) dbapiNamer namedModuledbapir,rr'poptwisted.internetrC_reactorconnargsconnkwCP_ARGSsetattrminmax connectionstwisted.pythonrErF getThreadIDthreadID ThreadPoolcallWhenRunning_startstartID) rrGrMrNrCargcpArgrErFs rrzConnectionPool.__init__sF#(33 4:z4 0 0E 9 9 G= > > > 4:~q 1 1A 5 5 GA B B B**\400 ? 0 0 0 0 0 0    < " "C#KKEc6%=1115Mtx**tx** :9999999". $//$(CC}44T[AA r c8d|_|Sr)rZstartrs rrYzConnectionPool._starts zz||r c|jsH|j|jdd|j|_d|_dSdS)z Start the connection pool. If you are using the reactor normally, this function does *not* need to be called. duringshutdownTN)runningrFr^rLaddSystemEventTrigger finalClose shutdownIDrs rr^zConnectionPool.starts[| O ! ! # # #"mAA*doDO DLLL   r cPtj|j|j|j|g|Ri|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}. )rdeferToThreadPoolrLrF_runWithConnection)rfuncargskws rrunWithConnectionz ConnectionPool.runWithConnectionsC,( M4?D,CT LP   TV   r c||} ||g|Ri|}||S#t$r= |n%#t$rt jddYnwxYwwxYwNr)connectionFactoryr"r#rrr$)rrirjrkconnresults rrhz!ConnectionPool._runWithConnections%%d++ T$,,,,,,F KKMMMM    1   1 1 1/00000 1   s-!9 BABA;8B:A;;BcPtj|j|j|j|g|Ri|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}. )rrgrLrF_runInteraction)r interactionrjrks rrunInteractionzConnectionPool.runInteraction$sJ8( M O            r c.|j|jg|Ri|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}. )ru _runQueryrrjrks rrunQueryzConnectionPool.runQueryIs)$#t"4>?D???B???r c.|j|jg|Ri|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}. )ru _runOperationrxs r runOperationzConnectionPool.runOperation]s*&#t"4#5CCCCCCCr c|jr&|j|jd|_|jr&|j|jd|_|dS)zC Close all pool connections and shutdown the pool. N)rerLremoveSystemEventTriggerrZrdrs rrzConnectionPool.closersl ? # M 2 24? C C C"DO < M 2 24< @ @ @DL r cd|_|jd|_|jD]}|||jdS)zE This should only be called by the shutdown trigger. NF)rerFstoprbrSvalues_closeclearrrps rrdzConnectionPool.finalClose~st  $++--  D KK          r c0|}|j|}|e|jrt jd|j|jj|j i|j }|j | |||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: ) rVrSgetr&rr'rGrIr*rMrNopenfun)rtidrps rr*zConnectionPool.connectsmmoo##C(( <z @>dn>>???%4:%t}D DDD|' T"""$(D S ! r c|}||j|urtd||||j|=dSdS)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)rVrSr Exceptionr)rrprs rr%zConnectionPool.disconnectsmmmoo t'++C00 0 09:: :   KK    %%%  r c|jrtjd|j |dS#t $rtjddYdSwxYw)Nzadbapi closing: zConnection close failed)r&rr'rGrr#r$rs rrzConnectionPool._closest : 9 G7t~77 8 8 8 5 JJLLLLL 5 5 5 GD3 4 4 4 4 4 4 5s;AAcZ||}|||} ||g|Ri|}|||S#t$r= |n%#t$rt jddYnwxYwwxYwrn)rotransactionFactoryrr"r#rrr$)rrtrjrkrptransrqs rrszConnectionPool._runInteractions%%d++''d33  [444444F KKMMM KKMMMM    1   1 1 1/00000 1   s/5A## B*.BB*B%"B*$B%%B*cD|j|i||Sr)r fetchallrrrjrks rrwzConnectionPool._runQuerys) t"r"""~~r c |j|i|dSr)r rs rr{zConnectionPool._runOperations t"r"""""r c h|j|j|j|j|j|j|j|jdS)NrGrQrRr&rr!rMrNrrs r __getstate__zConnectionPool.__getstate__s988Z  k   r cT||_|j|jg|jRi|jdSr)__dict__rrGrMrN)rstates r __setstate__zConnectionPool.__setstate__s7  dnDt}DDD DDDDDr )%rr r r splitrOr&rQrRr.rrr!rbrror1rrerrYr^rlrhruryr|rrdr*r%rrsrwr{rrr r rr;r;~s&>CCEEG E C C DGIHG"$JEBEBEBN       4   # # # J@@@(DDD*    ! ! !. & & &555      ###    EEEEEr r;N) r rKrrTrrrrrr1r;__all__r r rrs  %$$$$$''''''''Y4/4/4/4/4/4/4/4/n.+.+.+.+.+.+.+.+b]E]E]E]E]E]E]E]E@ * +r