pj%dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z e rddl!m"Z"ej#j$Z$Gdde defZ%dS)z Psycopg AsyncCursor object. ) annotations) TracebackType)Any AsyncIteratorIterable TYPE_CHECKINGoverload)asynccontextmanager)pq)errors)QueryParams) AsyncCopy AsyncWriter)RowRowMakerAsyncRowFactory)Self)Pipeline) BaseCursor)AsyncConnectioncDeZdZdZdZedGdZedHd Zd d dIfd ZdJdZdKdZdLdZe dMdZ e j dNdZ dOdZ dPd d ddQd&Z d'd(dRd-Z dPd d.d/dSd3ZdTd5ZdUdVd8ZdWd9ZdXd:ZdYdZd?Ze dPd d@d[dEZdLdFZxZS)\ AsyncCursorpsycopg connectionAsyncConnection[Row]cdSNr)selfrs d/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/psycopg/cursor_async.py__init__zAsyncCursor.__init__ s:=#AsyncConnection[Any] row_factoryAsyncRowFactory[Row]cdSr r)r!rr&s r"r#zAsyncCursor.__init__#s  sr$N)r&AsyncRowFactory[Row] | Nonecft||p|j|_dSr )superr#r& _row_factory)r!rr& __class__s r"r#zAsyncCursor.__init__(s3 $$$'A:+Ar$returnrc K|Sr rr!s r" __aenter__zAsyncCursor.__aenter__1s  r$exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec>K|d{VdSr )close)r!r2r4r6s r" __aexit__zAsyncCursor.__aexit__4s. jjllr$c2K|dS)zI Close the current cursor and free associated resources. N)_closer0s r"r:zAsyncCursor.close<s r$c|jS)z9Writable attribute to control how result rows are formed.r,r0s r"r&zAsyncCursor.row_factoryBs   r$cF||_|jr|||_dSdSr )r,pgresult _make_row)r!r&s r"r&zAsyncCursor.row_factoryGs3' = /([..DNNN / /r$ RowMaker[Row]c,||Sr r?r0s r"_make_row_makerzAsyncCursor._make_row_makerMs  &&&r$preparebinaryqueryrparams Params | NonerG bool | NonerHc @K |jj4d{V|j|||||d{Vdddd{Vn#1d{VswxYwYn,#tj$r}|dd}~wwxYw|S)z= Execute a query or command to the database. NrF)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)r!rIrJrGrHexs r"executezAsyncCursor.executePsQ *z        joo%%eVWV%TT                            * * *##D)) ) * s:A28A A2 A**A2-A*.A22BBBF) returning params_seqIterable[Params]rWboolc K tjr/|jj4d{V|jj}|r5|j||||d{Vdddd{Vn#1d{VswxYwY|s|j4d{V|jj4d{V|j||||d{Vdddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYdSdSdS|jj4d{V|j||||d{Vdddd{VdS#1d{VswxYwYdS#tj $r}| dd}~wwxYw)zI Execute the same command with a sequence of input data. N) r is_supportedrNrO _pipelinerP_executemany_gen_pipelinepipeline_executemany_gen_no_pipelinerRrSrT)r!rIrXrWprUs r" executemanyzAsyncCursor.executemanydsn *$&&  :? ,A"joo ::5*iXX#z2244djo"joo ::5*iXX  :?*//99%YWW * * *##D)) ) *s&F5AB.F5 B  F5 B $F52D06D ;D0 D D0D D0F50 D::F5=D:>F5F56F"F5" F,,F5/F,0F55GGGr rHsizerdintAsyncIterator[Row]c K|jjrtjd|jj4d{V |j|||||d{Vd}|j||d{Vrpt|D]+}|j ||j }|n|WV,d}|j||d{Vpn,#tj $r}|dd}~wwxYw|jjt kr|jd{V |j|dd{Vr5 |j|dd{V5n#t$$rYnwxYw |j|dd{Vn #t$$rYnwxYwn#|jjt kr|jd{V |j|dd{Vr5 |j|dd{V5n#t$$rYnwxYw |j|dd{Vw#t$$rYwwxYwwxYwdddd{VdS#1d{VswxYwYdS)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 modeNrcTF)first)_pgconnpipeline_statusrRProgrammingErrorrNrOrP_stream_send_gen_stream_fetchone_genrange_txload_rowrBrSrTtransaction_statusACTIVE _try_cancel Exception) r!rIrJrHrdrhposrecrUs r"streamzAsyncCursor.streams < ' Q$%OPP P:?# # # # # # # # " joo))%T)RR JOOD,E,Ee,L,LMMMMMMMM"$T{{"""h//T^DD;!E! !E !JOOD,E,Ee,L,LMMMMMMMM"? . . .''--- .<2f<<*00222222222$(JOO 55E5BB%%!!%)JOO 55E5BB%%!% "jood.G.Ge.G.T.TUUUUUUUUUU$!=4<2f<<*00222222222$(JOO 55E5BB%%!!%)JOO 55E5BB%%!% "jood.G.Ge.G.T.TUUUUUUUUUU$!=%# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # sMCDH7D=#D88D==H74M5A)GM G,)M+G,,M04H&$M& H30M2H33M75L.-A)KL. K$ !L.#K$ $L.(4LL. L* 'L.)L* *L..M MM Row | NonecK|d{V||j|j|j}||xjdz c_|S)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_fetchrorp_posrB)r!records r"fetchonezAsyncCursor.fetchoness""$$$$$$$$$ $$&&&""49dn==   IINII r$r list[Row]cPK|d{V||jsJ|s|j}|j|jt|j|z|jj|j }|xjt|z c_|S)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) rzr{rA arraysizero load_rowsr|minntuplesrBlen)r!rdrecordss r" fetchmanyzAsyncCursor.fetchmanys""$$$$$$$$$ $$&&&}} ">D($$ I  D $-"7 8 8 N   S\\! r$cK|d{V||jsJ|j|j|jj|j}|jj|_|S)z Return all the remaining records from the current recordset. :rtype: Sequence[Row], with Row defined by `row_factory` N)rzr{rArorr|rrB)r!rs r"fetchallzAsyncCursor.fetchalls~ ""$$$$$$$$$ $$&&&}}($$TY 0Et~VVM) r$cKd{Vdfd } |j}|dSxjdz c_|WV*) Nrurer.rxcDj|jSr )rorprB)rur!s r"loadz#AsyncCursor.__aiter__..loads8$$S$.99 9r$Tr )rurer.rx)rzr{r|)r!rrows` r" __aiter__zAsyncCursor.__aiter__s""$$$$$$$$$ $$&&& : : : : : : $ty//C{ IINIIIIII  r$relativevaluemodestrcjK|d{V|||dS)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)rz_scroll)r!rrs r"scrollzAsyncCursor.scrollsF""$$$$$$$$$ UD!!!!!r$writer statementrAsyncWriter | NoneAsyncIterator[AsyncCopy]cK |jj4d{V|j|||d{Vdddd{Vn#1d{VswxYwYt ||4d{V}|WVdddd{Vn#1d{VswxYwYn,#t j$r}|dd}~wwxYw|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) rNrOrP_start_copy_genrrRrSrT_select_current_result)r!rrJrcopyrUs r"rzAsyncCursor.copy s; *z O O O O O O O Ojood&:&:9f&M&MNNNNNNNNN O O O O O O O O O O O O O O O O O O O O O O O O O O O!f555                                    * * *##D)) ) * ##A&&&&&s^B/5A B/ A''B/*A'+B/B B/ B''B/*B'+B//C>CCc8K|jdur|js|jjr{|jj4d{V|j|jjdd{Vdddd{VdS#1d{VswxYwYdSdSdSdS)NFT)flush)_execmany_returningrArNr]rOrP _fetch_genr0s r"rzzAsyncCursor._fetch_pipeline#sk  $E 1 1M 2 $ 2z S S S S S S S Sjoodj&:&E&ED&E&Q&QRRRRRRRRR S S S S S S S S S S S S S S S S S S S S S S S S S S S S S S 2 1 1 1 1 1s?B B B )rr)rr%r&r')rr%r&r))r.r)r2r3r4r5r6r7r.r8)r.r8)r.r')r&r'r.r8)r.rCr ) rIrrJrKrGrLrHrLr.r)rIrrXrYrWrZr.r8) rIrrJrKrHrLrdrer.rf)r.rx)r)rdrer.r)r.r)r.rf)r)rrerrr.r8)rrrJrKrrr.r)__name__ __module__ __qualname__ __slots__r r#r1r;r:propertyr&setterrErVrbrwr~rrrrr rrz __classcell__)r-s@r"rrsoJI ===X=    X 48 BBBBBBBB !!!X!//// '''' !% $" 2 * * * * * *J!%5 # 555555n    ,         " " " " "!%' &* ''''''.SSSSSSSSr$rr%N)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r rRabcrrrrrrowsrrr_compatrr]r _cursor_baserconnection_asyncrTransactionStatusrrrrr$r"rs #"""""HHHHHHHHHHHHHH******((((((((0000000000$$$$$$2111111  $NSNSNSNSNS*3S89NSNSNSNSNSr$