ۇeh#ddlZddlmZddlmZGddejZGddejZGd d eZGd d eZdS) N) connresource) exceptionscleZdZdZdZfdZejdZejdZ dZ xZ S) CursorFactoryzA cursor interface for the results of a query. A cursor interface can be used to initiate efficient traversal of the results of a large query. )_state_args _prefetch_query_timeout _record_classct|||_||_||_||_||_||_||dSdSN) super__init__r r r r rr attach) self connectionquerystateargsprefetchtimeout record_class __class__s @/var/www/api/venv/lib/python3.11/site-packages/asyncpg/cursor.pyrzCursorFactory.__init__sg $$$ !   )   LLNNNNN  c |jdn|j}t|j|j|j|j|j||jS)N2)r CursorIterator _connectionr rr r r )rrs r __aiter__zCursorFactory.__aiter__2sI/22T^   K K J    M   rc|jtjdt|j|j|j|j|j}| |j  S)Nz;prefetch argument can only be specified for iterable cursor) r rInterfaceErrorCursorr!r rr r _initr __await__)rcursors rr'zCursorFactory.__await__?sq > %+MOO O   K K J     ||DM**44666rc|j:|j|j|jdSdSrrdetachr!_maybe_gc_stmtrs r__del__zCursorFactory.__del__ME ; " K     + +DK 8 8 8 8 8 # "r) __name__ __module__ __qualname____doc__ __slots__rrguardedr"r'r. __classcell__rs@rrrs I(      7 7 79999999rrcLeZdZdZfdZdZdZdZdZdZ dZ d Z xZ S) BaseCursor)rr _portal_name _exhaustedr r ct|||_||_||d|_d|_||_||_dS)NF) rrr rrr:r;r r )rrrrrrrs rrzBaseCursor.__init__^s` $$$    LLNNN  )rc|jtjd|jjrtjd|jjstjddS)Nz(cursor: no associated prepared statementz(cursor: the prepared statement is closedz1cursor cannot be created outside of a transaction)rrr$closedr! _top_xactNoActiveSQLTransactionErrorr-s r _check_readyzBaseCursor._check_readyis ; +:<< < ;  <+:<< <) E8CEE E E Erc(K||jrtjd|j}|j}|d|_||j|j |j|d|d{V\}}|_ |S)N!cursor already has an open portalportalT) rAr:rr$r! _protocol_get_unique_id bind_executerr r;)rnrconprotocolbuffer_s r _bind_execzBaseCursor._bind_execvs    5+355 5=..x88+3+@+@ KT%64,J,J&J&J&J&J&J&J"4? rcK||jrtjd|j}|j}|d|_||j|j |j|d{V}|S)NrCrD) rAr:rr$r!rErFbindrr )rrrIrJrKs r_bindzBaseCursor._binds    5+355 5=..x88}}T[$*%)%6%,........ rcK||jstjd|jj}||j|j|d|d{V\}}|_|S)N#cursor does not have an open portalT) rAr:rr$r!rEexecuterr;)rrHrrJrKrLs r_execzBaseCursor._execs   7+577 7#-+3+;+; K*AtW,>,>&>&>&>&>&>&>"4? rcK||jstjd|jj}||j|d{Vd|_dS)NrR)rAr:rr$r!rE close_portal)rrrJs r _close_portalzBaseCursor._close_portals   7+577 7#-##D$5w????????? rc ^g}|jr|d|d|jjdrd}n |jj}d||jj|jjd |t|S)N exhaustedzasyncpg.asyncpgz<{}.{} "{!s:.30}" {}{:#x}> ) r;appendrr1 startswithformatr0rrjoinid)rattrsmods r__repr__zBaseCursor.__repr__s ? & LL % % % R > $ / / ; ; ,CC.+C+22 ( K  HHUOORXX'' 'rc|j:|j|j|jdSdSrr*r-s rr.zBaseCursor.__del__r/r) r0r1r2r4rrArMrPrTrWrdr.r6r7s@rr9r9SsI * * * * * E E E       ! ! !''' 9999999rr9cbeZdZdZfdZejdZejdZxZ S)r )_bufferr r ct||||||dkrtjdt j|_||_||_dS)Nrz+prefetch argument must be greater than zero) rrrr$ collectionsdequergr r ) rrrrrrrrrs rrzCursorIterator.__init__sj UE4FFF q==+=?? ?#(** ! rc|Srr-s rr"zCursorIterator.__aiter__s rcK|jQ|j|j|jd|jd{V|_|j|jsG|js@| |j |jd{V}|j ||j sG|js@| |j |jd{V}|j ||jr'|jr ||jd{V|j r|j St NT)namedr)rr!_get_statementr r r rr:r;rMr rgextendrTrWpopleftStopAsyncIteration)rrKs r __anext__zCursorIterator.__anext__sz ;  $ 0 ? ?  !/ !@!!DK K     ( (??4>4=IIIIIIIIF L   ' ' '| (DO (::dndmDDDDDDDDF L   ' ' '   4 4$$T]33 3 3 3 3 3 3 3 < *<'')) )  r) r0r1r2r4rrr5r"rtr6r7s@rr r s{4I     (!!!!!!!rr ceZdZdZdZdZejdddZejdddZ ejddde fd Z dS) r%z-An open *portal* into the results of a query.rlcK|jL|j|j|d|jd{V|_|j|||d{V|Srn)rr!rpr r rrArP)rrs rr&z Cursor._inits ;  $ 0 ? ? !/ !@!!DK K    jj!!!!!!!!! rN)rcK||dkrtjd|jrgS|||d{V}t ||krd|_|S)zReturn the next *n* rows as a list of :class:`Record` objects. :param float timeout: Optional timeout value in seconds. :return: A list of :class:`Record` instances. rn must be greater than zeroNT)rArr$r;rTlen)rrHrrecss rfetchz Cursor.fetch s  66+,IJJ J ? IZZ7++++++++ t99q=="DO rcK||jrdS|d|d{V}t|dkr d|_dS|dS)zReturn the next row. :param float timeout: Optional timeout value in seconds. :return: A :class:`Record` instance. NrTr)rAr;rTry)rrrzs rfetchrowzCursor.fetchrowsr  ? 4ZZ7++++++++ t99q=="DO4AwrreturncPK||dkrtjd|jj}|d||j|d{V}t| d}||krd|_ |S)zSkip over the next *n* rows. :param float timeout: Optional timeout value in seconds. :return: A number of rows actually skipped over (<= *n*). rrxzMOVE FORWARD {:d} {}NrT) rArr$r!rErr_r:intsplitr;)rrHrrJstatusadvanceds rforwardzCursor.forward/s  66+,IJJ J#-~~&<&C&C t '"'"#*,,,,,,,,v||~~a()) a<<"DOr) r0r1r2r3r4r&rr5r{r}rrrlrrr%r%s77I   (,"(, *.3rr%) rirZrrConnectionResourcerr9r r%rlrrrs B9B9B9B9B9L3B9B9B9Jl9l9l9l9l90l9l9l9^5!5!5!5!5!Z5!5!5!pIIIIIZIIIIIr