pj\(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:ej;j<Zj?Z?ej>j@Z@ejAjBZBejAjCZCejAjDZDejEjFZFejEjGZGe2jHZIejJdZKGddeZLdeL_MeejNgdfZOdePd <eeLgdfZQdePd!<Gd"d#ee!ZRdS)$z psycopg connection objects ) annotationsN)CallableGeneric NamedTuple TYPE_CHECKING)ref ReferenceType)warn)partial)pq)errors)postgres) generators)PQGen PQGenConnQuery) ComposableSQL)Xid)Row) AdaptersMap)IsolationLevel) LiteralStringSelf TypeAliasTypeVar)connection_summary) BasePipeline)PrepareManager) capabilities)ConnectionInfo)PGconnPGresult)BasePool CursorRowpsycopgc6eZdZUdZded< ded< ded<dS)Notifyz8An asynchronous notification received from the database.strchannelpayloadintpidN)__name__ __module__ __qualname____doc____annotations__h/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/psycopg/_connection_base.pyr)r);s9BBLLLILLL3 HHHEEr5r)r NoticeHandler NotifyHandlerceZdZdZejZejZejZejZej Z ej Z ej Z ej Z ej Z ejZdidZdjdZdkd Zedld Zedld Zedld ZejdmdZdmdZdndZedodZejdpdZdpdZdqdZedrdZejdsdZdsdZdtdZedrdZejdsdZdsdZ dtd Z!dud"Z"edvd$Z#edwd&Z$edxd(Z%dyd*Z&djd+Z'dld,Z(dzd0Z)d{d3Z*d{d4Z+e,d|d9Z-d}d;Z.d}d<Z/e,d~d?Z0eddAZ1e1jddBZ1eddCZ2e2jddDZ2e3 ddFdGddJZ4e5fddPZ6ddSZ7djdTZ8ddUZ9ddWZ:ddXZ;ddYZddbZ?ddfZ@djdgZAdhS)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|}ttj ||_ttj||_|d|_|d|_d|_d|_d|_dS)NFrr5)r; _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)9-1 D  '(F N N '(F N N #.2 7;'+(, #r5returnNonect|dsdS|jrdSt|drdStd|dtdS)Nr;_poolz connection zV was deleted while still open. Please use 'with' or '.close()' to close the connection)hasattrclosedr ResourceWarningrNs r6__del__zBaseConnection.__del__s|tX&&  F ;  F 4 ! !  F  G$ G G G      r5r*c|jjd|jj}t|j}d|d|dt |ddS)N.< z at 0xx>) __class__r0r1rr;id)rNclsinfos r6__repr__zBaseConnection.__repr__sX*JJT^-HJJ!$+..232222RXX22222r5boolc,|jjtkS)z$`!True` if the connection is closed.)r;statusBADrXs r6rVzBaseConnection.closeds{!S((r5c<|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;rgrhrBrXs r6brokenzBaseConnection.brokens{!S(=-==r5c|jS)z'The autocommit state of the connection.)r=rXs r6 autocommitzBaseConnection.autocommits r5valuec0||dSN)_set_autocommitrNrms r6rlzBaseConnection.autocommit U#####r5ctroNotImplementedErrorrqs r6rpzBaseConnection._set_autocommit!!r5 PQGen[None]c#hK|dEd{Vt||_dS)Nrl)_check_intrans_genrer=rqs r6_set_autocommit_genz"BaseConnection._set_autocommit_gens@**<888888888;;r5IsolationLevel | Nonec|jS)zX The isolation level of the new transactions started on the connection. )rJrXs r6isolation_levelzBaseConnection.isolation_levels $$r5c0||dSro)_set_isolation_levelrqs r6r}zBaseConnection.isolation_levels !!%(((((r5ctrortrqs r6rz#BaseConnection._set_isolation_levelrvr5c#~K|dEd{V|t|nd|_d|_dS)Nr}r5)ryrrJrMrqs r6_set_isolation_level_genz'BaseConnection._set_isolation_level_gensV**+<=========9>9Ju 5 5 5PT #r5 bool | Nonec|jS)zX The read-only state of the new transactions started on the connection. )rKrXs r6 read_onlyzBaseConnection.read_onlys r5c0||dSro)_set_read_onlyrqs r6rzBaseConnection.read_onlys E"""""r5ctrortrqs r6rzBaseConnection._set_read_onlyrvr5c#~K|dEd{V|t|nd|_d|_dS)Nrr5)ryrerKrMrqs r6_set_read_only_genz!BaseConnection._set_read_only_gensP**;777777777).):$u+++ #r5c|jS)zY The deferrable state of the new transactions started on the connection. )rLrXs r6 deferrablezBaseConnection.deferrables r5c0||dSro)_set_deferrablerqs r6rzBaseConnection.deferrablerrr5ctrortrqs r6rzBaseConnection._set_deferrablervr5c#~K|dEd{V|t|nd|_d|_dS)Nrr5)ryrerLrMrqs r6_set_deferrable_genz"BaseConnection._set_deferrable_gensQ**<888888888*/*;4;;; #r5 attributec#RK|jj}|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_statusIDLErI _sync_genrAeProgrammingErrorr TransactionStatusname)rNrrgs r6ryz!BaseConnection._check_intrans_gens/ T>>dn>~//11 1 1 1 1 1 1 1[3F T>>% (CICCC (;I;;+F338;; >r5r"c*t|jS)z>A `ConnectionInfo` attribute to inspect connection properties.)r"r;rXs r6rczBaseConnection.infosdk***r5rcZ|jsttj|_|jSro)r>rradaptersrXs r6rzBaseConnection.adapters s&~ <():;;DN~r5BaseConnection[Row]c|Sror4rXs r6 connectionzBaseConnection.connections  r5r-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;socketrXs r6filenozBaseConnection.filenos{!!r5c|r/|j}|dSdS)z/Cancel the current operation on the connection.N)_should_cancelr; get_cancelcancel)rNcs r6rzBaseConnection.cancel"sD      &&((A HHJJJJJ  r5ch|jrdS|jr!|jdrtjddS)zhCheck whether the current command should actually be cancelled when invoking cancel*(). Fr z=cancel() cannot be used with a prepared two-phase transactionT)rVrDrrrXs r6rzBaseConnection._should_cancel(sI ; 5 9 1 $O tr5timeoutfloatPQGenConn[None]c#K|j}|tj||Ed{VdS)Nr)r; cancel_connstartrr)rNrrs r6 _cancel_genzBaseConnection._cancel_gen7sYk--// $['BBBBBBBBBBBBr5callbackr7c:|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)r?appendrNrs r6add_notice_handlerz!BaseConnection.add_notice_handler<! $$X.....r5c:|j|dS)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_handlerErr5rO"ReferenceType[BaseConnection[Row]]resr$c|}|r|jsdStj||jj}|jD]A} ||#t $r&}t d||Yd}~:d}~wwxYwdS)Nz)error processing notice callback '%s': %s)r?r Diagnosticr; _encoding Exceptionlogger exception)rOrrNdiagcbexs r6rEzBaseConnection._notice_handlerNsuww .  F|C!677' V VB V4 V V V  !LbRTUUUUUUUU V V Vs A  A;A66A;r8c:|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)r@rrs r6add_notify_handlerz!BaseConnection.add_notify_handler]rr5c:|j|dS)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_handlerfrr5pgn pq.PGnotifyc|}|r|jsdS|jj}t|j||j||j}|jD] }||dSro)r@r;rr)relnamedecodeextrabe_pid)rOrrNencnrs r6rGzBaseConnection._notify_handlerosuww .  Fk# 3;%%c**CI,<,  -+ C N ( / / 4 4 4 N ' . .t 4 4 44 D K " "7 + + + + K ) )'4} ) U U U'/ <<<<<<<D((BSD))$// / / / / / / / F !!### >  /C N ( / / 4 4 4 N ' . .t 4 4 4 F ''---'/ <<<<<<< 2~//11 1 1 1 1 1 1 1 1 1 2 2r5bytesc|jr|jSdg}|jit|j}|d||jdd|j||jrdnd|j||jrdndd ||_|jS) NsBEGINsISOLATION LEVEL_r]s READ ONLYs READ WRITEs DEFERRABLEsNOT DEFERRABLE ) rMr}rrrreplacerrrjoin)rNpartsvals r6r z$BaseConnection._get_tx_start_commands   )( (    + !566C LL+ , , , LL))#s33::<< = = = > % LLJ] K K K ? & LL$/P?P Q Q Q $ % 0 0$$r5c#*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) rArrrDr;rrrrIrrXs r6 _commit_genzBaseConnection._commit_gen2s  ! $4  9 $H  ; )T 1 1 F%%i000000000 > 2~//11 1 1 1 1 1 1 1 1 1 2 2r5c#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) rArrrDrIrr;rrrrCclear maintain_genrXs r6 _rollback_genzBaseConnection._rollback_genFs9  ! $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 2r5 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)rNrrrs r6xidzBaseConnection.xidas( ~i666r5r! 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)rrr from_stringr;rrrrr rrr=rDrr rNr!s r6_tpc_begin_genzBaseConnection._tpc_begin_genns #s## '/#&&C ; )T 1 1$P()GHHMPP    $E %L %%d&@&@&B&BCCCCCCCCCCCr5c#K|jstjd|jdrtjd|jd}|df|_|t dt |Ed{V|jr!|jEd{VdSdS)Nz='tpc_prepare()' must be called inside a two-phase transactionr zF'tpc_prepare()' cannot be used during a prepared two-phase transactionrTzPREPARE TRANSACTION {}) rDrrrrformatr*rIrr%s r6_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 2r5actionrXid | 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 transactionr r_genz{} PREPARED {}) lowerrDrrrrr$getattrrrr(r*)rNr*r!fnamemeths r6_tpc_finish_genzBaseConnection._tpc_finish_gens*v||~~))) ;9 (===)A,CCy (777c3'' +oc** 9 TYq\ 4!9V\\^^!9!9!9::DDItvv         ))$%%,,S[[#c((CC       DIIIr5cdS)z0Raise NotSupportedError if TPC is not supported.Nr4rXs r6rzBaseConnection._check_tpcs  r5N)r;r#)rQrR)rQr*)rQre)rmrerQrR)rmrerQrw)rQr{)rmr{rQrR)rmr{rQrw)rQr)rmrrQrR)rmrrQrw)rr*rQrw)rQr")rQr)rQr)rQr-)rrrQr)rr7rQrR)rOrrr$rQrR)rr8rQrR)rOrrrrQrR)rQr)rmrrQrR)r)rr*rrrQr)rrrrrQr)rrrQrw)rQrw)rQr )rr-rr*rr*rQr)r!r"rQrw)r*rr!r+rQrw)Br/r0r1r2rWarningErrorr DatabaseError DataErrorr IntegrityError InternalErrorrNotSupportedErrorrPrYrdpropertyrVrjrlsetterrprzr}rrrrrrrrryrcrrrrrrrr staticmethodrErrrGrr classmethodrrrrrr r rrr!r&r)r3rr4r5r6r:r:NsiG GE%NOM I)%NOM)+!$!$!$!$F    &3333 )))X)>>>X>   X $$$$""""''''%%%X% ))))""""$$$$ X ####""""$$$$    X $$$$""""$$$$ &+++X+X X""""    CCCC //////// V V V\ V////////   \  0 0 0X 01111111X1,,,,58[:>,,,,,\%%%%N      2 2 2 2%%%%(2222(22226 7 7 7 7DDDD( 2 2 2 2>      r5r:)Sr2 __future__rrloggingtypingrrrrweakrefrr warningsr functoolsr rr rrrrabcrrrsqlrrrDrrowsradaptr_enumsr_compatrrrrpq.miscrrIr _preparingr _capabilitiesr!_connection_infor"pq.abcr#r$psycopg_pool.baser%r&FormatrBINARY ConnStatusrrhrrrrrrINTRANShas_send_close_preparedr getLoggerrr)r0rr7r3r8r:r4r5r6rXsW #""""" ????????????&&&&&&&&(((((((((( """"""<<<<<<<<<<<<''''''######&&&&&&'''''',,,,,,+((((((((****** GK  y~  ]m ] % M # m'     &6,688  9 % % F F F F FZ F F F#Q\ND$89 9999#VHdN3 3333e  e  e  e  e  WS\e  e  e  e  e  r5