qj\UdZddlmZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4e rddl5m6Z6m7Z7ddl8m9Z9e*dZ:ejvjxZd Zed?d Zej,d@d Zd@d ZdAdZedBdZej,dCdZdCdZdDdZedBdZej,dCdZdCdZ dDdZ!dEdZ"edFdZ#edGdZ$edHdZ%dIdZ&d:dZ'ddUd5Z? dYd6Z@d:d7ZAy8)ZBaseConnectionz Base class for different types of connections. Share common functionalities such as access to the wrapped PGconn, but allow different interfaces (sync/async). cr||_d|_d|_g|_g|_d|_d|_t|_d|_ t|}ttj||_ttj||_|d|_|d|_d|_d|_d|_y)NFrr5)pgconn _autocommit _adapters_notice_handlers_notify_handlers_num_transactions_closedr _prepared_tpcrr r:_notice_handlernotice_handler_notify_handlernotify_handler _pipeline_isolation_level _read_only _deferrable_begin_statement)selfr<wselfs r6__init__zBaseConnection.__init__bs  .25757"# )7)9-1 D  '(F(F N '(F(F N .2 7;'+(, #r5czt|dsy|jryt|drytd|dty)Nr<_poolz connection zV was deleted while still open. Please use 'with' or '.close()' to close the connection)hasattrclosedr ResourceWarningrNs r6__del__zBaseConnection.__del__sHtX&  ;;  4 !  $ G G  r5c|jjd|jj}t|j}d|d|dt |ddS)N.< z at 0xx>) __class__r0r1rr<id)rNclsinfos r6__repr__zBaseConnection.__repr__sW**+1T^^-H-H,IJ!$++.3%qfRXaL22r5c<|jjtk(S)z$`!True` if the connection is closed.)r<statusBADrVs r6rTzBaseConnection.closeds{{!!S((r5cZ|jjtk(xr |j S)z `!True` if the connection was interrupted. A broken connection is always `closed`, but wasn't closed in a clean way, such as using `close()` or a `!with` block. )r<rdrerBrVs r6brokenzBaseConnection.brokens%{{!!S(=-==r5c|jS)z'The autocommit state of the connection.)r=rVs r6 autocommitzBaseConnection.autocommitsr5c&|j|yN)_set_autocommitrNvalues r6rizBaseConnection.autocommit U#r5ctrkNotImplementedErrorrms r6rlzBaseConnection._set_autocommit!!r5c#bK|jdEd{t||_y7w)Nri)_check_intrans_genboolr=rms r6_set_autocommit_genz"BaseConnection._set_autocommit_gens+**<888; 9s /-/c|jS)zX The isolation level of the new transactions started on the connection. )rJrVs r6isolation_levelzBaseConnection.isolation_levels $$$r5c&|j|yrk)_set_isolation_levelrms r6ryzBaseConnection.isolation_levels !!%(r5ctrkrqrms r6r{z#BaseConnection._set_isolation_levelrsr5c#xK|jdEd{| t|nd|_d|_y7 w)Nryr5)rurrJrMrms r6_set_isolation_level_genz'BaseConnection._set_isolation_level_gens=**+<===9>9Ju 5PT # > :8!:c|jS)zX The read-only state of the new transactions started on the connection. )rKrVs r6 read_onlyzBaseConnection.read_onlys r5c&|j|yrk)_set_read_onlyrms r6rzBaseConnection.read_onlys E"r5ctrkrqrms r6rzBaseConnection._set_read_onlyrsr5c#xK|jdEd{| t|nd|_d|_y7 w)Nrr5)rurvrKrMrms r6_set_read_only_genz!BaseConnection._set_read_only_gens9**;777).):$u+ # 8rc|jS)zY The deferrable state of the new transactions started on the connection. )rLrVs r6 deferrablezBaseConnection.deferrables r5c&|j|yrk)_set_deferrablerms r6rzBaseConnection.deferrableror5ctrkrqrms r6rzBaseConnection._set_deferrablersr5c#xK|jdEd{| t|nd|_d|_y7 w)Nrr5)rurvrLrMrms r6_set_deferrable_genz"BaseConnection._set_deferrable_gens:**<888*/*;4; # 9rc#K|jj}|tk(rD|jr8|jj Ed{|jj}|tk7r]|j rt jd|dt jd|dtj|jy7w)Nz can't change z2 now: connection.transaction() context in progressz' now: connection in transaction status ) r<transaction_statusIDLErI _sync_genrAeProgrammingErrorr TransactionStatusname)rN attributerds r6ruz!BaseConnection._check_intrans_gens// T>dnn~~//1 1 1[[33F T>%%((#I=1CC ((#I=18++F3889;  2sA C C  BCc,t|jS)z>A `ConnectionInfo` attribute to inspect connection properties.)r"r<rVs r6razBaseConnection.infosdkk**r5cn|jsttj|_|jSrk)r>rradaptersrVs r6rzBaseConnection.adapters s%~~():):;DN~~r5c|Srkr4rVs r6 connectionzBaseConnection.connections  r5c.|jjS)zReturn the file descriptor of the connection. This function allows to use the connection as file-like object in functions waiting for readiness, such as the ones defined in the `selectors` module. )r<socketrVs r6filenozBaseConnection.filenos{{!!!r5cz|jr+|jj}|jyy)z/Cancel the current operation on the connection.N)_should_cancelr< get_cancelcancel)rNcs r6rzBaseConnection.cancel"s.     &&(A HHJ !r5c~|jry|jr$|jdrtjdy)zhCheck whether the current command should actually be cancelled when invoking cancel*(). Fr z=cancel() cannot be used with a prepared two-phase transactionT)rTrDrrrVs r6rzBaseConnection._should_cancel(s: ;; 991$$O r5c#K|jj}|jtj||Ed{y7w)Ntimeout)r< cancel_connstartrr)rNrrs r6 _cancel_genzBaseConnection._cancel_gen7s:kk--/ $$['BBBsAAA Ac:|jj|y)z Register a callable to be invoked when a notice message is received. :param callback: the callback to call upon message received. :type callback: Callable[[~psycopg.errors.Diagnostic], None] N)r?appendrNcallbacks r6add_notice_handlerz!BaseConnection.add_notice_handler< $$X.r5c:|jj|y)z Unregister a notice message callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.errors.Diagnostic], None] N)r?removers r6remove_notice_handlerz$BaseConnection.remove_notice_handlerErr5c|}|r |jsytj||jj}|jD] } || y#t $r!}t jd||Yd}~4d}~wwxYw)Nz)error processing notice callback '%s': %s)r?r Diagnosticr< _encoding Exceptionlogger exception)rOresrNdiagcbexs r6rEzBaseConnection._notice_handlerNs{w.. ||C!6!67''B V4( V  !LbRTUU VsA B%BBc:|jj|y)z Register a callable to be invoked whenever a notification is received. :param callback: the callback to call upon notification received. :type callback: Callable[[~psycopg.Notify], None] N)r@rrs r6add_notify_handlerz!BaseConnection.add_notify_handler]rr5c:|jj|y)z Unregister a notification callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.Notify], None] N)r@rrs r6remove_notify_handlerz$BaseConnection.remove_notify_handlerfrr5c |}|r |jsy|jj}t|jj ||j j ||j}|jD] }|| yrk)r@r<rr)relnamedecodeextrabe_pid)rOpgnrNencnrs r6rGzBaseConnection._notify_handlerosow.. kk## 3;;%%c*CII,<,> --+ C NN ( ( / / 4 NN ' ' . .t 4 D KK " "7 + KK ) )'4} ) U'// <H c#&K|ts#|d|znd}|j|Ed{y|j|jrkt |j j |}|jjj||jjjdy|j j |tj|j Ed{d}|jtk7r~|jtk(r+tj ||j j"tj$dt'j(|jj*dy7d7w)ae Deallocate one, or all, prepared statement in the session. ``name == None`` stands for DEALLOCATE ALL. If possible, use protocol-level commands; otherwise use SQL statements. Note that PgBouncer doesn't support DEALLOCATE name, but it supports protocol-level Close from 1.21 and DEALLOCATE ALL from 1.22. Ns DEALLOCATE sDEALLOCATE ALLrrrz0 from sending closing prepared statement message)_HAS_SEND_CLOSErrrIr r<send_close_preparedrrrrrrdrrrrrrr rr)rNrstmtrrs r6 _deallocatezBaseConnection._deallocatesD <,0,<>D(BSD))$/ / /  !!# >> //C NN ( ( / / 4 NN ' ' . .t 4  ''-'// <>~~//1 1 1  D 1s$AB B-B >B?B B c|jr |jSdg}|j_t|j}|jd|j|jj ddj |j|j|jrdnd|j|j|jrdndd j||_|jS) NsBEGINsISOLATION LEVEL_r[s READ ONLYs READ WRITEs DEFERRABLEsNOT DEFERRABLE ) rMryrrrreplacerrrjoin)rNpartsvals r6rz$BaseConnection._get_tx_start_commands  (( (     + !5!56C LL+ , LL))#s3::< = >> % LL] K ?? & LL$//?P Q $ % 0$$$r5c#dK|jrtjd|jrtjd|jj t k(ry|jdEd{|jr#|jjEd{yy747w)z-Generator implementing `Connection.commit()`.zExplicit commit() forbidden within a Transaction context. (Transaction will be automatically committed on successful exit from context.)z6commit() cannot be used during a two-phase transactionNsCOMMIT) rArrrDr<rrrrIrrVs r6 _commit_genzBaseConnection._commit_gen2s  ! !$$4  99$$H  ;; ) )T 1 %%i000 >>~~//1 1 1  1 2s$A5B07B,8-B0%B.&B0.B0c#BK|jrtjd|jrtjd|jr"|jj Ed{|j jtk(ry|jdEd{|jj|jj|Ed{|jr#|jj Ed{yy77s787 w)z/Generator implementing `Connection.rollback()`.zExplicit rollback() forbidden within a Transaction context. (Either raise Rollback() or allow an exception to propagate out of the context.)z8rollback() cannot be used during a two-phase transactionNsROLLBACK) rArrrDrIrr<rrrrCclear maintain_genrVs r6 _rollback_genzBaseConnection._rollback_genFs  ! !$$A  99$$J  >>~~//1 1 1 ;; ) )T 1 %%k222 >>..t444 >>~~//1 1 1  2 34 2sHA,D.D/6D%D&>~~//1 1 1  V 1s$BCC -CC C Cc#pKd|jd}|4|jstj|d|jd}nI|jrtj|dt |t st j |}|jrD|jds5t|d|jd}d|_|Ed{y|jtd jt|t|Ed{d|_y7Q7w) Ntpc_z()z? without xid must must be called inside a two-phase transactionrz= with xid must must be called outside a two-phase transactionr r_genz{} PREPARED {}) lowerrDrrrrrgetattrrrrr*)rNactionrfnamemeths r6_tpc_finish_genzBaseConnection._tpc_finish_gens&v||~&b) ;99((g==))A,Cyy((g77c3'ooc* 99TYYq\41V\\^$4D!9:DDIv  ))$%,,S[#c(C  DI  s%CD6 D2!AD6%D4& D64D6cy)z0Raise NotSupportedError if TPC is not supported.Nr4rVs r6r zBaseConnection._check_tpcs r5N)r<r#)returnNone)r#r*)r#rv)rnrvr#r$)rnrvr# PQGen[None])r#IsolationLevel | None)rnr&r#r$)rnr&r#r%)r# bool | None)rnr'r#r$)rnr'r#r%)rr*r#r%)r#r")r#r)r#zBaseConnection[Row])r#r-)rfloatr#zPQGenConn[None])rr7r#r$)rO"ReferenceType[BaseConnection[Row]]rr$r#r$)rr8r#r$)rOr)rz pq.PGnotifyr#r$)r# int | None)rnr*r#r$))rr*rr(r#zPQGenConn[Self])rrrz pq.Formatr#zPQGen[PGresult | None])rz bytes | Noner#r%)r#r%)r#bytes)rr-rr*rr*r#r)rz Xid | strr#r%)rrrzXid | str | Noner#r%)Br/r0r1r2rWarningErrorr DatabaseError DataErrorrIntegrityError InternalErrorrNotSupportedErrorrPrWrbpropertyrTrgrisetterrlrwryr{r~rrrrrrrurarrrrrrrr staticmethodrErrrGrr classmethodrrrrrrrrr rrrr!r r4r5r6r:r:NsViiG GGE%%NOOM I))%%NOOM))++!$F &3 ))>>  $$"'%% ))"$  ##"$   $$"$ &++ " C // V1 V8@ V  V V// 1 8C     0 01111,,58-2 :>,,-6, ,\%N   2%(2(26 7D( 2#*: > r5r:)Sr2 __future__rrloggingtypingrrrrweakrefrr warningsr functoolsr r+r rrrrabcrrrsqlrrrDrrowsradaptr_enumsr_compatrrrrpq.miscrrIr _preparingr _capabilitiesr!_connection_infor"pq.abcr#r$psycopg_pool.baser%r&FormatrBINARY ConnStatusrrerrrrrrINTRANShas_send_close_preparedr getLoggerrr)r0rr7r3r8r:r4r5r6rPss # ??&(( "<<'#&',(* K  yy~~   ]]mm ]] % % MM # # mm''       & &6,668   9 % FZ F#Q\\ND$89 y9#VHdN3 y3e  WS\e  r5