o "4g%@sdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z e rbddl!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)AsyncConnectioncs"eZdZdZdZedWddZedXd dZd d dYfddZdZddZd[ddZd\ddZe d]ddZ e j d^d dZ d_d"d#Z d`d d d$dad,d-Z d.d/dbd4d5Z d`d d6d7dcd;d<Zddd>d?ZdedfdBdCZdgdDdEZdhdFdGZdidjdLdMZe d`d dNdkdSdTZd\dUdVZZS)l AsyncCursorpsycopg connectionAsyncConnection[Row]cCdSNr)selfrrr]/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/cursor_async.py__init__ szAsyncCursor.__init__AsyncConnection[Any] row_factoryAsyncRowFactory[Row]cCrrrrrr#rrr r!#sN)r#AsyncRowFactory[Row] | Nonecst||p |j|_dSr)superr!r# _row_factoryr% __class__rr r!(s returnrcs|Srrrrrr __aenter__1szAsyncCursor.__aenter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonecs|IdHdSr)close)rr.r0r2rrr __aexit__4szAsyncCursor.__aexit__cs|dS)zI Close the current cursor and free associated resources. N)_closer,rrr r5<s zAsyncCursor.closecCs|jS)z9Writable attribute to control how result rows are formed.r(r,rrr r#BszAsyncCursor.row_factorycCs||_|jr |||_dSdSr)r(pgresult _make_row)rr#rrr r#Gs RowMaker[Row]cCs ||Srr8r,rrr _make_row_makerMs zAsyncCursor._make_row_makerpreparebinaryqueryr params Params | Noner> bool | Noner?c sz3|jj4IdH|j|j||||dIdHWdIdHW|S1IdHs-wYW|StjyF}z|dd}~ww)z= Execute a query or command to the database. Nr=)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)rr@rAr>r?exrrr executePs   zAsyncCursor.executeF) returning params_seqIterable[Params]rMboolc sztr|jj4IdH|jj}|r#|j||||IdHWdIdHn 1IdHs3wY|s|j4IdH7|jj4IdH|j||||IdHWdIdHn 1IdHskwYWdIdHn1IdHswYWdSWdSWdS|jj4IdH|j||||IdHWdIdHWdS1IdHswYWdSt j y}z| dd}~ww)zI Execute the same command with a sequence of input data. N) r is_supportedrDrE _pipelinerF_executemany_gen_pipelinepipeline_executemany_gen_no_pipelinerHrIrJ)rr@rNrMprKrrr executemanyds8   (&  X  2 zAsyncCursor.executemanyr r?sizerYintAsyncIterator[Row]c Cs||jjr td|jj4IdHzzF|j|j||||dIdHd}|j||IdHrZt |D]}|j ||j }|durHn|Vq8d}|j||IdHs4Wntj ym}z|dd}~wwW|jjtkr|jIdHz|j|jddIdHr |j|jddIdHsWn tyYnwz|j|jddIdHWnftyYniwnQ|jjtkr|jIdHz|j|jddIdHr |j|jddIdHsWn tyYnwz|j|jddIdHWwty YwwwWdIdHdSWdIdHdSWdIdHdS1IdHs7wYdS)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 modeNrXTF)first)_pgconnpipeline_statusrHProgrammingErrorrDrErF_stream_send_gen_stream_fetchone_genrange_txload_rowr:rIrJtransaction_statusACTIVE _try_cancel Exception) rr@rAr?rYr\posrecrKrrr streams                !#0zAsyncCursor.stream Row | NonecsD|IdH||j|j|j}|dur |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_fetchrcrd_posr:)rrecordrrr fetchoneszAsyncCursor.fetchoner list[Row]csf|IdH||jsJ|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) rmrnr9 arraysizerc load_rowsrominntuplesr:len)rrYrecordsrrr fetchmanys zAsyncCursor.fetchmanycsH|IdH||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)rmrnr9rcrtrorvr:)rrxrrr fetchalls  zAsyncCursor.fetchallcsTIdHd fdd } |j}|durdSjd7_|Vq) NrirZr+rlcsj|jSr)rcrdr:)rir,rr loadsz#AsyncCursor.__aiter__..loadTr )rirZr+rl)rmrnro)rr{rowrr,r __aiter__s zAsyncCursor.__aiter__relativevaluemodestrcs |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)rm_scroll)rrrrrr scrolls zAsyncCursor.scrollwriter statementrAsyncWriter | NoneAsyncIterator[AsyncCopy]c CszP|jj4IdH|j|||IdHWdIdHn 1IdHs(wYt||d4IdH}|VWdIdHn 1IdHsKwYWntjyc}z|dd}~ww|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) rDrErF_start_copy_genrrHrIrJ_select_current_result)rrrArcopyrKrrr r s (( zAsyncCursor.copyc s|jdur=|js?|jjrA|jj4IdH|j|jjjddIdHWdIdHdS1IdHs6wYdSdSdSdS)NFT)flush)_execmany_returningr9rDrRrErF _fetch_genr,rrr rm#s  . zAsyncCursor._fetch_pipeline)rr)rr"r#r$)rr"r#r&)r+r)r.r/r0r1r2r3r+r4)r+r4)r+r$)r#r$r+r4)r+r;r) r@r rArBr>rCr?rCr+r)r@r rNrOrMrPr+r4) r@r rArBr?rCrYrZr+r[)r+rl)r)rYrZr+rr)r+rr)r+r[)r~)rrZrrr+r4)rr rArBrrr+r)__name__ __module__ __qualname__ __slots__rr!r-r6r5propertyr#setterr<rLrWrkrqryrzr}rr rrm __classcell__rrr)r rsP       % 7   rr"N)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r rHabcr rrrrrowsrrr_compatrrRr _cursor_baserconnection_asyncrTransactionStatusrfrrrrr s"