j]UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZdd lmZdd lmZmZmZdd lmZmZmZdd lmZmZdd lm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6erddl7m8Z8ddl9m:Z:m;Z;e,dZejzj~Z?ejjZAejjZBejjZDejjZEejjZFejjZHejjZIe2jZKejdZMGdde ZNdeN_OeejgdfZQdeRd<eeNgdfZSdeRd<Gd d!e e"ZTy)"z psycopg connection objects ) annotationsN) TYPE_CHECKINGAnyCallableGeneric NamedTuple) ReferenceTyperef)warn)partial)errors) generatorspostgrespq)PQGen PQGenConnQuery)SQL Composable)Xid)Row) AdaptersMap)IsolationLevel)Deque LiteralStringSelf TypeAliasTypeVar)connection_summary)PrepareManager) capabilities) BasePipeline)ConnectionInfo)BasePool)PGconnPGresult CursorRowpsycopgc4eZdZUdZded< ded< ded<y)Notifyz8An asynchronous notification received from the database.strchannelpayloadintpidN)__name__ __module__ __qualname____doc____annotations___/var/www/html/kelly/kelly-backend/venv/lib/python3.12/site-packages/psycopg/_connection_base.pyr+r+:sB LI L3 HEr7r+r NoticeHandler NotifyHandlerc*eZdZdZej Zej ZejZejZejZ ejZ ejZ ejZ ejZ ejZd9dZefd:dZd;dZedd Zed?d Zej.d@d Zd@d ZdAdZedBdZej.dCdZdCdZdDdZedBdZ e j.dCdZ dCdZ!dDdZ"dEdZ#edFdZ$edGdZ%edHdZ&dIdZ'dJdZ(ddYd4Z?dVd5Z@ dZd6ZAdJd7ZBy8)[BaseConnectionz 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)NFrr7)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 r8__init__zBaseConnection.__init__as  .25757"# )7)9-1 8=wD  '(F(F N '(F(F N .2 7;'+(, #r7ct|dsy|jryt|dry|tj|dty)Nr>_poolz_ was deleted while still open. Please use 'with' or '.close()' to close the connection properly)hasattrclosedobject__repr__ResourceWarning)rQ_BaseConnection__warns r8__del__zBaseConnection.__del__sPtX&  ;;  4 ! t$%&P P  r7c|jjd|jj}t|j}d|d|dt |ddS)N.< z at 0xx>) __class__r2r3r r>id)rQclsinfos r8rYzBaseConnection.__repr__sW**+1T^^-H-H,IJ!$++.3%qfRXaL22r7c<|jjtk(S)z$`!True` if the connection is closed.)r>statusBADrQs r8rWzBaseConnection.closeds{{!!S((r7cZ|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>rhrirDrjs r8brokenzBaseConnection.brokens%{{!!S(=-==r7c|jS)z'The autocommit state of the connection.)r?rjs r8 autocommitzBaseConnection.autocommitsr7c&|j|yN)_set_autocommitrQvalues r8rnzBaseConnection.autocommit U#r7ctrpNotImplementedErrorrrs r8rqzBaseConnection._set_autocommit!!r7c#bK|jdEd{t||_y7w)Nrn)_check_intrans_genboolr?rrs r8_set_autocommit_genz"BaseConnection._set_autocommit_gens+**<888; 9s /-/c|jS)zX The isolation level of the new transactions started on the connection. )rMrjs r8isolation_levelzBaseConnection.isolation_levels $$$r7c&|j|yrp)_set_isolation_levelrrs r8r~zBaseConnection.isolation_levels !!%(r7ctrprvrrs r8rz#BaseConnection._set_isolation_levelrxr7c#xK|jdEd{| t|nd|_d|_y7 w)Nr~r7)rzrrMrPrrs r8_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. )rNrjs r8 read_onlyzBaseConnection.read_onlys r7c&|j|yrp)_set_read_onlyrrs r8rzBaseConnection.read_onlys E"r7ctrprvrrs r8rzBaseConnection._set_read_onlyrxr7c#xK|jdEd{| t|nd|_d|_y7 w)Nrr7)rzr{rNrPrrs r8_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. )rOrjs r8 deferrablezBaseConnection.deferrables r7c&|j|yrp)_set_deferrablerrs r8rzBaseConnection.deferrablertr7ctrprvrrs r8rzBaseConnection._set_deferrablerxr7c#xK|jdEd{| t|nd|_d|_y7 w)Nrr7)rzr{rOrPrrs r8_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_statusIDLErL _sync_genrCeProgrammingErrorrTransactionStatusname)rQ attributerhs r8rzz!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>rjs r8rfzBaseConnection.info sdkk**r7cn|jsttj|_|jSrp)r@rradaptersrjs r8rzBaseConnection.adapterss%~~():):;DN~~r7c|Srpr6rjs r8 connectionzBaseConnection.connections  r7c.|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>socketrjs r8filenozBaseConnection.filenos{{!!!r7cz|jr+|jj}|jyy)z/Cancel the current operation on the connection.N)_should_cancelr> get_cancelcancel)rQcs r8rzBaseConnection.cancel$s.     &&(A HHJ !r7c~|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)rWrFrrrjs r8rzBaseConnection._should_cancel*s: ;; 991$$O r7c#K|jj}|jtj||Ed{y7w)Ntimeout)r> cancel_connstartrr)rQrrs r8 _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)rAappendrQcallbacks r8add_notice_handlerz!BaseConnection.add_notice_handler> $$X.r7c:|jj|y)z Unregister a notice message callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.errors.Diagnostic], None] N)rAremovers r8remove_notice_handlerz$BaseConnection.remove_notice_handlerGrr7c|}|r |jsytj||jj}|jD] } || y#t $r!}t jd||Yd}~4d}~wwxYw)Nz)error processing notice callback '%s': %s)rAr Diagnosticr> _encoding Exceptionlogger exception)rRresrQdiagcbexs r8rHzBaseConnection._notice_handlerPs{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)rBrrs r8add_notify_handlerz!BaseConnection.add_notify_handler_rr7c:|jj|y)z Unregister a notification callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.Notify], None] N)rBrrs r8remove_notify_handlerz$BaseConnection.remove_notify_handlerhrr7cb|x}sy|jj}t|jj ||j j ||j }|jr|jD] }|| y|jx}|j|yyrp) r>rr+relnamedecodeextrabe_pidrBrGr)rRpgnrQencnrds r8rJzBaseConnection._notify_handlerqs kk## 3;;%%c*CII,<,conns r8 _connect_genzBaseConnection._connect_gens1 '..xII6{ Js .,.c#K|jt|tr&|j|jj }n!t|t r|j|}|jrnt|jj|d|}|jjj||jjjdy|tk(r|jj|n|jj|d|t!j"|jEd{d}|j$t&k7r|j$t(k7r|j$t*k(r+t-j.||jj t-j0dt3j4|j$j6d|j9|S7w)z Generator to send a command and receive the result to the backend. Only used to implement internal commands such as "commit", with eventual arguments bound client-side. The cursor can do more complex stuff. N) result_formatencodingunexpected result z from command )_check_connection_ok isinstancer,encoder>rras_bytesrLr send_query_params command_queuer result_queueTEXT send_queryrexecuterh COMMAND_OK TUPLES_OK FATAL_ERRORrerror_from_resultInterfaceErrorr ExecStatusrr)rQcommandrcmdresults r8 _exec_commandzBaseConnection._exec_commands !!# gs #nnT[[%:%:;G  ,&&t,G >> --+ 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_CLOSErrrLr r>send_close_preparedrrrrrrhrrrrrrrrr)rQrstmtrrs r8 _deallocatezBaseConnection._deallocates> <,0,<>D(BSD))$/ / /  !!# >>$++994@C NN ( ( / / 4 NN ' ' . .t 4  ''-'// <rhOKrirOperationalErrorrrjs r8rz#BaseConnection._check_connection_ok sh ;;   #  ;;   $$$%?@ @ ++,,- /  r7c#K|jry|jjtk7ry|j |j Ed{|j r#|j jEd{yy747w)z.Generator to start a transaction if necessary.N)r?r>rrr_get_tx_start_commandrLrrjs r8 _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 ) rPr~rrrreplacerrrjoin)rQpartsvals r8rz$BaseConnection._get_tx_start_command"s  (( (     + !5!56C LL+ , LL))#s3::< = >> % LL] K ?? & LL$//?P Q $ % 0$$$r7c#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) rCrrrFr>rrrrLrrjs r8 _commit_genzBaseConnection._commit_gen6s  ! !$$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) rCrrrFrLrr>rrrrEclear maintain_genrjs r8 _rollback_genzBaseConnection._rollback_genJs  ! !$$A  99$$J  >>~~//1 1 1 ;; ) )T 1 %%k222 >>..t444 >>~~//1 1 1  2 34 2sHA,D.D/6D%D&rrrrrrrr?rFrrrQrs r8_tpc_begin_genzBaseConnection._tpc_begin_genrs #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 {}) rFrrrrformatr,rLrrs r8_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 {}) lowerrFrrrrrgetattrrrrr,)rQactionrfnamemeths r8_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.Nr6rjs r8rzBaseConnection._check_tpcs r7N)r>r&)r[rreturnNone)r)r,)r)r{)rsr{r)r*)rsr{r) PQGen[None])r)IsolationLevel | None)rsr,r)r*)rsr,r)r+)r) bool | None)rsr-r)r*)rsr-r)r+)rr,r)r+)r)r$)r)r)r)zBaseConnection[Row])r)r/)r)r*)rfloatr)zPQGenConn[None])rr9r)r*)rR"ReferenceType[BaseConnection[Row]]rr'r)r*)rr:r)r*)rRr/rz pq.PGnotifyr)r*)r) int | None)rsr0r)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+)Cr1r2r3r4rWarningErrorr DatabaseError DataErrorrIntegrityError InternalErrorrNotSupportedErrorrSr r\rYpropertyrWrlrnsetterrqr|r~rrrrrrrrrzrfrrrrrrrr staticmethodrHrrrJrr classmethodrrrrrrrr rrrrr'rr6r7r8r<r<Ms[iiG GGE%%NOOM I))%%NOOM))++%$N%) &3 ))>>  $$"'%% ))"$  ##"$   $$"$ $++ " C // V1 V8@ V  V V//18C " 0 01111,,58-2 :>,,-6, ,\"H   2%(2(26 7D( 2#*: > r7r<)Ur4 __future__rrloggingtypingrrrrrweakrefr r warningsr functoolsr r1rrrrrabcrrrsqlrrrFrrowsradaptr_enumsr_compatrrrrrpq.miscr _preparingr! _capabilitiesr"_pipeline_baser#_connection_infor$psycopg_pool.baser%pq.abcr&r'r(FormatrBINARY ConnStatusrrirrrrrrINTRANShas_send_close_preparedr getLoggerrr+r2rr9r5r:r<r6r7r8rWsp # DD&&&(( "CC'&'(,*( K  yy~~   ]]mm ]] % % MM # # mm''       & &6,668   9 % FZ F#Q\\ND$89 y9#VHdN3 y3j  WS\j  r7