h]UdZddlmZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZmZmZdd lmZmZmZdd lmZmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5erddl6m7Z7ddl8m9Z9m:Z:e+dZ;ejxjzZ=ejxj|Z>ej~jZ@ej~jZAejjZCejjZDejjZEejjZGejjZHe3jZJejdZLGdde ZMdeM_NeejgdfZPdeQd<eeMgdfZRdeQd<Gd d!ee!ZSy)"z psycopg connection objects ) annotationsN) TYPE_CHECKINGCallableGeneric NamedTuple) ReferenceTyperef)warn)partial)errors) generatorspostgrespq)PQGen PQGenConnQuery)SQL Composable)Xid)Row) AdaptersMap)IsolationLevel)Deque LiteralStringSelf TypeAliasTypeVar)connection_summary) BasePipeline)PrepareManager) capabilities)ConnectionInfo)BasePool)PGconnPGresult CursorRowpsycopgc4eZdZUdZded< ded< ded<y)Notifyz8An asynchronous notification received from the database.strchannelpayloadintpidN)__name__ __module__ __qualname____doc____annotations__R/var/www/tech_portal/venv/lib/python3.12/site-packages/psycopg/_connection_base.pyr*r*:sB LI L3 HEr6r*r NoticeHandler NotifyHandlerc&eZdZdZej Zej ZejZejZejZ ejZ ejZ ejZ ejZ ejZd9dZd:dZd;dZedd 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). c||_d|_d|_g|_g|_d|_d|_t|_d|_ t|_ t|}ttj||_ttj"||_|d|_|d|_d|_d|_d|_y)NFrr6)pgconn _autocommit _adapters_notice_handlers_notify_handlers_num_transactions_closedr! _prepared_tpcr_notifies_backlogr r r;_notice_handlernotice_handler_notify_handlernotify_handler _pipeline_isolation_level _read_only _deferrable_begin_statement)selfr=wselfs r7__init__zBaseConnection.__init__as  .25757"# )7)9-1 8=wD  '(F(F N '(F(F N .2 7;'+(, #r6czt|dsy|jryt|drytd|dty)Nr=_poolz connection zV was deleted while still open. Please use 'with' or '.close()' to close the connection)hasattrclosedr ResourceWarningrPs r7__del__zBaseConnection.__del__sHtX&  ;;  4 !  $ G G  r6c|jjd|jj}t|j}d|d|dt |ddS)N.< z at 0xx>) __class__r1r2rr=id)rPclsinfos r7__repr__zBaseConnection.__repr__sW**+1T^^-H-H,IJ!$++.3%qfRXaL22r6c<|jjtk(S)z$`!True` if the connection is closed.)r=statusBADrXs r7rVzBaseConnection.closeds{{!!S((r6cZ|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=rfrgrCrXs r7brokenzBaseConnection.brokens%{{!!S(=-==r6c|jS)z'The autocommit state of the connection.)r>rXs r7 autocommitzBaseConnection.autocommitsr6c&|j|yN)_set_autocommitrPvalues r7rkzBaseConnection.autocommit U#r6ctrmNotImplementedErrorros r7rnzBaseConnection._set_autocommit!!r6c#bK|jdEd{t||_y7w)Nrk)_check_intrans_genboolr>ros r7_set_autocommit_genz"BaseConnection._set_autocommit_gens+**<888; 9s /-/c|jS)zX The isolation level of the new transactions started on the connection. )rLrXs r7isolation_levelzBaseConnection.isolation_levels $$$r6c&|j|yrm)_set_isolation_levelros r7r{zBaseConnection.isolation_levels !!%(r6ctrmrsros r7r}z#BaseConnection._set_isolation_levelrur6c#xK|jdEd{| t|nd|_d|_y7 w)Nr{r6)rwrrLrOros r7_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. )rMrXs r7 read_onlyzBaseConnection.read_onlys r6c&|j|yrm)_set_read_onlyros r7rzBaseConnection.read_onlys E"r6ctrmrsros r7rzBaseConnection._set_read_onlyrur6c#xK|jdEd{| t|nd|_d|_y7 w)Nrr6)rwrxrMrOros r7_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. )rNrXs r7 deferrablezBaseConnection.deferrables r6c&|j|yrm)_set_deferrableros r7rzBaseConnection.deferrablerqr6ctrmrsros r7rzBaseConnection._set_deferrablerur6c#xK|jdEd{| t|nd|_d|_y7 w)Nrr6)rwrxrNrOros r7_set_deferrable_genz"BaseConnection._set_deferrable_gens:**<888*/*;4; # 9rc#K|jjx}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_statusIDLErK _sync_genrBeProgrammingErrorrTransactionStatusname)rP attributerfs r7rwz!BaseConnection._check_intrans_genskk44 4F =$..~~//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=rXs r7rczBaseConnection.info sdkk**r6cn|jsttj|_|jSrm)r?rradaptersrXs r7rzBaseConnection.adapterss%~~():):;DN~~r6c|Srmr5rXs r7 connectionzBaseConnection.connections  r6c.|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=socketrXs r7filenozBaseConnection.filenos{{!!!r6cz|jr+|jj}|jyy)z/Cancel the current operation on the connection.N)_should_cancelr= get_cancelcancel)rPcs r7rzBaseConnection.cancel$s.     &&(A HHJ !r6c~|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)rVrErrrXs r7rzBaseConnection._should_cancel*s: ;; 991$$O r6c#K|jj}|jtj||Ed{y7w)Ntimeout)r= cancel_connstartrr)rPrrs r7 _cancel_genzBaseConnection._cancel_gen9s: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@appendrPcallbacks r7add_notice_handlerz!BaseConnection.add_notice_handler> $$X.r6c:|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 r7remove_notice_handlerz$BaseConnection.remove_notice_handlerGrr6c|}|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)rQresrPdiagcbexs r7rGzBaseConnection._notice_handlerPsw.. ||C!6!67'' VB V4 V 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)rArrs r7add_notify_handlerz!BaseConnection.add_notify_handler_rr6c:|jj|y)z Unregister a notification callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.Notify], None] N)rArrs r7remove_notify_handlerz$BaseConnection.remove_notify_handlerhrr6cF|x}sy|jj}t|jj ||j j ||j }|jx}|j||jD] }|| yrm) r=rr*relnamedecodeextrabe_pidrFrrA)rQpgnrPencndrs r7rIzBaseConnection._notify_handlerqs kk## 3;;%%c*CII,<,> --+ C NN ( ( / / 4 NN ' ' . .t 4 D KK " "7 + KK ) )'4} ) U'1'9'9$++'FFK ==J &6==I+E}} +))&4;;;P;PQQ&&(v}})E)J)J(K$W^^%5$8: GsEH  H B>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_CLOSErrrKr r=send_close_preparedrrrrrrfrrrrrrrrr)rPrstmtrrs r7 _deallocatezBaseConnection._deallocates> <,0,<>D(BSD))$/ / /  !!# >>$++994@C NN ( ( / / 4 NN ' ' . .t 4  ''-'// <r=rrr_get_tx_start_commandrKrrXs r7 _start_queryzBaseConnection._start_querysp     ;; ) )T 1 %%d&@&@&BCCC >>~~//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 ) rOr{rrrreplacerrrjoin)rPpartsvals r7rz$BaseConnection._get_tx_start_command!s  (( (     + !5!56C LL+ , LL))#s3::< = >> % LL] K ?? & LL$//?P Q $ % 0$$$r6c#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) rBrrrEr=rrrrKrrXs r7 _commit_genzBaseConnection._commit_gen5s  ! !$$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) rBrrrErKrr=rrrrDclear maintain_genrXs r7 _rollback_genzBaseConnection._rollback_genIs  ! !$$A  99$$J  >>~~//1 1 1 ;; ) )T 1 %%k222 >>..t444 >>~~//1 1 1  2 34 2sHA,D.D/6D%D&rErrrPrs r7_tpc_begin_genzBaseConnection._tpc_begin_genqs #s#//#&C ;; ) )T 1$$(()G)GHMMNP    $$E %L %%d&@&@&BCCCsC(C2*C0+C2c#K|jstjd|jdrtjd|jd}|df|_|jt dj t |Ed{|jr#|jjEd{yy747w)Nz='tpc_prepare()' must be called inside a two-phase transactionr zF'tpc_prepare()' cannot be used during a prepared two-phase transactionrTzPREPARE TRANSACTION {}) rErrrrformatr+rKrrs r7_tpc_prepare_genzBaseConnection._tpc_prepare_gensyy$$O  99Q<$$X iil$K %%c*B&C&J&J3s8&TUUU >>~~//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 {}) lowerrErrrrrgetattrrrrr+)rPactionrfnamemeths r7_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.Nr5rXs r7rzBaseConnection._check_tpcs r6N)r=r%)returnNone)r&r+)r&rx)rprxr&r')rprxr& PQGen[None])r&IsolationLevel | None)rpr)r&r')rpr)r&r()r& bool | None)rpr*r&r')rpr*r&r()rr+r&r()r&r#)r&r)r&zBaseConnection[Row])r&r.)rfloatr&zPQGenConn[None])rr8r&r')rQ"ReferenceType[BaseConnection[Row]]rr&r&r')rr9r&r')rQr,rz pq.PGnotifyr&r')r& int | None)rpr-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()r!rrzXid | str | Noner&r()Br0r1r2r3rWarningErrorr DatabaseError DataErrorrIntegrityError InternalErrorrNotSupportedErrorrRrYrdpropertyrVrirksetterrnryr{r}rrrrrrrrwrcrrrrrrrr staticmethodrGrrrIrr classmethodrrrrrrrr r rrrr$rr5r6r7r;r;MsViiG GGE%%NOOM I))%%NOOM))++%$N &3 ))>>  $$"'%% ))"$  ##"$   $$"$ $++ " C // V1 V8@ V  V V//18C   0 01111,,58-2 :>,,-6, ,\"H   2%(2(26 7D( 2#*: > r6r;)Tr3 __future__rrloggingtypingrrrrweakrefrr warningsr functoolsr r.r rrrrabcrrrsqlrrrErrowsradaptr_enumsr_compatrrrrrpq.miscrrKr _preparingr! _capabilitiesr"_connection_infor#psycopg_pool.baser$pq.abcr%r&r'FormatrBINARY ConnStatusrrgrrrrrrINTRANShas_send_close_preparedr getLoggerrr*r1rr8r4r9r;r5r6r7rSsp # ??&&&(( "CC'#&',*( K  yy~~   ]]mm ]] % % MM # # mm''       & &6,668   9 % FZ F#Q\\ND$89 y9#VHdN3 y3i  WS\i  r6