U ڲg$@sdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z e rddl!m"Z"ej#j$Z$Gddde defZ%dS)z Psycopg Cursor object. ) annotations) TracebackType)AnyIteratorIterable TYPE_CHECKINGoverload)contextmanager)pq)errors)QueryParams)CopyWriter)RowRowMaker RowFactory)Self)Pipeline) BaseCursor) ConnectioncseZdZdZdZeddddZeddd d dZd d dd d fddZddddZdddddddZddddZe ddddZ e j dddddZ d dd!d"Z dQd d d#d$d%d&d&dd'd(d)Z d*d+d$d,d-dd.d/d0ZdRd d1d2d$d%d&d3d4d5d6d7Zd8dd9d:ZdSd3dd?Zd)rr7rCrApr?rrr executemanyds&     zCursor.executemanyr r6sizeintz Iterator[Row])r7r8r6rIr&c csH|jjrtd|jj"zzj|j |j ||||dd}|j | |rt |D]&}|j||j}|dkrzq|VqZd}q@Wn.tjk r}z|dW5d}~XYnXW5|jjtkr8|jz|j |j ddrqWnt k rYnXz|j |j ddWnt k r6YnXXW5QRXdS)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 modeF)firstrHTN)Z_pgconnZpipeline_statusr<ZProgrammingErrorr9r:Ztransaction_statusACTIVEZ _try_cancelr;Z_stream_fetchone_gen ExceptionZ_stream_send_genrange_txload_rowr2r=r>) rr7r8r6rIrKposZrecr?rrrstreams8      z Cursor.stream Row | NonecCs<|||j|j|j}|dk r8|jd7_|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_posr2)rrecordrrrfetchones zCursor.fetchonerz list[Row])rIr&cCs^|||jst|s$|j}|j|jt|j||jj |j }|jt |7_|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` ) rTrUr1AssertionErrorZ arraysizerO load_rowsrVminntuplesr2len)rrIrecordsrrr fetchmanys zCursor.fetchmanycCs@|||jst|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` ) rTrUr1rYrOrZrVr\r2)rr^rrrfetchalls   zCursor.fetchallc#sRdddfdd }|j}|dkr8qNjd7_|Vq$dS)NrJrS)rQr&csj|jSr)rOrPr2)rQr'rrloadszCursor.__iter__..loadr )rTrUrV)rrarowrr'r__iter__s zCursor.__iter__relativestr)valuemoder&cCs||||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)rTZ_scroll)rrfrgrrrscrolls z Cursor.scrollwriterz Writer | NonezIterator[Copy]) statementr8rjr&c cszJ|jj|j|||W5QRXt||d }|VW5QRXWn.tjk rx}z|dW5d}~XYnX|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. riNr) r9r:r;Z_start_copy_genrr<r=r>Z_select_current_result)rrkr8rjcopyr?rrrrls  z Cursor.copyc CsH|jdk rD|jsD|jjrD|jj|j|jjjddW5QRXdS)NFT)flush)Z_execmany_returningr1r9rEr:r;Z _fetch_genr'rrrrTs zCursor._fetch_pipeline)N)N)r)rd)N)__name__ __module__ __qualname__ __slots__rrr(r.r-propertyr setterr3r@rGrRrXr_r`rcrhr rlrT __classcell__rrr$rrsN!3 rrN)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r r<abcr rrlrrZrowsrrr_compatrrErZ _cursor_baserrrZTransactionStatusrLrrrrrs