o DDi,@sdZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZdd lmZmZddl m!Z!ddl"m#Z#erjddl$m%Z%ej&j'Z'Gddde!defZ(dS)z Psycopg AsyncCursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyoverload)asynccontextmanager) AsyncIteratorIterable)errors)pq)ParamsQueryQueryNoTemplate) AsyncCopy AsyncWriter)AsyncRowFactoryRowRowMaker)SelfTemplate) BaseCursor) AsyncPipeline)AsyncConnectioncspeZdZdZdZeddddZeded dZd d dffddZdgddZdhddZdiddZe djddZ e j dkd dZ dld"d#Z e dmd d d$dnd,d-Z ed d d$dod/d-Z dmd d d$dpd1d-Z d2d3dqd8d9Z dmd d:d;drd?d@ZdsdBdCZdtdEdFZdudHdIZdvdwdLdMZdxdNdOZdgdPdQZdydSdTZdzd{dYdZZe dmd d[d|d`daZdidbdcZZS)} AsyncCursorpsycopg connectionAsyncConnection[Row]cCdSNr)selfrrrM/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/cursor_async.py__init__!zAsyncCursor.__init__AsyncConnection[Any] row_factoryAsyncRowFactory[Row]cCrr rr!rr&rrr"r#$sN)r&AsyncRowFactory[Row] | Nonecst||p |j|_dSr )superr#r& _row_factoryr( __class__rr"r#)s returnrcs|Sr rr!rrr" __aenter__2szAsyncCursor.__aenter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonecs|IdHdSr )close)r!r1r3r5rrr" __aexit__5szAsyncCursor.__aexit__cs|dS)zI Close the current cursor and free associated resources. N)_closer/rrr"r8=s zAsyncCursor.closecCs|jS)z9Writable attribute to control how result rows are formed.r+r/rrr"r&CszAsyncCursor.row_factorycCs||_|jr |||_dSdSr )r+pgresult _make_row)r!r&rrr"r&Hs RowMaker[Row]cCs ||Sr r;r/rrr"_make_row_makerNs zAsyncCursor._make_row_makerpreparebinaryqueryrparams Params | NonerA bool | NonerBcdSr r)r!rCrDrArBrrr"executeQszAsyncCursor.executercrGr r)r!rCrArBrrr"rH[src 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)r!rCrDrArBexrrr"rHds   F) returning params_seqIterable[Params]rQboolc sDz|jj4IdHxtrP|jjr"|j||||IdHnT|j4IdH|j||||IdHWdIdHn 1IdHsJwYn|j||||IdHWdIdHWdSWdIdHWdSWdIdHWdS1IdHswYWdSt j y}z| dd}~ww)zI Execute the same command with a sequence of input data. N) rIrJr is_supported _pipelinerK_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerMrNrO)r!rCrRrQrPrrr" executemanyxs8   (  2 zAsyncCursor.executemanyr rBsizer\intAsyncIterator[Row]c Csv|jjr td|jj4IdHzzC|j|j||||dIdHd}|j||IdH}rWt |j D] }|j ||j Vq;d}|j||IdH}s6Wntjyj}z|dd}~wwW|jjtkr|jIdHz|j|jddIdHr |j|jddIdHsWn tyYnwz|j|jddIdHWnftyYniwnQ|jjtkr |jIdHz|j|jddIdHr |j|jddIdHsWn tyYnwz|j|jddIdHWwty YwwwWdIdHdSWdIdHdSWdIdHdS1IdHs4wYdS)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 modeNr[TF)first)_pgconnpipeline_statusrMProgrammingErrorrIrJrK_stream_send_gen_stream_fetchone_genrangentuples_txload_rowr=rNrOtransaction_statusACTIVE _try_cancel Exception) r!rCrDrBr\r_resposrPrrr"streams               0zAsyncCursor.streamAsyncIterator[Self]cCs"|jr |V|sdSqdS)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)r<nextsetr/rrr"resultsszAsyncCursor.resultsindexcsft|j |krt|jks!ntd|dt|jd|dkr,t|j|}|||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!rsrrr" set_results$ zAsyncCursor.set_result Row | NonecsL|IdH|}|j|jkr$|j|j|j}|jd7_|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_posrfrgrhr=r!rmrecordrrr"fetchones zAsyncCursor.fetchoner list[Row]csZ|IdH|}|s|j}|j|jt|j||j|j}|jt |7_|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) rzr{ arraysizerg load_rowsr|minrfr=rt)r!r\rmrecordsrrr" fetchmanyszAsyncCursor.fetchmanycs:|IdH|}|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)rzr{rgrr|rfr=)r!rmrrrr"fetchalls zAsyncCursor.fetchallcCs|Sr rr/rrr" __aiter__ r$zAsyncCursor.__aiter__rcsP|IdH|}|j|jkr$|j|j|j}|jd7_|Std)Nr zno more records to return)rzr{r|rfrgrhr=StopAsyncIterationr}rrr" __anext__#s zAsyncCursor.__anext__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)rz_scroll)r!rrrrr"scroll,s zAsyncCursor.scrollwriter statementrAsyncWriter | NoneAsyncIterator[AsyncCopy]c CszP|jj4IdH;|j|||IdHt||d4IdH}|VWdIdHn 1IdHs6wYWdIdHn 1IdHsKwYWntjyc}z|dd}~ww|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) rIrJrK_start_copy_genrrMrNrOrw)r!rrDrcopyrPrrr"r:s *( zAsyncCursor.copyc s|jdur=|js?|jjrA|jj4IdH|j|jjjddIdHWdIdHdS1IdHs6wYdSdSdSdS)NFT)flush)_execmany_returningr<rIrVrJrK _fetch_genr/rrr"rzRs  . zAsyncCursor._fetch_pipeline)rr)rr%r&r')rr%r&r))r.r)r1r2r3r4r5r6r.r7)r.r7)r.r')r&r'r.r7)r.r>r ) rCrrDrErArFrBrFr.r)rCrrArFrBrFr.r) rCrrDrErArFrBrFr.r)rCrrRrSrQrTr.r7) rCrrDrErBrFr\r]r.r^)r.rp)rsr]r.r)r.ry)r)r\r]r.r)r.r)r.r)r)rr]rrr.r7)rrrDrErrr.r)__name__ __module__ __qualname__ __slots__rr#r0r9r8propertyr&setterr?rHrZrorrrxrrrrrrrrrz __classcell__rrr,r"rsj         ! 2      rr%N))__doc__ __future__rtypesrtypingrrr contextlibrcollections.abcrr r rMr abcr rrrrrrowsrrr_compatrr _cursor_baser_pipeline_asyncrconnection_asyncrTransactionStatusrjrrrrr"s$