qj EdZddlmZddlZddlmZddlmZddlm 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@Z@ddlAZAddlAmBZBddl)mCZCerddlEmFZFdZGejjZIejjZJejjZLejjZMejjZN eAjePfZQejdZSGd d!e=e!ZTy)"z+ Psycopg connection object (async version) ) annotationsN) monotonic) TracebackType)AnyAsyncGenerator AsyncIteratorcastoverload TYPE_CHECKING)asynccontextmanager)pq)errors)waiting) AdaptContextConnDict ConnParamParamsPQGenQueryRV)Xid)RowAsyncRowFactory tuple_rowargs_row) AdaptersMap)IsolationLevel)Self) make_conninfoconninfo_to_dict)conninfo_attempts_asynctimeout_from_conninfo) AsyncPipeline)notifies)AsyncTransaction) AsyncCursor) capabilities)AsyncServerCursor)BaseConnection CursorRowNotify)Lock) to_thread)PGconng?psycopgceZdZUdZdZded<ded<ded<d ed <eeeef d9fd Z e d:d d dddd d;dZ ddZd?dZed dd@dZed d dAdZed dd d dBdZed dd d dCdZ d:d ddd d dDdZ dEdd d dFdZd?dZd?d Zd!d"dGd#Zd$d"dGd%Ze dH dId&Zddd' dJd(ZedKd)ZefdLd*ZdMd+ZdMd,ZdNd-ZdNd.Z dOd/Z!dOd0Z"dOd1Z#dOd2Z$ dPd3Z%dQd4Z&d?d5Z'dEdRd6Z(dEdRd7Z)dSd8Z*xZ+S)TAsyncConnectionz3 Wrapper for a connection to the database. r0ztype[AsyncCursor[Row]]cursor_factoryztype[AsyncServerCursor[Row]]server_cursor_factoryAsyncRowFactory[Row] row_factoryzAsyncPipeline | None _pipelinec|t||||_t|_t |_t|_yN) super__init__r6r-lockr'r3r)r4)selfpgconnr6 __class__s Y/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/connection_async.pyr;zAsyncConnection.__init__Js3  &F )%6"FN) autocommitprepare_thresholdcontextr6r3c K tjdk(rCtj}t |tj rt jd|j|fi|d{} t| } d} t| d{} | D]E} td i| }|j|| }tj|td{} n| ssJ|j-dt/|| _|r|| _|r|| _|rt7|j8| _|| _| S777i#t j $rj}t#| dkDrOt$j'd| j)d| j)d | j)d t+||}Yd}~$d}~wwxYww) z[ Connect to a database server and return a new `AsyncConnection` instance. win32zPsycopg cannot use the 'ProactorEventLoop' to run in async mode. Please use a compatible event loop, for instance by setting 'asyncio.set_event_loop_policy(WindowsSelectorEventLoopPolicy())'Ntimeoutintervalr z=connection attempt failed: host: %r port: %r, hostaddr %r: %shostporthostaddr)sysplatformasyncioget_running_loop isinstanceProactorEventLoopeInterfaceError_get_connection_paramsr#r"r _connect_genrwait_conn_async_WAIT_INTERVAL _NO_TRACEBACKlenloggerdebuggetstrwith_tracebackbool _autocommitr6r3radapters _adaptersrD)clsconninforCrDrEr6r3kwargsloopparamsrIrvattemptsattemptgenexlast_exs r@connectzAsyncConnection.connectUs ||w&//1dG$=$=>**>2s11(EfEE'/ 088G (7w7&&x&A"223PP! $N7((. .j) (BN  .B  &w'7'78BL0 EF9 Q?? x=1$LLW F+ F+ J/B  sbA.G0D=1GD? Gfinishrvs r@rzAsyncConnection.closes) ;;   s02)binarycyr9ru)r=rs r@cursorzAsyncConnection.cursorsCFrAcyr9ru)r=rr6s r@rzAsyncConnection.cursors"%rA)r scrollablewithholdcyr9ru)r=namerrrs r@rzAsyncConnection.cursors"%rAcyr9ru)r=rrr6rrs r@rzAsyncConnection.cursors(+rA)rr6rrc|j|s |j}|r|j|||||}n|j||}|r t|_|S)z\ Return a new `AsyncCursor` to send commands and queries to the connection. )rr6rrr6)_check_connection_okr6r4r3BINARYformat)r=rrr6rrcurs r@rzAsyncConnection.cursorsm !!#**K ,,'%! -C%%d %DC CJ rA)preparercK |j}|r t|_|j|||d{S7#tj $r}|j dd}~wwxYww)z8Execute a query and return a cursor to read its results.)rN)rrrexecuterWr]rc)r=queryrlrrrrqs r@rzAsyncConnection.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_genrvs r@r~zAsyncConnection.commitsD999))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_genrvs r@r{zAsyncConnection.rollbacksD999))D..01 1 199 1999rg>@rHcK|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()`. NrHrJ) _should_cancelr(has_cancel_saferr[ _cancel_genr\r.cancel)r=rIs r@ cancel_safezAsyncConnection.cancel_safesp ""$   ' ' )))   1N   ,,,   -s$AA?A;A?5A=6A?=A?@cK |j|d{y7#t$r }tjd|Yd}~yd}~wwxYww)NrHzquery cancellation failed: %s)rr|r_r})r=rIrqs r@ _try_cancelzAsyncConnection._try_cancel5sF @""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&r7pipeline)r=savepoint_nameforce_rollbacktxs r@ transactionzAsyncConnection.transaction;sdNN 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)rI stop_aftercK|t|z}t|t}nd}t}d}|j4d{|jj } |j t|j|d{}|D]U} t| jj|| jj|| j} | |dz }W|||k\rn$|r!tt|tz }|dkrndddd{y77#tj$r}|jdd}~wwxYw75#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. NrrJr g)rminr\r<r> _encodingrr%rWr]rcr,relnamedecodeextrabe_pid) r=rIrdeadlinerK nreceivedencnsrqpgnns r@r%zAsyncConnection.notifiesPs<   {W,H7N3HH%H 999++''C2#yy$++)>yRRB C **3/1A1A#1F AGNI )i:.E">8ik3IJH#~/99S2++D112 999s|;E&DE&E)D#D!D#BE E&EE&!D##E 6EE  EE&E#E E#E&c K|j4d{|j|j}|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)z 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 .E66E9cK 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). rJNrrH) r wait_asyncr>socket _INTERRUPTEDtransaction_statusACTIVErrW QueryCanceled)r=rprKs r@rzAsyncConnection.waits  ++C1C1ChWW WW {{--7&&s&333!,,S$++2D2DxXXX    saC/:8:C:;C 5A86C 77 #1d{7swYyxYww)z: Commit a prepared two-phase transaction. NCOMMITr<r_tpc_finish_genrs r@ tpc_commitzAsyncConnection.tpc_commitsJ999))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. NROLLBACKrrs r@ tpc_rollbackzAsyncConnection.tpc_rollbacksJ999))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 r@ tpc_recoverzAsyncConnection.tpc_recovers --;;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>r/r6r5rO)rirbrCrdrD int | NonerEzAdaptContext | Noner6zAsyncRowFactory[Row] | Noner3ztype[AsyncCursor[Row]] | Nonerjrreturnr)rr)rztype[BaseException] | NonerzBaseException | NonerzTracebackType | NonerNone)rirbrjrrr)rr )rrdrAsyncCursor[Row])rrdr6AsyncRowFactory[CursorRow]rzAsyncCursor[CursorRow]) rrbrrdr bool | NonerrdrzAsyncServerCursor[Row]) rrbrrdr6r rr rrdrzAsyncServerCursor[CursorRow]) rrbrrdr6zAsyncRowFactory[Any] | Nonerr rrdrz)AsyncCursor[Any] | AsyncServerCursor[Any]r9) rrrlz Params | Nonerr rrdrr )rIfloatrr )NF)rz str | NonerrdrzAsyncIterator[AsyncTransaction])rI float | NonerrrzAsyncGenerator[Notify, None])rzAsyncIterator[AsyncPipeline])rpz PQGen[RV]rKrrr)rrdrr )rzIsolationLevel | Nonerr )rr rr )rrbrr )rz Xid | strrr )rzXid | str | Nonerr )rz list[Xid]),__name__ __module__ __qualname____doc____annotations__r rrrr; classmethodrsrwrrYrr rrr~r{rrr rr%rr\rrrrrrrrrrrrrrr __classcell__)r?s@r@r2r2>s~J**77%%## -11Ey,Q 7 7* 7;!()'+378<;; ; & ; % ;1;6;; ;;z,&%   .44',FF %%%4N% %% "& %% %  %  % %% "&++ + 0 +  ++ &++ 37"&    1     3 J!%* $ ***  *  * *&0 2 59:58@ HM(AE (**.0&0;E0 %0d**&CQ(' = , B & < ' =    69AC rAr2)Ur __future__rloggingtimertypesrtypingrrrr r r contextlibr rPrrrWrabcrrrrrrr_tpcrrowsrrrradaptr_enumsr_compatrrir r!r"r#r7r$ generatorsr%rr& cursor_asyncr' _capabilitiesr( server_cursorr)_connection_baser*r+r,rQrSr-r. threadingpq.abcr/r\FormatTEXTrTransactionStatusrrrCancelledErrorKeyboardInterruptr getLoggerr_r2rurAr@r/s  #TT*LLL;;"5D$ )%',??" yy~~         $ $   & &**,=>L   9 %InS)IrA