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 AsyncCursor object. ) annotations) TracebackType)Any AsyncIteratorIterable TYPE_CHECKINGoverload)asynccontextmanager)pq)errors)QueryParams) AsyncCopy AsyncWriter)RowRowMakerAsyncRowFactory)Self)Pipeline) BaseCursor)AsyncConnectioncseZdZdZdZeddddZeddd 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|jj4IdH"|j||||IdHW5QIdHRXWn0t j k r,}z| dW5d}~XYnXdS)zI Execute the same command with a sequence of input data. N) r is_supportedr9r: _pipeliner;Z_executemany_gen_pipelineZpipelineZ_executemany_gen_no_pipeliner<r=r>)rr7rCrApr?rrr executemanyds&  & , zAsyncCursor.executemanyr r6sizeintzAsyncIterator[Row])r7r8r6rIr&c Csv|jjrtd|jj4IdHBzzv|j |j ||||dIdHd}|j | |IdHrt |D]&}|j||j}|dkrq|Vqnd}qNWn.tjk r}z|dW5d}~XYnXW5|jjtkr`|jIdHz"|j |j ddIdHrqWnt k r(YnXz|j |j ddIdHWnt k r^YnXXW5QIdHRXdS)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 modeNF)firstrHT)Z_pgconnZpipeline_statusr<ZProgrammingErrorr9r:Ztransaction_statusACTIVEZ _try_cancelr;Z_stream_fetchone_gen ExceptionZ_stream_send_genrange_txload_rowr2r=r>) rr7r8r6rIrKposZrecr?rrrstreams<      zAsyncCursor.stream Row | NonecsB|IdH||j|j|j}|dk r>|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 zAsyncCursor.fetchonerz list[Row])rIr&csd|IdH||js t|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` N) rTrUr1AssertionErrorZ arraysizerO load_rowsrVminntuplesr2len)rrIrecordsrrr fetchmanys zAsyncCursor.fetchmanycsF|IdH||js t|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) rTrUr1rYrOrZrVr\r2)rr^rrrfetchalls   zAsyncCursor.fetchallcsXIdHdddfdd }|j}|dkr>qTjd7_|Vq*dS)NrJrS)rQr&csj|jSr)rOrPr2)rQr'rrloadsz#AsyncCursor.__aiter__..loadr )rTrUrV)rrarowrr'r __aiter__s zAsyncCursor.__aiter__relativestr)valuemoder&cs|IdH|||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 zAsyncCursor.scrollwriterzAsyncWriter | NonezAsyncIterator[AsyncCopy]) statementr8rjr&c Cszl|jj4IdH |j|||IdHW5QIdHRXt||d4IdH }|VW5QIdHRXWn.tjk r}z|dW5d}~XYnX|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrir) r9r:r;Z_start_copy_genrr<r=r>Z_select_current_result)rrkr8rjcopyr?rrrrl s *zAsyncCursor.copyc s\|jdk rX|jsX|jjrX|jj4IdH$|j|jjjddIdHW5QIdHRXdS)NFT)flush)Z_execmany_returningr1r9rEr:r;Z _fetch_genr'rrrrT#szAsyncCursor._fetch_pipeline)N)N)r)rd)N)__name__ __module__ __qualname__ __slots__rrr(r.r-propertyr setterr3r@rGrRrXr_r`rcrhr rlrT __classcell__rrr$rrsN %7 rrN)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r r<abcr rrlrrZrowsrrr_compatrrErZ _cursor_baserZconnection_asyncrZTransactionStatusrLrrrrrs