j,dZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZdd lmZmZddl m!Z!ddl"m#Z#erddl$m%Z%ej&j'Z'Gdde!defZ(dS)z Psycopg AsyncCursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyoverload)asynccontextmanager) AsyncIteratorIterable)errors)pq)ParamsQueryQueryNoTemplate) AsyncCopy AsyncWriter)AsyncRowFactoryRowRowMaker)SelfTemplate) BaseCursor) AsyncPipeline)AsyncConnectionceZdZdZdZedQdZedRd Zd d dSfd ZdTdZdUdZdVdZe dWdZ e j dXdZ dYdZ e dZd d dd[d&Z ed d dd\d(Z dZd d dd]d*Z d+d,d^d1Z dZd d2d3d_d7Zd`d9Zdad;Zdbd=Zdcddd@ZdedAZdTdBZdfdDZdgdhdIZe dZd dJdidOZdVdPZxZS)j AsyncCursorpsycopg connectionAsyncConnection[Row]cdSNr)selfrs ^/workspace/show-city-briefing-agent/.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__2s  r&exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec>K|d{VdSr")close)r#r4r6r8s r$ __aexit__zAsyncCursor.__aexit__5s. jjllr&c2K|dS)zI Close the current cursor and free associated resources. N)_closer2s r$r<zAsyncCursor.close=s r&c|jS)z9Writable attribute to control how result rows are formed.r.r2s r$r(zAsyncCursor.row_factoryCs   r&cF||_|jr|||_dSdSr")r.pgresult _make_row)r#r(s r$r(zAsyncCursor.row_factoryHs3' = /([..DNNN / /r& RowMaker[Row]c,||Sr"rAr2s r$_make_row_makerzAsyncCursor._make_row_makerNs  &&&r&preparebinaryqueryrparams Params | NonerI bool | NonerJc KdSr"r)r#rKrLrIrJs r$executezAsyncCursor.executeQs sr&rc KdSr"r)r#rKrIrJs r$rPzAsyncCursor.execute[s sr&rc @K |jj4d{V|j|||||d{Vdddd{Vn#1d{VswxYwYn,#tj$r}|dd}~wwxYw|S)z= Execute a query or command to the database. NrH)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)r#rKrLrIrJexs r$rPzAsyncCursor.executedsQ *z        joo%%eVWV%TT                            * * *##D)) ) * s:A28A A2 A**A2-A*.A22BBBF) returning params_seqIterable[Params]r[boolc K |jj4d{Vtjr|jjr6|j||||d{Vn|j4d{V|j||||d{Vdddd{Vn#1d{VswxYwYn5|j||||d{Vdddd{VdS#1d{VswxYwYdS#tj $r}| dd}~wwxYw)zI Execute the same command with a sequence of input data. N) rSrTr is_supported _pipelinerU_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerWrXrY)r#rKr\r[rZs r$ executemanyzAsyncCursor.executemanyxs  *z         -//z+ "joo ::5*iXX $(:#>#>#@#@"&*// $ > >$):y!"!"##*//99%YWW#                              ( * * *##D)) ) *s_EA5D- 6CD- C D- C !9D-E- D77E:D7;EE)E$$E)r rJsizergintAsyncIterator[Row]c K|jjrtjd|jj4d{V |j|||||d{Vd}|j||d{Vx}rqt|j D]%}|j ||j WV&d}|j||d{Vx}qn,#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 modeNrfTF)first)_pgconnpipeline_statusrWProgrammingErrorrSrTrU_stream_send_gen_stream_fetchone_genrangentuples_txload_rowrDrXrYtransaction_statusACTIVE _try_cancel Exception) r#rKrLrJrgrkresposrZs r$streamzAsyncCursor.streams < ' Q$%OPP P:?         joo))%T)RR#':??43L3LU3S3S#T#TTTTTTTTc"$S[11EE"h//T^DDDDDDD!E$(:??43L3LU3S3S#T#TTTTTTTTc"? . . .''--- .<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$!=                              sMCDH:E&D;;EH:4M8A)G"!M" G/,M.G//M34H)'M) H63M5H66M:5L10A)KL1 K' $L1&K' 'L1+4L L1 L- *L1,L- -L11M MMAsyncIterator[Self]cRK|jr |WV|sdSdS)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. TN)rCnextsetr2s r$resultszAsyncCursor.resultssF =   ||~~E    r&indexc&Kt|j |cxkrt|jks*ntd|dt|jd|dkrt|j|z}|||S)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#rs r$ set_resultzAsyncCursor.set_results DM"""e@@@@c$-.@.@@@@@WWWs4=/A/AWWW  199 &&.E ##E*** r& Row | NonecK|d{V|}|j|jkr7|j|j|j}|xjdz c_|SdS)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_posrrrsrtrDr#ryrecords r$fetchonezAsyncCursor.fetchones""$$$$$$$$$**,, 9s{ " "X&&ty$.AAF IINIIMtr&r list[Row]c4K|d{V|}|s|j}|j|jt |j|z|j|j}|xjt|z c_|S)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) rr arraysizers load_rowsrminrrrDr)r#rgryrecordss r$ fetchmanyzAsyncCursor.fetchmanys""$$$$$$$$$**,, ">D($$ Is49t+S[994>   S\\! r&cK|d{V|}|j|j|j|j}|j|_|S)z Return all the remaining records from the current result set. :rtype: Sequence[Row], with Row defined by `row_factory` N)rrrsrrrrrD)r#ryrs r$fetchallzAsyncCursor.fetchallsj ""$$$$$$$$$**,,($$TY T^LLK r&c|Sr"rr2s r$ __aiter__zAsyncCursor.__aiter__ s r&rcK|d{V|}|j|jkr7|j|j|j}|xjdz c_|Std)Nr zno more records to return)rrrrrrsrtrDStopAsyncIterationrs r$ __anext__zAsyncCursor.__anext__#s""$$$$$$$$$**,, 9s{ " "X&&ty$.AAF IINIIM !<===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)r_scroll)r#rrs r$scrollzAsyncCursor.scroll,sF""$$$$$$$$$ UD!!!!!r&writer statementrAsyncWriter | NoneAsyncIterator[AsyncCopy]cK |jj4d{V|j|||d{Vt ||4d{V}|WVdddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYn,#t j$r}|dd}~wwxYw|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) rSrTrU_start_copy_genrrWrXrYr)r#rrLrcopyrZs r$rzAsyncCursor.copy:s *z        jood&:&:9f&M&MNNNNNNNNN$T&999TJJJJ                             * * *##D)) ) * ##A&&&&&s_B/A B"A:(B: B BB B B/ B''B/*B'+B//C>CCc8K|jdur|js|jjr{|jj4d{V|j|jjdd{Vdddd{VdS#1d{VswxYwYdSdSdSdS)NFT)flush)_execmany_returningrCrSrarTrU _fetch_genr2s r$rzAsyncCursor._fetch_pipelineRsk  $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+)r0r)r4r5r6r7r8r9r0r:)r0r:)r0r))r(r)r0r:)r0rEr") rKrrLrMrIrNrJrNr0r)rKrrIrNrJrNr0r) rKrrLrMrIrNrJrNr0r)rKrr\r]r[r^r0r:) rKrrLrMrJrNrgrhr0ri)r0r|)rrhr0r)r0r)r)rgrhr0r)r0r)r0r)r)rrhrrr0r:)rrrLrMrrr0r)__name__ __module__ __qualname__ __slots__rr%r3r=r<propertyr(setterrGrPrer{rrrrrrrrrrr __classcell__)r/s@r$rrsJI ===X=    X 48 BBBBBBBB !!!X!//// ''''!% $" X $" X!% $" *PU******B!%0 # 000000d    4 &    >>>> " " " " "!%' &* ''''''.SSSSSSSSr&rr'N))__doc__ __future__rtypesrtypingrrr contextlibrcollections.abcr r r rWr abcrrrrrrrowsrrr_compatrr _cursor_baser_pipeline_asyncrconnection_asyncrTransactionStatusrvrrr&r$rs #"""""//////////******33333333//////////((((((((0000000000########$$$$$$******2111111  $|S|S|S|S|S*3S89|S|S|S|S|Sr&