qj%dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z e rddl!m"Z"ejFjHZ$Gdde defZ%y)z Psycopg AsyncCursor object. ) annotations) TracebackType)Any AsyncIteratorIterable TYPE_CHECKINGoverload)asynccontextmanager)pq)errors)QueryParams) AsyncCopy AsyncWriter)RowRowMakerAsyncRowFactory)Self)Pipeline) BaseCursor)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 U/var/www/html/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 tjr|jj4d{|jj}|r4|jj |j |||d{dddd{s|jj4d{|jj4d{|jj |j |||d{dddd{dddd{y|jj4d{|jj |j|||d{dddd{yy7_77 #1d{7swYxYw7777#1d{7swYxYw7#1d{7swYyxYw77k7]#1d{7swYyxYw#tj$r}|jdd}~wwxYww)zI Execute the same command with a sequence of input data. N) r is_supportedr@rA _pipelinerB_executemany_gen_pipelinepipeline_executemany_gen_no_pipelinerDrErF)rrG params_seqrKprIs r! executemanyzAsyncCursor.executemanyds *$$& ::??? ,,A"jjoo ::5*iX+?#zz2244djjooo"jjoo ::5*iX7Fo444  ::???**//99%YW+?? ++??? 5o7Fooo4444 ++??? *##D) ) *sH,/HFHAF?FF HF$H4F15H8GF3G0F9F5F9 GF7G H'G(H,H,-HG%H 0G+;G'<G+ H G) HH,HFHF.!F$ "F.) H3G5F97G9G ?GG GHG"G G"H!H,"H'G+)H+G=1G4 2G=9H<H,=HH)H$$H))H,r r>sizec *K|jjrtjd|jj 4d{ |jj |j||||d{d}|jj |j|d{rtt|D]1}|jj||j}|n|3d}|jj |j|d{rt|jjt k(r|jj#d{ |jj |jdd{r5 |jj |jdd{r5 |jj |jdd{dddd{y777l7#tj$r}|jdd}~wwxYw777#t$$rYwxYw7a#t$$rYjwxYw#|jjt 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 modeNrUTF)first)_pgconnpipeline_statusrDProgrammingErrorr@rArB_stream_send_gen_stream_fetchone_genrange_txload_rowr9rErFtransaction_statusACTIVE _try_cancel Exception) rrGrHr>rVrXposrecrIs r!streamzAsyncCursor.streams << ' '$$%OP P::???" jjoo))%T)R JJOOD,E,Ee,LMMM$T{"hh//T^^D;!! + "E !JJOOD,E,Ee,LMMM<<22f<**00222$(JJOO 55E5B%!%)JJOO 55E5B%"jjood.G.Ge.G.TUUUC#??NM?? .''-- .3% V$!<<22f<**00222$(JJOO 55E5B%!%)JJOO 55E5B%% "jjood.G.Ge.G.TUUU$!=%#???sANH N M>1H$?H3H$3H4A2H$&H"'H$,:M>&I'M>,.II3III.I'I%I' NM;NH$H$"H$$I 7II  I6M>II I"M>!I""M>%I'' I30M>2I33M>6;M81J4 2M88.L#&K) '4L#L L#"M8# L/ ,M8.L/ /M83.M(!M$ "M('M8( M4 1M83M4 4M88M>;N>NN N NcK|jd{|j|jj|j|j }||xjdz c_|S7]w)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_fetchr_r`_posr9)rrecords r!fetchonezAsyncCursor.fetchonesa""$$$ $$&""499dnn=   IINI %sA6A4AA6cK|jd{|j|jsJ|s |j}|jj |j t|j |z|jj|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) rirjr8 arraysizer_ load_rowsrkminntuplesr9len)rrVrecordss r! fetchmanyzAsyncCursor.fetchmanys""$$$ $$&}}}>>D(($$ II  D $--"7"7 8 NN  S\!  %sCCB-Cc>K|jd{|j|jsJ|jj |j |jj |j}|jj |_|S7w)z Return all the remaining records from the current recordset. :rtype: Sequence[Row], with Row defined by `row_factory` N)rirjr8r_rprkrrr9)rrts r!fetchallzAsyncCursor.fetchallsw ""$$$ $$&}}}(($$TYY 0E0Et~~VMM))  %sBBBBcKjd{jdfd } |j}|yxjdz c_|07Kw)NcPjj|jSr)r_r`r9)rers r!loadz#AsyncCursor.__aiter__..loads88$$S$..9 9r#r )reintreturn Row | None)rirjrk)rrzrows` r! __aiter__zAsyncCursor.__aiter__s`""$$$ $$& :tyy/C{ IINII  %sA%A#A A%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)ri_scroll)rvaluemodes r!scrollzAsyncCursor.scrolls-""$$$ UD! %s 0.0writerc K |jj4d{|jj|j||d{dddd{t ||4d{}|dddd{|jdy77S7E#1d{7swYUxYw7F73#1d{7swYCxYw#t j $r}|jdd}~wwxYww)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) r@rArB_start_copy_genrrDrErF_select_current_result)r statementrHrcopyrIs r!rzAsyncCursor.copy s *zzjjood&:&:9f&MNNN'!f55 65 ##A&'N'65555 *##D) ) *sDCB#C/B)B%B) C!B'"C7B>8C;C C C CD#C%B)'C)B;/B2 0B;7CCCC CCD*C;;DDcK|jdur|js|jjrw|jj4d{|jj |jjj dd{dddd{yyyy7_77 #1d{7swYyxYww)NFT)flush)_execmany_returningr8r@rNrArB _fetch_genr+s r!rizAsyncCursor._fetch_pipeline#s  $ $E 1MM $$zzjjoodjj&:&:&E&ED&E&QRRR'%" 2'R's[A C B-CAB3B/B3 C$B1% C/B31C3C9B< :CC)r zAsyncConnection[Row])r AsyncConnection[Any]r%AsyncRowFactory[Row])r rr%zAsyncRowFactory[Row] | None)r|r)r/ztype[BaseException] | Noner0zBaseException | Noner1zTracebackType | Noner|None)r|r)r|r)r%rr|r)r|z RowMaker[Row]r) rGrrH Params | Noner= bool | Noner>rr|r)rGrrRzIterable[Params]rKboolr|r) rGrrHrr>rrVr{r|AsyncIterator[Row])r|r})r)rVr{r| list[Row])r|r)r|r)relative)rr{rstrr|r)rrrHrrzAsyncWriter | Noner|zAsyncIterator[AsyncCopy])__name__ __module__ __qualname__ __slots__r r"r,r2r.propertyr%setterr;rJrTrgrmrurwrrr rri __classcell__)r)s@r!rrsJI ==  . @T  48 B(B1 B,&%    !!// ' !% $"      2 * *% *  *  *J!%5 # 555  5  5 5n ,   "!%' &* ''' # ' " ''.Sr#rrN)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r rDabcrrrrrrowsrrr_compatrrNr _cursor_baserconnection_asyncrTransactionStatusrbrrr#r!rsh #HH*(00$1   $ $NS*3S89NSr#