YtjB^0UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZd d lmZd d lmZmZmZd d lmZmZmZd dl m!Z!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,m-Z-m.Z.d dl/m0Z0d dl1m2Z2d dl3m4Z4d dl5m6Z6d dl7m8Z8erddl9m:Z:d dl;mej?j@Z@ej?jAZAejBjCZCejBjDZDejEjFZFejEjGZGejEjHZHejIjJZJejIjKZKe4jLZMejNdZOGdde ZPdeP_QeejRgdfZSdeTd <eePgdfZUdeTd!<Gd"d#ee&ZVdS)$z psycopg connection objects ) annotationsN) TYPE_CHECKINGAnyGeneric NamedTuple TypeAlias) ReferenceTyperef)warn)partial)deque)Callable)errors) generatorspostgrespq)PQGen PQGenConnQueryNoTemplate)SQL Composable)Xid)Row) AdaptersMap)IsolationLevel) LiteralStringSelfTypeVar)connection_summary)PrepareManager) capabilities) BasePipeline)ConnectionInfo)BasePool)PGconnPGresult CursorRowpsycopgc6eZdZUdZded< ded< ded<dS)Notifyz8An asynchronous notification received from the database.strchannelpayloadintpidN)__name__ __module__ __qualname____doc____annotations__V/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg/_connection_base.pyr+r+<s9BBLLLILLL3 HHHEEr7r+r NoticeHandler NotifyHandlerceZdZdZejZejZejZejZej Z ej Z ej Z ej Z ej Z ejZdkdZefdld Zdmd Zednd ZedndZedndZejdodZdodZdpdZedqdZejdrdZdrdZdsdZedtdZejdudZdudZdvdZedtdZ e jdud Z dud!Z!dvd"Z"dwd$Z#edxd&Z$edyd(Z%edzd*Z&d{d,Z'd|d-Z(dnd.Z)d}d2Z*d~d5Z+d~d6Z,e-dd;Z.dd=Z/dd>Z0e-ddAZ1eddCZ2e2jddDZ2eddEZ3e3jddFZ3e4 ddHdIddLZ5e6fddRZ7ddUZ8d|dVZ9ddWZ:ddYZ;ddZZddcZ?dddZ@ddhZAd|diZBdjS)BaseConnectionz Base class for different types of connections. Share common functionalities such as access to the wrapped PGconn, but allow different interfaces (sync/async). pgconnr&c||_d|_d|_g|_g|_d|_d|_t|_d|_ t|_ t|}ttj||_ttj||_|d|_||d|_d|_d|_d|_dS)NFrr7)r= _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__cs  .25757"# )7)9)9-1 8=wwD  '(F N N '(F N N #.2  7;'+(, #r7_BaseConnection__warnrreturnNonect|dsdS|jrdSt|drdS|t|dtdS)Nr=_poolz_ was deleted while still open. Please use 'with' or '.close()' to close the connection properly)hasattrclosedobject__repr__ResourceWarning)rQrTs r8__del__zBaseConnection.__del__stX&&  F ;  F 4 ! !  Ft$$ P P P      r7r,c|jjd|jj}t|j}d|d|dt |ddS)N.< z at 0xx>) __class__r2r3r r=id)rQclsinfos r8r\zBaseConnection.__repr__sX*JJT^-HJJ!$+..232222RXX22222r7boolc,|jjtkS)z$`!True` if the connection is closed.)r=statusBADrQs r8rZzBaseConnection.closeds{!S((r7c<|jjtko|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=rkrlrDrms r8brokenzBaseConnection.brokens{!S(=-==r7c|jS)z'The autocommit state of the connection.)r?rms r8 autocommitzBaseConnection.autocommits r7valuec0||dSN)_set_autocommitrQrrs r8rqzBaseConnection.autocommit U#####r7ctrtNotImplementedErrorrvs r8ruzBaseConnection._set_autocommit!!r7 PQGen[None]c#hK|dEd{Vt||_dS)Nrq)_check_intrans_genrir?rvs r8_set_autocommit_genz"BaseConnection._set_autocommit_gens@**<888888888;;r7IsolationLevel | Nonec|jS)zX The isolation level of the new transactions started on the connection. )rMrms r8isolation_levelzBaseConnection.isolation_levels $$r7c0||dSrt)_set_isolation_levelrvs r8rzBaseConnection.isolation_levels !!%(((((r7ctrtryrvs r8rz#BaseConnection._set_isolation_levelr{r7c#~K|dEd{V|t|nd|_d|_dS)Nrr7)r~rrMrPrvs r8_set_isolation_level_genz'BaseConnection._set_isolation_level_gensV**+<=========9>9Ju 5 5 5PT #r7 bool | Nonec|jS)zX The read-only state of the new transactions started on the connection. )rNrms r8 read_onlyzBaseConnection.read_onlys r7c0||dSrt)_set_read_onlyrvs r8rzBaseConnection.read_onlys E"""""r7ctrtryrvs r8rzBaseConnection._set_read_onlyr{r7c#~K|dEd{V|t|nd|_d|_dS)Nrr7)r~rirNrPrvs r8_set_read_only_genz!BaseConnection._set_read_only_gensP**;777777777).):$u+++ #r7c|jS)zY The deferrable state of the new transactions started on the connection. )rOrms r8 deferrablezBaseConnection.deferrables r7c0||dSrt)_set_deferrablervs r8rzBaseConnection.deferrablerwr7ctrtryrvs r8rzBaseConnection._set_deferrabler{r7c#~K|dEd{V|t|nd|_d|_dS)Nrr7)r~rirOrPrvs r8_set_deferrable_genz"BaseConnection._set_deferrable_gensQ**<888888888*/*;4;;; #r7 attributec#RK|jjx}tkr2|jr+|jEd{V|jj}|tkrP|jrt jd|dt jd|dtj |j dS)Nz can't change z2 now: connection.transaction() context in progressz' now: connection in transaction status ) r=transaction_statusIDLErL _sync_genrCeProgrammingErrorrTransactionStatusname)rQrrks r8r~z!BaseConnection._check_intrans_gensk4 4F = =$. =~//11 1 1 1 1 1 1 1[3F T>>% (CICCC (;I;;+F338;; >r7r$c*t|jS)z>A `ConnectionInfo` attribute to inspect connection properties.)r$r=rms r8rhzBaseConnection.infosdk***r7rcZ|jsttj|_|jSrt)r@rradaptersrms r8rzBaseConnection.adapterss&~ <():;;DN~r7BaseConnection[Row]c|Srtr6rms r8 connectionzBaseConnection.connections  r7r/c|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=socketrms r8filenozBaseConnection.filenos{!!r7c|r/|j}|dSdS)z/Cancel the current operation on the connection.N)_should_cancelr= get_cancelcancel)rQcs r8rzBaseConnection.cancel(sD      &&((A HHJJJJJ  r7ch|jrdS|jr!|jdrtjddS)zhCheck whether the current command should actually be cancelled when invoking cancel*(). Frz=cancel() cannot be used with a prepared two-phase transactionT)rZrFrrrms r8rzBaseConnection._should_cancel.sI ; 5 9 1 $O tr7timeoutfloatPQGenConn[None]c#K|j}|tj||Ed{VdS)Nr)r= cancel_connstartrr)rQrrs r8 _cancel_genzBaseConnection._cancel_gen=sYk--// $['BBBBBBBBBBBBr7callbackr9c:|j|dS)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)rAappendrQrs r8add_notice_handlerz!BaseConnection.add_notice_handlerB! $$X.....r7c:|j|dS)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_handlerKrr7rR"ReferenceType[BaseConnection[Row]]resr'c|}|r|jsdStj||jj}|jD]A} ||#t $r&}t d||Yd}~:d}~wwxYwdS)Nz)error processing notice callback '%s': %s)rAr Diagnosticr= _encoding Exceptionlogger exception)rRrrQdiagcbexs r8rHzBaseConnection._notice_handlerTsuww .  F|C!677' V VB V4 V V V  !LbRTUUUUUUUU V V Vs A  A;A66A;r:c:|j|dS)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_handlercrr7c:|j|dS)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_handlerlrr7pgn pq.PGnotifycB|x}sdS|jj}t|j||j||j}|jr|jD] }||dS|jx}| |dSdSrt) r=rr+relnamedecodeextrabe_pidrBrGr)rRrrQencnrds r8rJzBaseConnection._notify_handlerus  Fk# 3;%%c**CI,<,  -+ C N ( / / 4 4 4 N ' . .t 4 4 44 D K " "7 + + + + K ) )'4} ) U U U'1'9$+'F'FFFFFFFK =J & &6=I+E+E} ++)&4;;PQQQQ&:v})E)E)J::%,^^%5%5:: r7r bytes | Nonec#K|ts&|d|znd}||Ed{VdS||jrZt |jj|}|jj||jj ddS|j|tj |jEd{Vd}|j tkrd|j tkr tj||jjtjdt'j|j jddS)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_preparedrrrrrrkrrrrrrrrr)rQrstmtrrs r8 _deallocatezBaseConnection._deallocatesr <<,0,<>D((BSD))$// / / / / / / / F !!### > $+94@@C N ( / / 4 4 4 N ' . .t 4 4 4 F ''---'/ <<<<<<< 2~//11 1 1 1 1 1 1 1 1 1 2 2r7bytesc|jr|jSdg}|jit|j}|d||jdd|j||jrdnd|j||jrdndd ||_|jS) NsBEGINsISOLATION LEVEL_rbs READ ONLYs READ WRITEs DEFERRABLEsNOT DEFERRABLE ) rPrrrrreplacerrrjoin)rQpartsvals r8rz$BaseConnection._get_tx_start_command&s   )( (    + !566C LL+ , , , LL))#s33::<< = = = > % LLJ] K K K ? & LL$/P?P Q Q Q $ % 0 0$$r7c#*K|jrtjd|jrtjd|jjt krdS|dEd{V|jr!|j Ed{VdSdS)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=rrrrLrrms r8 _commit_genzBaseConnection._commit_gen:s  ! $4  9 $H  ; )T 1 1 F%%i000000000 > 2~//11 1 1 1 1 1 1 1 1 1 2 2r7c#K|jrtjd|jrtjd|jr|jEd{V|jjtkrdS| dEd{V|j |j |Ed{V|jr!|jEd{VdSdS)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_genrms r8 _rollback_genzBaseConnection._rollback_genNs9  ! $A  9 $J  > 2~//11 1 1 1 1 1 1 1 ; )T 1 1 F%%k222222222 >..t444444444 > 2~//11 1 1 1 1 1 1 1 1 1 2 2r7 format_idgtridbqualrcV|tj|||S)aL Returns a `Xid` to pass to the `!tpc_*()` methods of this connection. The argument types and constraints are explained in :ref:`two-phase-commit`. The values passed to the method will be available on the returned object as the members `~Xid.format_id`, `~Xid.gtrid`, `~Xid.bqual`. ) _check_tpcr from_parts)rQr!r"r#s r8xidzBaseConnection.xidis( ~i666r7r' Xid | strc#K|t|tstj|}|jjt kr8tjdtj |jjj |j rtjd|df|_ ||Ed{VdS)Nz8can't start two-phase transaction: connection in status z3can't use two-phase transactions in autocommit modeF)r%rr from_stringr=rrrrrrrr?rFrrrQr's r8_tpc_begin_genzBaseConnection._tpc_begin_genvs #s## '/#&&C ; )T 1 1$P()GHHMPP    $E %L %%d&@&@&B&BCCCCCCCCCCCr7c#K|jstjd|jdrtjd|jd}|df|_|t dt |Ed{V|jr!|jEd{VdSdS)Nz='tpc_prepare()' must be called inside a two-phase transactionrzF'tpc_prepare()' cannot be used during a prepared two-phase transactionrTzPREPARE TRANSACTION {}) rFrrrrformatr,rLrr+s r8_tpc_prepare_genzBaseConnection._tpc_prepare_gensy $O  9Q< $X il$K %%c*B&C&C&J&J3s88&T&TUUUUUUUUU > 2~//11 1 1 1 1 1 1 1 1 1 2 2r7actionrXid | str | Nonec#Kd|d}|,|jstj|d|jd}nG|jrtj|dt |t st j|}|jrL|jds?t|d|d}d|_|Ed{VdS|td  t|t|Ed{Vd|_dS) Ntpc_z()z? without xid must must be called inside a two-phase transactionrz= with xid must must be called outside a two-phase transactionrr_genz{} PREPARED {}) lowerrFrrrrr*getattrrrr.r,)rQr0r'fnamemeths r8_tpc_finish_genzBaseConnection._tpc_finish_gens*v||~~))) ;9 (===)A,CCy (777c3'' +oc** 9 TYq\ 4!9V\\^^!9!9!9::DDItvv         ))$%%,,S[[#c((CC       DIIIr7cdS)z0Raise NotSupportedError if TPC is not supported.Nr6rms r8r%zBaseConnection._check_tpcs  r7N)r=r&)rTrrUrV)rUr,)rUri)rrrirUrV)rrrirUr|)rUr)rrrrUrV)rrrrUr|)rUr)rrrrUrV)rrrrUr|)rr,rUr|)rUr$)rUr)rUr)rUr/)rUrV)rrrUr)rr9rUrV)rRrrr'rUrV)rr:rUrV)rRrrrrUrV)rUr)rrrrUrV)r)rr,rrrUr)rrrrrUr)rrrUr|)rUr|)rUr)r!r/r"r,r#r,rUr)r'r(rUr|)r0rr'r1rUr|)Cr1r2r3r4rWarningErrorr DatabaseError DataErrorrIntegrityError InternalErrorrNotSupportedErrorrSr r^r\propertyrZrorqsetterrurrrrrrrrrrr~rhrrrrrrrr staticmethodrHrrrJrr classmethodrrrr rrrrr r'r,r/r9r%r6r7r8r<r<OsiG GE%NOM I)%NOM)+'$'$'$'$R%)     &3333 )))X)>>>X>   X $$$$""""''''%%%X% ))))""""$$$$ X ####""""$$$$    X $$$$""""$$$$ $+++X+X X""""    CCCC //////// V V V\ V////////\" 0 0 0X 01111111X1,,,,58[DH,,,,,\""""H      2 2 2 2%%%%(2222(22226 7 7 7 7DDDD( 2 2 2 2>      r7r<)Wr4 __future__rrloggingtypingrrrrrweakrefr r warningsr functoolsr collectionsr collections.abcrrrrrrrabcrrrsqlrrrFrrowsradaptr_enumsr_compatrrrpq.miscr _preparingr! _capabilitiesr"_pipeline_baser#_connection_infor$psycopg_pool.baser%pq.abcr&r'r(FormatrBINARY ConnStatusrrlrrrrrrINTRANShas_send_close_preparedr  getLoggerrr+r2rr9r5r:r<r6r7r8rasc #""""" EEEEEEEEEEEEEE&&&&&&&&$$$$$$&&&&&&&&&&2222222222 """"""1111111111''''''&&&&&&''''''((((((,,,,,,)******(((((((( GK  y~  ]m ] % M # m'     &6,688  9 % % F F F F FZ F F F#Q\ND$89 9999#VHdN3 3333l  l  l  l  l  WS\l  l  l  l  l  r7