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 Cursor object. ) annotations) TracebackType)AnyIteratorIterable TYPE_CHECKINGoverload)contextmanager)pq)errors)QueryParams)CopyWriter)RowRowMaker RowFactory)Self)Pipeline) BaseCursor) Connectioncs"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)lCursorpsycopg connectionConnection[Row]cCdSNr)selfrrrW/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/cursor.py__init__#zCursor.__init__Connection[Any] row_factoryRowFactory[Row]cCrrrrrr$rrr r!&sN)r$RowFactory[Row] | Nonecst||p |j|_dSr)superr!r$ _row_factoryr& __class__rr r!+s returnrcCs|Srrrrrr __enter__1r"zCursor.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonecCs |dSr)close)rr/r1r3rrr __exit__4s zCursor.__exit__cCs |dS)zI Close the current cursor and free associated resources. N)_closer-rrr r6<s z Cursor.closecCs|jS)z9Writable attribute to control how result rows are formed.r)r-rrr r$BszCursor.row_factorycCs||_|jr |||_dSdSr)r)pgresult _make_row)rr$rrr r$Gs RowMaker[Row]cCs ||Srr9r-rrr _make_row_makerMs zCursor._make_row_makerpreparebinaryqueryr params Params | Noner? bool | Noner@c Csrz&|jj|j|j||||dWdW|S1swYW|Stjy8}z|dd}~ww)z= Execute a query or command to the database. r>N)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)rrArBr?r@exrrr executePs   zCursor.executeF) returning params_seqIterable[Params]rNboolc Cs>ztri|jj|jj}|r|j||||Wdn1s%wY|sc|j'|jj|j||||Wdn1sLwYWdn1s[wYWdSWdSWdS|jj|j||||WdWdS1swYWdSt j y}z| dd}~ww)zI Execute the same command with a sequence of input data. N) r is_supportedrErF _pipelinerG_executemany_gen_pipelinepipeline_executemany_gen_no_pipelinerIrJrK)rrArOrNprLrrr executemanyds:   "  & zCursor.executemanyr r@sizerZint Iterator[Row]c cs|jjr td|jjzz=|j|j||||dd}|j||rLt |D]}|j ||j }|dur=n|Vq-d}|j||s)Wntj y_}z|dd}~wwW|jjtkr|jz|j|jddr |j|jddswWn 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 moderYTNF)first)_pgconnpipeline_statusrIProgrammingErrorrErFrG_stream_send_gen_stream_fetchone_genrange_txload_rowr;rJrKtransaction_statusACTIVE _try_cancel Exception) rrArBr@rZr]posrecrLrrr streamsv           $z Cursor.stream Row | NonecCs<|||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_fetchrdre_posr;)rrecordrrr fetchones zCursor.fetchoner list[Row]cCs^|||js J|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` ) rnror: arraysizerd load_rowsrpminntuplesr;len)rrZrecordsrrr fetchmanys zCursor.fetchmanycCs@|||js J|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` )rnror:rdrurprwr;)rryrrr fetchalls   zCursor.fetchallc#sNd fdd } |j}|durdSjd7_|Vq) Nrjr[r,rmcsj|jSr)rdrer;)rjr-rr loadszCursor.__iter__..loadTr )rjr[r,rm)rnrorp)rr|rowrr-r __iter__s zCursor.__iter__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)rn_scroll)rrrrrr scrolls z Cursor.scrollwriter statementr Writer | NoneIterator[Copy]c csz9|jj|j|||Wdn1swYt||d }|VWdn1s4wYWntjyL}z|dd}~ww|ddS)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) rErFrG_start_copy_genrrIrJrK_select_current_result)rrrBrcopyrLrrr rs  z Cursor.copycCsj|jdur/|js1|jjr3|jj|j|jjjddWddS1s(wYdSdSdSdS)NFT)flush)_execmany_returningr:rErSrFrG _fetch_genr-rrr rns  " zCursor._fetch_pipeline)rr)rr#r$r%)rr#r$r')r,r)r/r0r1r2r3r4r,r5)r,r5)r,r%)r$r%r,r5)r,r<r) rAr rBrCr?rDr@rDr,r)rAr rOrPrNrQr,r5) rAr rBrCr@rDrZr[r,r\)r,rm)r)rZr[r,rs)r,rs)r,r\)r)rr[rrr,r5)rr rBrCrrr,r)__name__ __module__ __qualname__ __slots__rr!r.r7r6propertyr$setterr=rMrXrlrrrzr{r~rr rrn __classcell__rrr*r rsP        ! 3   rr#N)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r rIabcr rrrrrowsrrr_compatrrSr _cursor_baserrrTransactionStatusrgrrrrr s"