j%dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z erddl!m"Z"ejFjHZ$GddedefZ%y)z Psycopg AsyncCursor object. ) annotations) TracebackType) TYPE_CHECKINGAny AsyncIteratorIterableoverload)asynccontextmanager)errors)pq)ParamsQuery) AsyncCopy AsyncWriter)AsyncRowFactoryRowRowMaker)Self) BaseCursor) AsyncPipeline)AsyncConnectionceZdZdZdZeddZe ddZdd d fdZd!dZ d"d Zd#d Ze d$d Z e jd%d Z d&d Z d'ddd d(dZ dd d)dZ d'ddd d*dZd+dZd,d-dZd.dZd/dZd0d1dZe d'dd d2dZd#dZxZS)3 AsyncCursorpsycopgcyNr)self connections [/var/www/html/kelly/kelly-backend/venv/lib/python3.12/site-packages/psycopg/cursor_async.py__init__zAsyncCursor.__init__ s:=cyrr)rr row_factorys r!r"zAsyncCursor.__init__#s r#N)r%cNt|||xs |j|_yr)superr"r% _row_factory)rr r% __class__s r!r"zAsyncCursor.__init__(s% $'A:+A+Ar#cK|Swrrrs r! __aenter__zAsyncCursor.__aenter__1s  sc@K|jd{y7wr)close)rexc_typeexc_valexc_tbs r! __aexit__zAsyncCursor.__aexit__4s jjls c,K|jyw)zI Close the current cursor and free associated resources. N)_closer+s r!r.zAsyncCursor.close<s sc|jS)z9Writable attribute to control how result rows are formed.r(r+s r!r%zAsyncCursor.row_factoryBs   r#cF||_|jr|||_yyr)r(pgresult _make_row)rr%s r!r%zAsyncCursor.row_factoryGs"' ==(.DN r#c$|j|Srr6r+s r!_make_row_makerzAsyncCursor._make_row_makerMs  &&r#preparebinaryc 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)rqueryparamsr=r>exs r!executezAsyncCursor.executePs *zzjjoo%%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) r@rAr is_supported _pipelinerB_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerDrErF)rrG params_seqrKrIs r! executemanyzAsyncCursor.executemanydsC  *zz --/zz++"jjoo ::5*iX $(::#>#>#@#@"&**// $ > >$):y!"#$A#@#@**//99%YW#'  $A$A#@#@#@#'( *##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 modeNrTTF)first)_pgconnpipeline_statusrDProgrammingErrorr@rArB_stream_send_gen_stream_fetchone_genrangentuples_txload_rowr9rErFtransaction_statusACTIVE _try_cancel Exception) rrGrHr>rUrWresposrIs r!streamzAsyncCursor.streams << ' '$$%OP P::??? jjoo))%T)R#'::??43L3LU3S#TTTcT$S[[1"hh//T^^DD 2!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 NNc K|jd{|j}|j|jkrG|jj |j|j }|xjdz c_|Sy7uw)z Return the next record from the current recordset. Return `!None` the recordset is finished. :rtype: Row | None, with Row defined by `row_factory` Nr _fetch_pipeline_check_result_for_fetch_posr^r_r`r9)rrerecords 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 recordset. `!size` default to `!self.arraysize` if not specified. :rtype: Sequence[Row], with Row defined by `row_factory` N) rjrk arraysizer_ load_rowsrlminr^r9len)rrUrerecordss 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 recordset. :rtype: Sequence[Row], with Row defined by `row_factory` N)rjrkr_rqrlr^r9)rrerts r!fetchallzAsyncCursor.fetchallsa ""$$$**,(($$TYY T^^LKK  %sA;A9A#A;c\K|jd{|j}|j|jkre|jj |j|j }|xjdz c_||j|jkrdyy7w)Nr ri)rrerows r! __aiter__zAsyncCursor.__aiter__s}""$$$**,ii#++%((##DIIt~~>C IINIIii#++% %sB,B*BB,(B,cdK|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)rj_scroll)rvaluemodes r!scrollzAsyncCursor.scrolls-""$$$ UD! %s 0.0writerc K |jj4d{|jj|j||d{t ||4d{}|dddd{dddd{|jdy77S7?7,#1d{7swYrrr)rGrrRzIterable[Params]rKboolrr) rGrrHrr>rrUintrAsyncIterator[Row])rz Row | None)r)rUrr list[Row])rr)rr)relative)r}rr~strrr)rrrHrrzAsyncWriter | NonerzAsyncIterator[AsyncCopy])__name__ __module__ __qualname__ __slots__r r"r,r2r.propertyr%setterr;rJrSrgrnrurwrzrr rrj __classcell__)r)s@r!rrsJI ==  . @T  48 B(B1 B,&%    !!// ' !% $"      *PU**(8*HL* *B!%0 # 000  0  0 0d &  "!%' &* ''' # ' " ''.Sr#rrN)&__doc__ __future__rtypesrtypingrrrrr contextlibr r rDr abcrrrrrrowsrrr_compatr _cursor_baser_pipeline_asyncrconnection_asyncrTransactionStatusrbrrr#r!rsh #HH*(00$*1   $ $}S*3S89}Sr#