jQdZddlmZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZddlmZdd lmZdd lmZmZdd lmZmZmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/ddl,m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>m?Z?m@Z@ddlAmBZB ddlCZCddlDZDddlDmEZEddl*mFZFe rddlHmIZIdZJejjZLejjZMejjZOejjZPejjZQ eDjeSfZTejdZVGd d!e>e#ZWy)"z+ Psycopg connection object (async version) ) annotationsN) monotonic) TracebackType) TYPE_CHECKINGAnyAsyncGenerator AsyncIteratorcastoverload)asynccontextmanager)errors)pqwaiting)RV AdaptContextConnDict ConnParamParamsPQGenQuery)Xid)AsyncRowFactoryRowargs_row tuple_row) AdaptersMap)IsolationLevel)Self)conninfo_attempts_asyncconninfo_to_dict make_conninfo)timeout_from_conninfo)notifies)AsyncTransaction) AsyncCursor) capabilities)gssapi_requested) AsyncPipeline)BaseConnection CursorRowNotify)AsyncServerCursor)Lock) to_thread)PGconng?psycopgceZdZUdZdZded<ded<ded<d ed <eeeef d:fd Z e d;d d dddd ddZ e d?dZd@dZed ddAdZed d dBdZed dd d dCdZed dd d dDdZ d;d ddd d dEdZ dFdd d dGdZd@dZd@d Zd!d"dHd#Zd$d"dHd%Ze dI dJd&Zddd' dKd(ZedLd)ZedLd*ZefdMd+ZdNd,ZdNd-ZdOd.Z dOd/Z!dPd0Z"dPd1Z#dPd2Z$dPd3Z% dQd4Z&dRd5Z'd@d6Z(dFdSd7Z)dFdSd8Z*dTd9Z+xZ,S)UAsyncConnectionz3 Wrapper for a connection to the database. r1ztype[AsyncCursor[Row]]cursor_factoryztype[AsyncServerCursor[Row]]server_cursor_factoryAsyncRowFactory[Row] row_factoryzAsyncPipeline | None _pipelinec|t||||_t|_t |_t|_yN) super__init__r7r.lockr&r4r-r5)selfpgconnr7 __class__s _/var/www/html/kelly/kelly-backend/venv/lib/python3.12/site-packages/psycopg/connection_async.pyr<zAsyncConnection.__init__Ls3  &F )%6"FN) autocommitprepare_thresholdcontextr7r4c K tjdk(rOtj}t |tj r!ddlm} tjdd| z|j|fi|d{} t| } d} t| d{} g}| D]}|jd|jd|jd f}d |z}tjd | t!di|}|j#|| }t%j&|t( d{} tjd|n| s|dd}t5|dk(r|j3dt-|g}|j/d|j7d|Dt9|dj;|j3dt=j>r;| j@jBr%tE| stGjHdtJtM|| _'|r|| _(|r|| _)|rtU|jV| _,|| _-| S777G#tj*$r>}tjd|t-||j/||fYd}~#d}~wtj0$r}|j3dd}~wwxYww)z[ Connect to a database server and return a new `AsyncConnection` instance. win32r )_asyncio_run_snippetztPsycopg cannot use the 'ProactorEventLoop' to run in async mode. Please use a compatible event loop, for instance by Nhostporthostaddrz host: %r, port: %r, hostaddr: %rzconnection attempt: %stimeoutintervalzconnection succeeded: %szconnection failed: %s: %srz7Multiple connection attempts failed. All failures were:c32K|]\}}d|d|yw)z- z: N).0errordescrs rA z*AsyncConnection.connect..s"O;<5%2eWBug.;s aathe connection was obtained using the GSSAPI relying on the 'gssencmode=prefer' libpq default. In a future psycopg[binary] version this default will be changed to 'disable'. If you wish to interact with the GSSAPI reliably please set the 'gssencmode' parameter in the connection string or the 'PGGSSENCMODE' environment variable to 'prefer' or 'require').sysplatformasyncioget_running_loop isinstanceProactorEventLoop_compatrIeInterfaceError_get_connection_paramsr#r getloggerdebugr" _connect_genrwait_conn_async_WAIT_INTERVALErrorstrappend _NO_TRACEBACKwith_tracebacklenextendtypejoinr'has_used_gssapir? used_gssapir(warningswarnDeprecationWarningbool _autocommitr7r4radapters _adaptersrE)clsconninforDrErFr7r4kwargslooprIparamsrOrvattempts conn_errorsattempttdescrrWgenexlast_exliness rAconnectzAsyncConnection.connectWs ||w&//1dG$=$=>=**U2345 2s11(EfEE'/ 08813 Gkk&)7;;v+> J@WXF6?E LL15 9 (7w7&&x&A"223PP 7? "!"oa(G;1$,,T22\NE LLR S LLO;O O$w- % 01@@F F  ( ( * %%$V, MMP # j) (BN  .B  &w'7'78BL0 sF9Q77 0 8%RI""B;//?? .''-- .snA:K7<I3=K7I6AK76s rA __aenter__zAsyncConnection.__aenter__s  sc@K|jry|r |jd{n|j d{t |dds|jd{yy7D#t$r!}tj d||Yd}~Nd}~wwxYw7Z77w)Nz#error ignored in rollback on %s: %s_pool)closedrollback Exceptionrgwarningcommitgetattrclose)r>exc_typeexc_valexc_tbexc2s rA __aexit__zAsyncConnection.__aexit__s ;;   Rmmo%%++-  tWd+**,  ,& RDdDQQ R  sZBA-A+A-BB$B$B%B+A-- B6B BBBBc"Kt|fi|Sw)z3Manipulate connection parameters before connecting.)r!)r~rrs rArez&AsyncConnection._get_connection_paramss 3F33s chK|jryd|_|jjyw)zClose the database connection.NT)r_closedr?finishrs rArzAsyncConnection.closes) ;;   s02)binarycyr:rT)r>rs rAcursorzAsyncConnection.cursorsCFrBcyr:rT)r>rr7s rArzAsyncConnection.cursors"%rB)r scrollablewithholdcyr:rT)r>namerrrs rArzAsyncConnection.cursors"%rBcyr:rT)r>rrr7rrs rArzAsyncConnection.cursors(+rB)rr7rrc|j|s |j}|r|j|||||}n|j||}|r t|_|S)z\ Return a new `AsyncCursor` to send commands and queries to the connection. )rr7rrr7)_check_connection_okr7r5r4BINARYformat)r>rrr7rrcurs rArzAsyncConnection.cursorsm !!#**K ,,'%! -C%%d %DC CJ rB)preparercK |j}|r t|_|j|||d{S7#tj $r}|j dd}~wwxYww)z8Execute a query and return a cursor to read its results.)rN)rrrexecutercrorp)r>queryrrrrrs rArzAsyncConnection.executes_ *++-C# UFGDD DD *##D) ) *s1A+4?=?A+?A(A##A((A+cK|j4d{|j|jd{dddd{y7<77 #1d{7swYyxYww)z/Commit any pending transaction to the database.N)r=wait _commit_genrs rArzAsyncConnection.commit)sD999))D,,./ / /99 /999SA+AA+#AAA A+ A A+AA+A(A A($A+cK|j4d{|j|jd{dddd{y7<77 #1d{7swYyxYww)z2Roll back to the start of any pending transaction.N)r=r _rollback_genrs rArzAsyncConnection.rollback.sD999))D..01 1 199 1999rg>@rNcK|jsytjr4tj|j |t d{y t|jd{y7$7w)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()`. NrNrP) _should_cancelr'has_cancel_saferrj _cancel_genrkr/cancel)r>rOs rA cancel_safezAsyncConnection.cancel_safe3sp ""$   ' ' )))   1N   ,,,   -s$AA?A;A?5A=6A?=A?@cK |j|d{y7#t$r }tjd|Yd}~yd}~wwxYww)NrNzquery cancellation failed: %s)rrrgr)r>rOrs rA _try_cancelzAsyncConnection._try_cancelPsF @""7"3 3 3 @ NN:B ? ? @s6A !!A ! A AA A  A c ,Kt|||}|jrs|j4d{|4d{|j4d{|dddd{dddd{dddd{y|4d{|dddd{y77v7_7L#1d{7swY\xYw7S#1d{7swYcxYw7Z#1d{7swYyxYw7e7R#1d{7swYyxYww)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: AsyncTransaction N)r%r8pipeline)r>savepoint_nameforce_rollbacktxs rA transactionzAsyncConnection.transactionVsdNN C >>}}DMMOO- D C$ DC;DC? D*C=+D2C&4C6C8C >C?C C C&C! CC! C&$D&C8,C/ -C84D=D?DD D D)rO stop_aftercK|t|z}t|t}nd}t}d}|jrt j dt |j4d{|jj}d|jc|_ } |r|r|j|dz }|rn |jt|j|d{}|D]S} t#| j$j'|| j(j'|| j*|dz }U|||k\rn$|r!tt|tz }|dkrn||_ dddd{y77#tj$r} | j!dd} ~ wwxYw#||_ wxYw7B#1d{7swYyxYww)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. Nrzusing 'notifies()' together with notifies handlers on the same connection is not reliable. Please use only one of these methodsr rPg)rminrk_notify_handlersrwrxRuntimeWarningr=r? _encoding_notifies_backlogpopleftrr$rcrorpr,relnamedecodeextrabe_pid) r>rOrdeadlinerQ nreceivedencdpgnsrpgns rAr$zAsyncConnection.notifiesks   {W,H7N3HH%H  MM8  999++''C)-d.D.D %D "A$ +"#))+-%NI :)- ( 5*3*$D$(C"( # 2 23 7 # 0 0 5 # # &NI $("-)z2I #&~x)+7M#N#c>!CF*+&W99"$ !:"$"3"3D"99:.*+&W999sA!G#E?$G'+F>F03F05)FFF#BF0'F>. G9F<:GFF-F((F--F00 F99F><G>GG G Gc K|j4d{|j|jx}t|x}|_dddd{ 4d{|dddd{|jdk(r=|j4d{||jusJd|_dddd{yy77t#1d{7swYxYw77l#1d{7swY|xYw7_7:#1d{7swYyxYw#jdk(rT|j4d{7||jusJd|_dddd{7w#1d{7swYwxYwwxYww)zrs rArzAsyncConnection.pipelines 999  % % ' NN*3,9$,??4> 9 *x x~~"999#t~~555%)DN%99#9999 xxxx%9999~~"999#t~~555%)DN%9999#s(E9CE91C  E9CE9DC D#C#) D4C!5D9E9C8E9C<4 E9?C:E9E9 CC CE9D!D#C5)C, *C51D8E9:E9<DD D E9 E61D4 2E66E  E6EE6 E2 &E)'E2 .E66E9cnK|jrJt|dx}|_ |4d{|dddd{|jdk(sJ||jusJd|_y7B7/#1d{7swY?xYw#|jdk(sJ||jusJd|_wxYww)zglike pipeline() but don't acquire a lock. Assume that the caller is holding the lock. T)_no_lockNr)r8r)rrs rA_pipeline_nolockz AsyncConnection._pipeline_nolocks>>!!$1$$FF4> "x x>>Q& &&t~~- --!DN  xxxx>>Q& &&t~~- --!DNs`#B5BA/BA3 BA1B)B5/B1B3B9A< :BB*B22B5cK tj||jj|d{S7#t$r|jj t k(rk|jdd{7 tj||jj|d{7#tj$rYwxYwwxYww)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). rPNrrN) r wait_asyncr?socket _INTERRUPTEDtransaction_statusACTIVErrc QueryCanceled)r>rrQs rArzAsyncConnection.waits  ++C1C1ChWW WW {{--7&&s&333!,,S$++2D2DxXXX    saC/:8:C:;C 5A86C values rA_set_autocommitzAsyncConnection._set_autocommit    | ,rBcK|j4d{|j|j|d{dddd{y7=77 #1d{7swYyxYww)z6Method version of the `~Connection.autocommit` setter.N)r=r_set_autocommit_genrs rArzAsyncConnection.set_autocommitF999))D44U;< < <99 <999TA,AA,$AAA A, A A,AA,A)A A)%A,c( |jdy)Nisolation_level)rset_isolation_levelrs rA_set_isolation_levelz$AsyncConnection._set_isolation_levels    0 1rBcK|j4d{|j|j|d{dddd{y7=77 #1d{7swYyxYww)z;Method version of the `~Connection.isolation_level` setter.N)r=r_set_isolation_level_genrs rArz#AsyncConnection.set_isolation_levelsF999))D99%@A A A99 A999rc( |jdy)N read_only)r set_read_onlyrs rA_set_read_onlyzAsyncConnection._set_read_only s    { +rBcK|j4d{|j|j|d{dddd{y7=77 #1d{7swYyxYww)z5Method version of the `~Connection.read_only` setter.N)r=r_set_read_only_genrs rArzAsyncConnection.set_read_onlysF999))D33E:; ; ;99 ;999rc( |jdy)N deferrable)rset_deferrablers rA_set_deferrablezAsyncConnection._set_deferrablerrBcK|j4d{|j|j|d{dddd{y7=77 #1d{7swYyxYww)z6Method version of the `~Connection.deferrable` setter.N)r=r_set_deferrable_genrs rArzAsyncConnection.set_deferrablerrc&td|d|d)Nz'the zD property is read-only on async connections: please use 'await .set_z ()' instead.)AttributeError)r> attributes rArzAsyncConnection._no_set_async!s(  }%++4+\C rBcK|j4d{|j|j|d{dddd{y7=77 #1d{7swYyxYww)zO Begin a TPC transaction with the given transaction ID `!xid`. N)r=r_tpc_begin_genr>xids rA tpc_beginzAsyncConnection.tpc_begin'sH999))D//45 5 599 5999rcPK |j4d{|j|jd{dddd{y7<77 #1d{7swYyxYw#tj$r$}tj t |dd}~wwxYww)zV Perform the first phase of a transaction started with `tpc_begin()`. N)r=r_tpc_prepare_genrcObjectNotInPrerequisiteStateNotSupportedErrorrm)r>rs rA tpc_preparezAsyncConnection.tpc_prepare.su 9yyyii 5 5 7888!yy8!yyy-- 9%%c"g.D 8 9sB&A,AA,#AAA A, A A,B&A,AA,A)A A)%A,(B&)A,,B#?BB##B&cK|j4d{|j|jd|d{dddd{y7>77 #1d{7swYyxYww)z: Commit a prepared two-phase transaction. NCOMMITr=r_tpc_finish_genr s rA tpc_commitzAsyncConnection.tpc_commit8sJ999))D003?@ @ @99 @999TA-AA-%AAA A- A A-AA-A*A! A*&A-cK|j4d{|j|jd|d{dddd{y7>77 #1d{7swYyxYww)z= Roll back a prepared two-phase transaction. NROLLBACKrr s rA tpc_rollbackzAsyncConnection.tpc_rollback?sJ999))D00SAB B B99 B999rcK|j|jj}|jt t j 4d{}|jt jd{|jd{}dddd{|tk(r5|jjtk(r|jd{S77n7X7J#1d{7swYZxYw7#w)Nr) _check_tpcinforrrr _from_recordr_get_recover_queryfetchallIDLEINTRANSr)r>statusrress rA tpc_recoverzAsyncConnection.tpc_recoverFs --;;8C4D4D+E;FF#++c4467 7 7 &CGF T>dii::gE--/ ! ! G 7&GFFF "sxAD C.D 'C6C0C6C2C6 D )C4*=D 'D (D 0C62C64D 6D<C? =DD )r?r0r7r6rZ)rrmrDrzrE int | NonerFzAdaptContext | Noner7zAsyncRowFactory[Row] | Noner4ztype[AsyncCursor[Row]] | Nonerrreturnr)r(r)rztype[BaseException] | NonerzBaseException | NonerzTracebackType | Noner(None)rrmrrr(r)r(r))rrzr(AsyncCursor[Row])rrzr7AsyncRowFactory[CursorRow]r(zAsyncCursor[CursorRow]) rrmrrzr bool | Nonerrzr(zAsyncServerCursor[Row]) rrmrrzr7r+rr,rrzr(zAsyncServerCursor[CursorRow]) rrmrrzr7zAsyncRowFactory[Any] | Nonerr,rrzr(z)AsyncCursor[Any] | AsyncServerCursor[Any]r:) rrrz Params | Nonerr,rrzr(r*)rOfloatr(r))NF)rz str | Nonerrzr(zAsyncIterator[AsyncTransaction])rOz float | Nonerr'r(zAsyncGenerator[Notify, None])r(zAsyncIterator[AsyncPipeline])rz PQGen[RV]rQr-r(r)rrzr(r))rzIsolationLevel | Noner(r))rr,r(r))rrmr(r))r z Xid | strr(r))r zXid | str | Noner(r))r(z list[Xid])-__name__ __module__ __qualname____doc____annotations__r rrrr< classmethodrrrrerr rrrrrrr rr$rrrkrrrrrrrrrrr rrrr& __classcell__)r@s@rAr3r3@sJ**77%%## -11Ey,Q 7 7* 7T!()'+378<TT T & T % T1T6TT TTl,&%   .44',FF %%%4N% %% "& %% %  %  % %% "&++ + 0 +  ++ &++ 37"&    1     3 J!%* $ ***  *  * *&0 2 59:58@ HM(AE (**.J+&J+;EJ+ %J+X**$""(rOs #TT*LLL;;"NN+ )%'-*??3" yy~~         $ $   & &**,=>L   9 %PnS)PrB