o DDil+@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 Cursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyoverload)contextmanager)IterableIterator)errors)pq)ParamsQueryQueryNoTemplate)CopyWriter)Row RowFactoryRowMaker)SelfTemplate)Pipeline) BaseCursor) ConnectioncspeZdZdZdZeddddZeded 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)}Cursorpsycopg connectionConnection[Row]cCdSNr)selfrrrG/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/cursor.py__init__$zCursor.__init__Connection[Any] row_factoryRowFactory[Row]cCrr rr!rr&rrr"r#'sN)r&RowFactory[Row] | Nonecst||p |j|_dSr )superr#r& _row_factoryr( __class__rr"r#,s returnrcC|Sr rr!rrr" __enter__2r$zCursor.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonecCs |dSr )close)r!r2r4r6rrr"__exit__5s zCursor.__exit__cCs |dS)zI Close the current cursor and free associated resources. N)_closer0rrr"r9=s z Cursor.closecCs|jS)z9Writable attribute to control how result rows are formed.r+r0rrr"r&CszCursor.row_factorycCs||_|jr |||_dSdSr )r+pgresult _make_row)r!r&rrr"r&Hs RowMaker[Row]cCs ||Sr r<r0rrr"_make_row_makerNs zCursor._make_row_makerpreparebinaryqueryrparams Params | NonerB bool | NonerCcCrr r)r!rDrErBrCrrr"executeQszCursor.executercCrr r)r!rDrBrCrrr"rH[src Csrz&|jj|j|j||||dWdW|S1swYW|Stjy8}z|dd}~ww)z= Execute a query or command to the database. rAN)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)r!rDrErBrCexrrr"rHds   F) returning params_seqIterable[Params]rQboolc Cszk|jj\tr;|jjr|j||||n>|j|j||||Wdn1s5wYn|j||||WdWdSWdWdSWdWdS1sdwYWdSt 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!rDrRrQrPrrr" executemanyxs6    & zCursor.executemanyr rCsizer\int Iterator[Row]c cs |jjr td|jjzz:|j|j||||dd}|j||}rIt |j D] }|j ||j Vq0d}|j||}s+Wntjy\}z|dd}~wwW|jjtkr|jz|j|jddr |j|jddstWn tyYnwz |j|jddWnUtyYnUwnC|jjtkr|jz|j|jddr |j|jddsWn tyYnwz |j|jddWwtyYwwwWddSWddSWddS1swYdS)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 moder[TFN)first)_pgconnpipeline_statusrMProgrammingErrorrIrJrK_stream_send_gen_stream_fetchone_genrangentuples_txload_rowr>rNrOtransaction_statusACTIVE _try_cancel Exception) r!rDrErCr\r_resposrPrrr"streamsp          "z Cursor.streamIterator[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=nextsetr0rrr"resultsszCursor.resultsindexcCsdt|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$ zCursor.set_result Row | NonecCsD||}|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` r N)_fetch_pipeline_check_result_for_fetch_posrfrgrhr>r!rmrecordrrr"fetchones zCursor.fetchoner list[Row]cCsR||}|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` ) rzr{ arraysizerg load_rowsr|minrfr>rt)r!r\rmrecordsrrr" fetchmanyszCursor.fetchmanycCs2||}|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` )rzr{rgrr|rfr>)r!rmrrrr"fetchalls zCursor.fetchallcCr/r rr0rrr"__iter__r$zCursor.__iter__rcCsH||}|j|jkr |j|j|j}|jd7_|Std)Nr zno more records to return)rzr{r|rfrgrhr> StopIterationr}rrr"__next__!s zCursor.__next__relativevaluemodestrcCs||||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 z Cursor.scrollwriter statementr Writer | NoneIterator[Copy]c csz9|jj+|j|||t||d }|VWdn1s%wYWdn1s4wYWntjyL}z|dd}~ww|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. rNr) rIrJrK_start_copy_genrrMrNrOrw)r!rrErcopyrPrrr"r8s  z Cursor.copycCsj|jdur/|js1|jjr3|jj|j|jjjddWddS1s(wYdSdSdSdS)NFT)flush)_execmany_returningr=rIrVrJrK _fetch_genr0rrr"rzPs  " zCursor._fetch_pipeline)rr)rr%r&r')rr%r&r))r.r)r2r3r4r5r6r7r.r8)r.r8)r.r')r&r'r.r8)r.r?r ) rDrrErFrBrGrCrGr.r)rDrrBrGrCrGr.r) rDrrErFrBrGrCrGr.r)rDrrRrSrQrTr.r8) rDrrErFrCrGr\r]r.r^)r.rp)rsr]r.r)r.ry)r)r\r]r.r)r.r)r.r)r)rr]rrr.r8)rrrErFrrr.r)__name__ __module__ __qualname__ __slots__rr#r1r:r9propertyr&setterr@rHrZrorrrxrrrrrrrrrz __classcell__rrr,r"r sj          ! 0      rr%N))__doc__ __future__rtypesrtypingrrr contextlibrcollections.abcrr r rMr abcr rrrrrrowsrrr_compatrrrVr _cursor_baserrrTransactionStatusrjrrrrr"s$