jl+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 Cursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyoverload)contextmanager)IterableIterator)errors)pq)ParamsQueryQueryNoTemplate)CopyWriter)Row RowFactoryRowMaker)SelfTemplate)Pipeline) BaseCursor) ConnectionceZdZdZdZedQdZedRd 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)jCursorpsycopg connectionConnection[Row]cdSNr)selfrs Y/DATA/AppData/hermes/projects/honcho/.venv/lib/python3.11/site-packages/psycopg/cursor.py__init__zCursor.__init__$s58SConnection[Any] row_factoryRowFactory[Row]cdSr"r)r#rr(s r$r%zCursor.__init__'s  sr&N)r(RowFactory[Row] | Nonecft||p|j|_dSr")superr%r( _row_factory)r#rr( __class__s r$r%zCursor.__init__,s3 $$$'A:+Ar&returnrc|Sr"rr#s r$ __enter__zCursor.__enter__2 r&exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec.|dSr")close)r#r5r7r9s r$__exit__zCursor.__exit__5s r&c.|dS)zI Close the current cursor and free associated resources. N)_closer2s r$r=z Cursor.close=s r&c|jS)z9Writable attribute to control how result rows are formed.r.r2s r$r(zCursor.row_factoryCs   r&cF||_|jr|||_dSdSr")r.pgresult _make_row)r#r(s r$r(zCursor.row_factoryHs3' = /([..DNNN / /r& RowMaker[Row]c,||Sr"rBr2s r$_make_row_makerzCursor._make_row_makerNs  &&&r&preparebinaryqueryrparams Params | NonerJ bool | NonerKcdSr"r)r#rLrMrJrKs r$executezCursor.executeQs sr&rcdSr"r)r#rLrJrKs r$rQzCursor.execute[s sr&rc   |jj5|j|||||dddn #1swxYwYn,#tj$r}|dd}~wwxYw|S)z= Execute a query or command to the database. rIN)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)r#rLrMrJrKexs r$rQzCursor.executeds *   %%eVWV%TT                * * *##D)) ) * s: A2A  A AAAAB'A<<BF) returning params_seqIterable[Params]r\boolc h |jj5tjr|jjr0|j||||n|j5|j||||dddn #1swxYwYn/|j||||ddddS#1swxYwYdS#tj $r}| dd}~wwxYw)zI Execute the same command with a sequence of input data. N) rTrUr is_supported _pipelinerV_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerXrYrZ)r#rLr]r\r[s r$ executemanyzCursor.executemanyxs  *  (**z+   ::5*iXX "Z88:: JOO $ > >$):y!"!"JOO99%YWW#                  ( * * *##D)) ) *s_ DA)C;70B3' C;3B7 7C;:B7 ;3C;. D;C??DC?DD1D,,D1r rKsizerhint Iterator[Row]c #K|jjrtjd|jj5 |j|||||d}|j||x}rjt|j D]$}|j ||j V%d}|j||x}jn,#tj$r}|dd}~wwxYw|jjt"kr|j |j|dr/ |j|d/n#t&$rYnwxYw |j|dn#t&$rYnwxYwn#|jjt"kr|j |j|dr/ |j|d/n#t&$rYnwxYw |j|dw#t&$rYwwxYwwxYwddddS#1swxYwYdS)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 modergTFN)first)_pgconnpipeline_statusrXProgrammingErrorrTrUrV_stream_send_gen_stream_fetchone_genrangentuples_txload_rowrErYrZtransaction_statusACTIVE _try_cancel Exception) r#rLrMrKrhrlresposr[s r$streamz Cursor.streams < ' Q$%OPP P Z_    ))%T)RR!Z__T-F-Fu-M-MNNNc"$S[11EE"h//T^DDDDDD!E"Z__T-F-Fu-M-MNNNc"? . . .''--- .<2f<<J**,,,"jood.G.Ge.G.T.TUU! #jood.G.Ge.G.T.TUU!$  (A(A(A(N(NOOOO$=4<2f<<J**,,,"jood.G.Ge.G.T.TUU! #jood.G.Ge.G.T.TUU!$  (A(A(A(N(NOOOO$=                  sK7C C>=H>D' D""D''H*.K7AF76K77 GK7GK7.G76K77 HK7HK7/K'8AJK' J# K'"J# #K''.KK' K# K'"K# #K''K77K;>K;Iterator[Self]c#PK|jr |V|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)rDnextsetr2s r$resultszCursor.resultssD =   ||~~E    r&indexc"t|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_resultzCursor.set_results DM"""e@@@@c$-.@.@@@@@WWWs4=/A/AWWW  199 &&.E ##E*** r& Row | Nonec||}|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` r N)_fetch_pipeline_check_result_for_fetch_posrsrtrurEr#rzrecords r$fetchonezCursor.fetchonesj **,, 9s{ " "X&&ty$.AAF IINIIMtr&r list[Row]c$||}|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` ) rr arraysizert load_rowsrminrsrEr)r#rhrzrecordss r$ fetchmanyzCursor.fetchmanys **,, ">D($$ Is49t+S[994>   S\\! r&c||}|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` )rrrtrrrsrE)r#rzrs r$fetchallzCursor.fetchallsT **,,($$TY T^LLK r&c|Sr"rr2s r$__iter__zCursor.__iter__r4r&rc||}|j|jkr7|j|j|j}|xjdz c_|Std)Nr zno more records to return)rrrrsrtrurE StopIterationrs r$__next__zCursor.__next__!sr **,, 9s{ " "X&&ty$.AAF IINIIM7888r&relativevaluemodestrcZ||||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$scrollz Cursor.scroll*s0  UD!!!!!r&writer statementr Writer | NoneIterator[Copy]c#K |jj5|j|||t ||5}|Vdddn #1swxYwYdddn #1swxYwYn,#t j$r}|dd}~wwxYw|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. rNr) rTrUrV_start_copy_genrrXrYrZr)r#rrMrcopyr[s r$rz Cursor.copy8sP *    4 4Y G GHHH$v...$JJJ                 * * *##D)) ) * ##A&&&&&s_ BAA8A! A8!A% %A8(A% )A8, B8A<<B?A<BB-B((B-c|jdurr|jsm|jjrc|jj5|j|jjdddddS#1swxYwYdSdSdSdS)NFT)flush)_execmany_returningrDrTrbrUrV _fetch_genr2s r$rzCursor._fetch_pipelinePs  $E 1 1] 2 $ 2 M M  4 ? ?d ? K KLLL M M M M M M M M M M M M M M M M M M 2 1 1 1 1 1s9A//A36A3)rr )rr'r(r))rr'r(r+)r0r)r5r6r7r8r9r:r0r;)r0r;)r0r))r(r)r0r;)r0rFr") rLrrMrNrJrOrKrOr0r)rLrrJrOrKrOr0r) rLrrMrNrJrOrKrOr0r)rLrr]r^r\r_r0r;) rLrrMrNrKrOrhrir0rj)r0r})rrir0r)r0r)r)rhrir0r)r0r)r0r)r)rrirrr0r;)rrrMrNrrr0r)__name__ __module__ __qualname__ __slots__rr%r3r>r=propertyr(setterrHrQrfr|rrrrrrrrrrr __classcell__)r/s@r$rr sJI 888X8    X UYBBBBBBBB  !!!X!//// ''''!% $" X $" X!% $" *PU******B!%. # ......`    4 &    9999 " " " " "!%' !% '''''^'.MMMMMMMMr&rr'N))__doc__ __future__rtypesrtypingrrr contextlibrcollections.abcr r r rXr abcrrrrrrrowsrrr_compatrrrbr _cursor_baserrrTransactionStatusrwrrr&r$rs #"""""//////////%%%%%%........//////////++++++++++########$$$$$$'&&&&&&  $wMwMwMwMwMZ)3. /wMwMwMwMwMr&