qjI=dZddlmZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZmZmZmZdd lmZddl m!Z!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-ddl+m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlZ>m?Z?ddl@mAZAerddlBmCZCdZDejjZFejjZGejjZIejjZJejjZKeLZMejdZOGdd e=e!ZPy)!z* Psycopg connection object (sync version) ) annotationsN) monotonic) TracebackType)Any GeneratorIteratorcastoverload TYPE_CHECKING)contextmanager)pq)errors)waiting) AdaptContextConnDict ConnParamParamsPQGenQueryRV)Xid)Row RowFactory tuple_rowargs_row) AdaptersMap)IsolationLevel)Self) make_conninfoconninfo_to_dict)conninfo_attemptstimeout_from_conninfo)Pipeline)notifies) Transaction)Cursor) capabilities) ServerCursor)BaseConnection CursorRowNotify)Lock)PGconng?psycopgceZdZUdZdZded<ded<ded<d ed <eeeef d8fd Z e d9d d dddd d:dZ d;dZ ddZed dd?dZed d d@dZed dd d dAdZed dd d dBdZ d9d ddd d dCdZ dDdd d dEdZd>dZd>d Zd!d"dFd#Zd$d"dFd%Ze dG dHd&Zddd' dId(ZedJd)ZefdKd*ZdLd+ZdLd,ZdMd-ZdMd.Z dNd/Z!dNd0Z"dNd1Z#dNd2Z$dOd3Z%d>d4Z&dDdPd5Z'dDdPd6Z(dQd7Z)xZ*S)R Connectionz3 Wrapper for a connection to the database. r/ztype[Cursor[Row]]cursor_factoryztype[ServerCursor[Row]]server_cursor_factoryRowFactory[Row] row_factoryzPipeline | None _pipelinec|t||||_t|_t |_t|_yN) super__init__r5r-lockr'r2r)r3)selfpgconnr5 __class__s S/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/connection.pyr:zConnection.__init__Ds3  &F $%1"FN) autocommitprepare_thresholdcontextr5r2c |j|fi|}t|} d} t|} | D]=} td i| }|j || } t j | t} n| ssJ|jdt!|| _|r|| _|r|| _|rt)|j*| _|| _| S#tj$rj}t| dkDrOtjd| jd| jd| jdt||}Yd}~d}~wwxYw) zV Connect to a database server and return a new `Connection` instance. Ntimeoutintervalr z=connection attempt failed: host: %r port: %r, hostaddr %r: %shostporthostaddr)_get_connection_paramsr#r"r _connect_genr wait_conn_WAIT_INTERVALe _NO_TRACEBACKlenloggerdebuggetstrwith_tracebackbool _autocommitr5r2radapters _adaptersrC)clsconninforBrCrDr5r2kwargsparamsrGrvattemptsattemptgenexlast_exs r?connectzConnection.connectOs? ,++H??'/ $V,G (7w7&&x&A&&s^D! $N7((. .j) (BN  .B  &w'7'78BL0 3?? x=1$LLW F+ F+ J/B  s9C  E AEE c|Sr8r<s r? __enter__zConnection.__enter__s r@c|jry|r |jn|j t |dds|jyy#t$r!}tj d||Yd}~Dd}~wwxYw)Nz#error ignored in rollback on %s: %s_pool)closedrollback ExceptionrVwarningcommitgetattrclose)r<exc_typeexc_valexc_tbexc2s r?__exit__zConnection.__exit__so ;;   R  KKMtWd+ JJL,  RDdDQQ RsA A;A66A;c t|fi|S)z3Manipulate connection parameters before connecting.)r!)r_r`ras r?rOz!Connection._get_connection_paramss 3F33r@c`|jryd|_|jjy)zClose the database connection.NT)rp_closedr=finishrls r?rvzConnection.closes% ;;   r@)binarycyr8rk)r<rs r?cursorzConnection.cursors>Ar@cyr8rk)r<rr5s r?rzConnection.cursors r@)r scrollablewithholdcyr8rk)r<namerrrs r?rzConnection.cursors r@cyr8rk)r<rrr5rrs r?rzConnection.cursors#&r@)rr5rrc|j|s |j}|r|j|||||}n|j||}|r t|_|S)zW Return a new `Cursor` to send commands and queries to the connection. )rr5rrr5)_check_connection_okr5r3r2BINARYformat)r<rrr5rrcurs r?rzConnection.cursorsm !!#**K ,,'%! -C%%d %DC CJ r@)preparerc |j}|r t|_|j|||S#tj $r}|j dd}~wwxYw)z8Execute a query and return a cursor to read its results.)rN)rrrexecuterSrTrZ)r<queryrbrrrrgs r?rzConnection.executesV *++-C# ;;ufg;> > *##D) ) *s03AAAc|j5|j|jdddy#1swYyxYw)z/Commit any pending transaction to the database.N)r;wait _commit_genrls r?rtzConnection.commits( YY IId&&( )YY 6?c|j5|j|jdddy#1swYyxYw)z2Roll back to the start of any pending transaction.N)r;r _rollback_genrls r?rqzConnection.rollbacks( YY IId((* +YYrg>@rFc|jsytjr,tj|j |t y|jy)aCancel the current operation on the connection. :param timeout: raise a `~errors.CancellationTimeout` if the cancellation request does not succeed within `timeout` seconds. Note that a successful cancel attempt on the client is not a guarantee that the server will successfully manage to cancel the operation. This is a non-blocking version of `~Connection.cancel()` which leverages a more secure and improved cancellation feature of the libpq, which is only available from version 17. If the underlying libpq is older than version 17, the method will fall back to using the same implementation of `!cancel()`. NrFrH)_should_cancelr(has_cancel_saferrQ _cancel_genrRcancel)r<rGs r? cancel_safezConnection.cancel_safesK ""$   ' ' )      1N  KKMr@@c |j|y#t$r }tjd|Yd}~yd}~wwxYw)NrFzquery cancellation failed: %s)rrrrVrs)r<rGrgs r? _try_cancelzConnection._try_cancel!s= @   W  - @ NN:B ? ? @s >9>c#@Kt|||}|jrB|j5|5|j5|dddddddddy|5|dddy#1swY*xYw#1swY.xYw#1swYyxYw#1swYyxYww)a Start a context block with a new transaction or nested transaction. :param savepoint_name: Name of the savepoint used to manage a nested transaction. If `!None`, one will be chosen automatically. :param force_rollback: Roll back the transaction at the end of the block even if there were no error (e.g. to try a no-op process). :rtype: Transaction N)r&r6pipeline)r<savepoint_nameforce_rollbacktxs r? transactionzConnection.transaction'sj~~ > >>"dmmo'6"'6o""sc*BBA:A.A: B B B% B.A7 3A::B ?BB BBB)rG stop_afterc#K|t|z}t|t}nd}t}d}|j5|jj } |j t|j|}|D]T} t| jj|| jj|| j} | |dz }V|||k\rn$|r!tt|tz }|dkrndddy#tj$r}|jdd}~wwxYw#1swYyxYww)a Yield `Notify` objects as soon as they are received from the database. :param timeout: maximum amount of time to wait for notifications. `!None` means no timeout. :param stop_after: stop after receiving this number of notifications. You might actually receive more than this number if more than one notifications arrives in the same packet. NrrHr g)rminrRr;r= _encodingrr%rSrTrZr,relnamedecodeextrabe_pid) r<rGrdeadlinerI nreceivedencnsrgpgnns r?r%zConnection.notifies<s+   {W,H7N3HH%H YY++''C28DKK#88LB C **3/1A1A#1F AGNI )i:.E">8ik3IJH#~/Y 2++D112 YsA7D>D2&D8BD2= D>D/D**D//D22D;7D>c#K|j5|j|j}|t|x}|_ddd 5|ddd|jdk(r-|j5||jusJd|_dddyy#1swYVxYw#1swYRxYw#1swYyxYw#jdk(r9|j5||jusJd|_dddw#1swYwxYwwxYww)z  *~~"YY#t~~555%)DNY#YY~~"YY#t~~555%)DNYY#s| D1BD B9 B!B9D3B- DBD!B*&B9-B62D9DC6- D6C? ;DDcd tj||jj|S#t$rx|jj t k(rY|jd tj||jj|#tj$rYwxYwwxYw)z Consume a generator operating on the connection. The function must be used on generators that don't change connection fd (i.e. not on connect and reset). rHrrF) rrr=socket _INTERRUPTEDtransaction_statusACTIVErrS QueryCanceled)r<rfrIs r?rzConnection.waits <<T[[%7%7(K K {{--7   -LLdkk&8&88L    s-+.8B/',BB/B*'B/)B**B/c&|j|yr8)set_autocommitr<values r?_set_autocommitzConnection._set_autocommit E"r@c|j5|j|j|dddy#1swYyxYw)z6Method version of the `~Connection.autocommit` setter.N)r;r_set_autocommit_genrs r?rzConnection.set_autocommit* YY IId..u5 6YY !7Ac&|j|yr8)set_isolation_levelrs r?_set_isolation_levelzConnection._set_isolation_levels   'r@c|j5|j|j|dddy#1swYyxYw)z;Method version of the `~Connection.isolation_level` setter.N)r;r_set_isolation_level_genrs r?rzConnection.set_isolation_levels* YY IId33E: ;YYrc&|j|yr8) set_read_onlyrs r?_set_read_onlyzConnection._set_read_onlys 5!r@c|j5|j|j|dddy#1swYyxYw)z5Method version of the `~Connection.read_only` setter.N)r;r_set_read_only_genrs r?rzConnection.set_read_onlys* YY IId--e4 5YYrc&|j|yr8)set_deferrablers r?_set_deferrablezConnection._set_deferrablerr@c|j5|j|j|dddy#1swYyxYw)z6Method version of the `~Connection.deferrable` setter.N)r;r_set_deferrable_genrs r?rzConnection.set_deferrablerrc|j5|j|j|dddy#1swYyxYw)zO Begin a TPC transaction with the given transaction ID `!xid`. N)r;r_tpc_begin_genr<xids r? tpc_beginzConnection.tpc_begins,YY IId))#. /YYrc |j5|j|jdddy#1swYyxYw#tj$r$}tj t |dd}~wwxYw)zV Perform the first phase of a transaction started with `tpc_begin()`. N)r;r_tpc_prepare_genrSObjectNotInPrerequisiteStateNotSupportedErrorrY)r<rgs r? tpc_preparezConnection.tpc_preparesY 9 $//12-- 9%%c"g.D 8 9s0 A 7AAAAA:A55A:c|j5|j|jd|dddy#1swYyxYw)z: Commit a prepared two-phase transaction. COMMITNr;r_tpc_finish_genrs r? tpc_commitzConnection.tpc_commits.YY IId**8S9 :YY "8Ac|j5|j|jd|dddy#1swYyxYw)z= Roll back a prepared two-phase transaction. ROLLBACKNrrs r? tpc_rollbackzConnection.tpc_rollbacks.YY IId**:s; <YYrc|j|jj}|jt t j 5}|jt j|j}ddd|tk(r-|jjtk(r|jS#1swYAxYw)Nr) _check_tpcinforrrr _from_recordr_get_recover_queryfetchallIDLEINTRANSrq)r<statusrress r? tpc_recoverzConnection.tpc_recovers -- [[Xc.>.>%?[ @C KK..0 1,,.CA T>dii::gE MMO A @s 4CC )r=r.r5r4rM)r`rYrBr[rC int | NonerDzAdaptContext | Noner5zRowFactory[Row] | Noner2ztype[Cursor[Row]] | Nonerarreturnr)rr)rwztype[BaseException] | NonerxzBaseException | NoneryzTracebackType | NonerNone)r`rYrarrr)rr)rr[r Cursor[Row])rr[r5RowFactory[CursorRow]rzCursor[CursorRow]) rrYrr[r bool | Nonerr[rzServerCursor[Row]) rrYrr[r5rrrrr[rzServerCursor[CursorRow]) rrYrr[r5zRowFactory[Any] | Nonerrrr[rzCursor[Any] | ServerCursor[Any]r8) rrrbz Params | Nonerrrr[rr)rGfloatrr)NF)rz str | Nonerr[rzIterator[Transaction])rG float | NonerrrzGenerator[Notify, None, None])rzIterator[Pipeline])rfz PQGen[RV]rIrrr)rr[rr)rzIsolationLevel | Nonerr)rrrr)rz Xid | strrr)rzXid | str | Nonerr)rz list[Xid])+__name__ __module__ __qualname____doc____annotations__r rrrr: classmethodrirmr{rOrvr rrrtrqrrr rr%rrRrrrrrrrrrrrrrr __classcell__)r>s@r?r1r18soJ%%22   (,JsOY'G 2 2% 21!()'+.23711 1 & 1 % 1,1111 11f,&%   .44',AA %  4I    "&             "&&& & + &  && !&& .2"&    ,     ) J!%* $ ***  *  * *$* , /34/2@ HM(AE **.0&0;E0 &0d**&=K(#7 (< "6 #7 09;= r@r1)Qr __future__rloggingtimertypesrtypingrrrr r r contextlibr rNrrrSrabcrrrrrrr_tpcrrowsrrrradaptr_enumsr_compatrr`r r!r"r#r6r$ generatorsr%rr&rr' _capabilitiesr( server_cursorr)_connection_baser*r+r, threadingr-pq.abcr.rRFormatTEXTrTransactionStatusrrrKeyboardInterruptr getLoggerrVr1rkr@r?rs #JJ%LLL66"5> $''?? yy~~         $ $   & &   9 %g$gr@