2i,$dZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZdd lmZmZddl m!Z!ddl"m#Z#erddl$m%Z%ejLjNZ'Gdde!defZ(y)z Psycopg AsyncCursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyoverload)asynccontextmanager) AsyncIteratorIterable)errors)pq)ParamsQueryQueryNoTemplate) AsyncCopy AsyncWriter)AsyncRowFactoryRowRowMaker)SelfTemplate) BaseCursor) AsyncPipeline)AsyncConnectionceZdZdZdZed#dZe d$dZdd d%fdZd&dZ d'd Zd(d Ze d)d Z e jd*d Z d+d Z e d,ddd d-dZ eddd d.dZ d,ddd d/dZ dd d0dZ d,ddd d1dZd2dZd3dZd4dZd5d6dZd7dZd&dZd8dZd9d:dZe d,dd  d;d!Zd(d"ZxZS)< AsyncCursorpsycopgcyNr)self connections S/var/www/html/secretsanta/venv/lib/python3.12/site-packages/psycopg/cursor_async.py__init__zAsyncCursor.__init__!s:=cyr r)r!r" row_factorys r#r$zAsyncCursor.__init__$s r%N)r'cNt|||xs |j|_yr )superr$r' _row_factory)r!r"r' __class__s r#r$zAsyncCursor.__init__)s% $'A:+A+Ar%cK|Swr rr!s r# __aenter__zAsyncCursor.__aenter__2s  sc@K|jd{y7wr )close)r!exc_typeexc_valexc_tbs r# __aexit__zAsyncCursor.__aexit__5s jjls c,K|jyw)zI Close the current cursor and free associated resources. N)_closer-s r#r0zAsyncCursor.close=s sc|jS)z9Writable attribute to control how result rows are formed.r*r-s r#r'zAsyncCursor.row_factoryCs   r%cF||_|jr|||_yyr )r*pgresult _make_row)r!r's r#r'zAsyncCursor.row_factoryHs"' ==(.DN r%c$|j|Sr r8r-s r#_make_row_makerzAsyncCursor._make_row_makerNs  &&r%preparebinaryc Kywr r)r!queryparamsr?r@s r#executezAsyncCursor.executeQs c Kywr r)r!rBr?r@s r#rDzAsyncCursor.execute[s rEc jK |jj4d{|jj|j||||d{dddd{|S7L77 #1d{7swY|SxYw#tj $r}|j dd}~wwxYww)z= Execute a query or command to the database. Nr>)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)r!rBrCr?r@exs r#rDzAsyncCursor.executeds *zz  jjoo%%eVWV%T            *##D) ) *sB3BA+B2A1A-A1 B$A/%B)B3+B-A1/B1B7A: 8B?BB3BB0B++B00B3F) returningc ,K |jj4d{tjr|jjr5|jj |j |||d{n|jj4d{|jj |j |||d{dddd{n4|jj |j|||d{dddd{y7777a7S#1d{7swY.xYw757'#1d{7swYyxYw#tj$r}|jdd}~wwxYww)zI Execute the same command with a sequence of input data. N) rHrIr is_supported _pipelinerJ_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerLrMrN)r!rB params_seqrPrOs r# executemanyzAsyncCursor.executemanyxsh  *zz   --/zz++"jjoo ::5*iX $(::#>#>#@"&**// $ > >$):y!"#**//99%YW#     #    ( *##D) ) *sFE(D/E(AE<D2=#E D4!E$0D:D6D: E$D8%4EEE E()E*E(.F/E(2E4E6D:8E:E EE EE(E%E E%!E($F%E((F;F  FFr r@sizec :K|jjrtjd|jj 4d{ |jj |j||||d{d}|jj |j|d{x}rzt|jD]+}|jj||j-d}|jj |j|d{x}rz|jj t"k(r|jj%d{ |jj |jdd{r5 |jj |jdd{r5 |jj |jdd{dddd{y777t7#tj$r}|jdd}~wwxYw777#t&$rYwxYw7a#t&$rYjwxYw#|jj t"k(r|jj%d{7 |jj |jdd{7r6 |jj |jdd{7r6n#t&$rYnwxYw |jj |jdd{7w#t&$rYwwxYwwxYw7h#1d{7swYyxYww)a Iterate row-by-row on a result from the database. :param size: if greater than 1, results will be retrieved by chunks of this size from the server (but still yielded row-by-row); this is only available from version 17 of the libpq. z(stream() cannot be used in pipeline modeNrYTF)first)_pgconnpipeline_statusrLProgrammingErrorrHrIrJ_stream_send_gen_stream_fetchone_genrangentuples_txload_rowr;rMrNtransaction_statusACTIVE _try_cancel Exception) r!rBrCr@rZr\resposrOs r#streamzAsyncCursor.streams4 << ' '$$%OP P::??   jjoo))%T)R#'::??43L3LU3S#TTTcT$S[[1E"hh//T^^DDE!E$(::??43L3LU3S#TTTcT<<22f<**00222$(JJOO 55E5B%!%)JJOO 55E5B%"jjood.G.Ge.G.TUUU9   UT?? .''-- . 3% V$!<<22f<**00222$(JJOO 55E5B%!%)JJOO 55E5B%% "jjood.G.Ge.G.TUUU$!=    sANH! N N1H,?H$3H,3H'4A8H,,H*-H,4:N.I/N4.I"I#3III.I/ I- I/ NNN$H,'H,*H,,I?III>NII I*'N)I**N-I// I;8N:I;;N>;N9J< :N.L+.K1 /4L+#L& $L+*N+ L7 4N6L7 7N;.M0)M, *M0/N0 M< 9N;M< <NNNN N NNcTK|jr ||jsyyw)z Iterate across multiple record sets received by the cursor. Multiple record sets are received after using `executemany()` with `!returning=True` or using `execute()` with more than one query in the command. N)r:nextsetr-s r#resultszAsyncCursor.resultss- == ||~  s&(cKt|j |cxkrt|jks'ntd|dt|jd|dkrt|j|z}|j||Sw)a Move to a specific result set. :arg index: index of the result to go to :type index: `!int` More than one result will be available after executing calling `executemany()` or `execute()` with more than one query. `!index` is 0-based and supports negative values, counting from the end, the same way you can index items in a list. The function returns self, so that the result may be followed by a fetch operation. See `results()` for details. zindex z out of range: z result(s) availabler)len_results IndexError_select_current_result)r!indexs r# set_resultzAsyncCursor.set_results DMM""e@c$--.@@s4==/A.BBVW  19 &.E ##E* sB B c K|jd{|j}|j|jkrG|jj |j|j }|xjdz c_|Sy7uw)z Return the next record from the current result set. Return `!None` the result set is finished. :rtype: Row | None, with Row defined by `row_factory` Nr )_fetch_pipeline_check_result_for_fetch_posrcrdrer;r!rjrecords r#fetchonezAsyncCursor.fetchonesp""$$$**, 99s{{ "XX&&tyy$..AF IINIM %sBB A6Bc^K|jd{|j}|s |j}|jj |j t |j |z|j|j}|xj t|z c_|S7w)z Return the next `!size` records from the current result set. `!size` default to `!self.arraysize` if not specified. :rtype: Sequence[Row], with Row defined by `row_factory` N) rxry arraysizerd load_rowsrzminrcr;rq)r!rZrjrecordss r# fetchmanyzAsyncCursor.fetchmanys""$$$**,>>D(($$ IIs499t+S[[94>>  S\!  %sB-B+BB-cK|jd{|j}|jj|j|j |j }|j |_|S7bw)z Return all the remaining records from the current result set. :rtype: Sequence[Row], with Row defined by `row_factory` N)rxryrdrrzrcr;)r!rjrs r#fetchallzAsyncCursor.fetchallsa ""$$$**,(($$TYY T^^LKK  %sA;A9A#A;c|Sr rr-s r# __aiter__zAsyncCursor.__aiter__ s r%c4K|jd{|j}|j|jkrG|jj |j|j }|xjdz c_|Std7w)Nr zno more records to return)rxryrzrcrdrer;StopAsyncIterationr{s r# __anext__zAsyncCursor.__anext__#sw""$$$**, 99s{{ "XX&&tyy$..AF IINIM !<== %sBBBBcdK|jd{|j||y7w)a Move the cursor in the result set to a new position according to mode. If `!mode` is ``'relative'`` (default), `!value` is taken as offset to the current position in the result set; if set to ``'absolute'``, `!value` states an absolute target position. Raise `!IndexError` in case a scroll operation would leave the result set. In this case the position will not change. N)rx_scroll)r!valuemodes r#scrollzAsyncCursor.scroll,s-""$$$ UD! %s 0.0writerc K |jj4d{|jj|j||d{t ||4d{}|dddd{dddd{|jdy77S7?7,#1d{7swY "!%' &* ''' # ' " ''.Sr%rrN))__doc__ __future__rtypesrtypingrrr contextlibrcollections.abcr r r rLr abcrrrrrrrowsrrr_compatrr _cursor_baser_pipeline_asyncrconnection_asyncrTransactionStatusrgrrr%r#rsn #//*3//(00#$*1   $ $|S*3S89|Sr%