o 6 fUz@sdZddlmZddlmZddlmZmZmZm Z m Z ddlm Z m Z m Z mZmZddlmZmZddlmZddlmZd d lmZd d lmZd d lmZd d lmZmZmZmZd dl m!Z!m"Z#d dl$m%Z%m&Z&m'Z'd dl(m)Z)d dl*m+Z+d dl,m-Z-d dl.m/Z/m0Z0d dl1m2Z2d dl3m4Z4d dl5m6Z6d dl7m8Z8m9Z9m:Z:erd dlm;Z;d dlZ>d dl?m@Z@ejAjBZBejAjCZCejDjEZEejDjFZFejDjGZGejDjHZHejDjIZIejDjJZJejDjKZKejDjLZLejDjMZMejNjOZOGdddeee%fZPGdddePde%fZQd S)!z psycopg cursor objects )partial) TracebackType)AnyGenericIterableIteratorList)OptionalNoReturnSequenceTupleType)overload TYPE_CHECKING)warn)contextmanager)pq)adapt)errors)ConnectionTypeQueryParamsPQGen)CopyWriter)RowRowMaker RowFactoryColumn)Self)connection_summary) PostgresQueryPostgresClientQuery)Pipeline)pgconn_encoding)Prepare)executefetchsend) Transformer)PGconnPGresult) Connectionc @seZdZUdZejZded<ee ed<ded<de fdd Z dmd e d d fddZ d efddZed e fddZed ejfddZed e fddZed eeefddZed efddZed eefddZdeed d fdd Zdnd!ed"eed d fd#d$Z d ee fd%d&Z!ed eefd'd(Z"d ee fd)d*Z# dnd d d+d,e$d-ee%d.ee d/ee d e&d f d0d1Z'd,e$d2e(e%d3e d e&d fd4d5Z)d,e$d2e(e%d3e d e&d fd6d7Z*d d d+d8e+d.ee d/ee d e&eed9fd:d;Z, dnd8e+d.ee d e-e.e/ffdd,e$d-ee%d/ee d e&d fd?d@Z1dAe d e&ed9fdBdCZ2dnd,ee$d e&d fdDdEZ3 dndFe$d-ee%d e&d fdGdHZ4dId dJd,e+dKe d/ee d d fdLdMZ5 dnd,e$d-ee%d e+fdNdOZ6dPed9d d fdQdRZ7dSd9d e8fdTdUZ9 dndVedWeej:d d fdXdYZ;dPed9d d fdZd[Zd\e/d8e+d/ee d d fd_d`Z>dodadbZ?dpdcddZ@deedfed d fdgdhZAdodidjZBed efdkdlZCd S)q BaseCursorz _conn format _adapters arraysize _closed _results pgresult _pos _iresult _rowcount _query _tx _last_query _row_factory _make_row _pgconn _execmany_returning __weakref__ r+_tx _make_rowr,_pgconn connectioncCs@||_t|_|j|_t|j|_d|_ d|_ d|_ | dS)NrF) _connTEXTformatpgconnr2r AdaptersMapadapters _adapters arraysize_closed _last_query_resetselfr3rAG/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/cursor.py__init__As zBaseCursor.__init__T reset_queryreturnNcCs:g|_d|_d|_d|_d|_|d|_|rd|_dSdS)Nr)_resultspgresult_pos_iresult _rowcount_execmany_returning_query)r@rDrArArBr>Ks zBaseCursor._resetc Csl|jjd|jj}t|j}|jrd}n|jr"t|jj j }nd}d|d|d|dt |dd S) N.closedz no result) __class__ __module__ __qualname__r"r2r<rHr ExecStatusstatusnameid)r@clsinforWrArArB__repr__Ws $zBaseCursor.__repr__cC|jS)z$The connection this cursor is using.)r4r@rArArBr3bzBaseCursor.connectioncCr]N)r:r^rArArBr9gszBaseCursor.adapterscCr])z`True` if the cursor is closed.)r<r^rArArBrOkr_zBaseCursor.closedcs@j}|r|js|jtks|jtkrfddt|jDSdS)z A list of `Column` objects describing the current resultset. `!None` if the current resultset didn't return tuples. csg|]}t|qSrAr).0ir^rArB sz*BaseCursor.description..N)rHnfieldsrW TUPLES_OK SINGLE_TUPLErange)r@resrAr^rB descriptionpszBaseCursor.descriptioncCr])z6Number of records affected by the precedent operation.)rKr^rArArBrowcountr_zBaseCursor.rowcountcCs |jo|jjtk}|r|jSdS)zoIndex of the next row to fetch in the current result. `!None` if there is no result to fetch. N)rHrWrerI)r@tuplesrArArB rownumberszBaseCursor.rownumbersizescCdSr`rA)r@rmrArArB setinputsizeszBaseCursor.setinputsizessizecolumncCrnr`rA)r@rqrrrArArB setoutputsizerpzBaseCursor.setoutputsizecCsH|jdur|jjrtdt|jt|jdkr"||jddSdS)a Move to the result set of the next query executed through `executemany()` or to the next result set if `execute()` returned more than one. Return `!True` if a new result is available, which will be the one methods `!fetch*()` will operate on. Nzusing nextset() in pipeline mode for several execute() is deprecated and will be dropped in 3.2; please use different cursors to receive more than one resultrT) rLr4 _pipelinerDeprecationWarningrJlenrG_select_current_resultr^rArArBnextsets zBaseCursor.nextsetcCs"|jr|jjnd}|r|SdS)z The command status tag from the last SQL command executed. `!None` if the cursor doesn't have a result available. N)rHcommand_statusdecode)r@msgrArArB statusmessageszBaseCursor.statusmessagecCstr`)NotImplementedErrorr^rArArB_make_row_makerzBaseCursor._make_row_makerpreparebinaryqueryparamsrrccs||EdH|||}|j|||dEdH}|jjr(|jjEdHn|dus.J||||_|d||_ |jj D] }|j |EdHqDdS)z*Generator implementing `Cursor.execute()`.Nrr) _start_query_convert_query_maybe_prepare_genr4rt_communicate_gen_check_resultsrGrwr= _preparedget_maintenance_commands _exec_command)r@rrrrpgqresultscmdrArArB _execute_gens      zBaseCursor._execute_gen params_seq returningc cs|jj}|s J||EdH|sd|_|jdusJ||_d}|D]&}|r4|||}||_d}n|||j|ddEdH| EdHq$||_ |rY|j ddEdH|jj D] }|j|EdHq_dS)zY Generator implementing `Cursor.executemany()` with pipelines available. NrTFrflush)r4rtrrKrLrrMdumprrr= _fetch_genrrr) r@rrrpipelinefirstrrrrArArB_executemany_gen_pipelines.  z$BaseCursor._executemany_gen_pipelinec cs||EdH|sd|_d}|D]B}|r"|||}||_d}n|||j|ddEdH}|dus7J|||rE|j|q|D] }|j|j pPd7_qGq|jr]| d||_ |j j D] } |j | EdHqfdS)z] Generator implementing `Cursor.executemany()` with pipelines not available. NrTFr)rrKrrMrrrrGextendcommand_tuplesrwr=r4rrr) r@rrrrrrrrhrrArArB_executemany_gen_no_pipelines2     z'BaseCursor._executemany_gen_no_pipelinerr-c cs|||\}}|tjur|j||dn-|tjur;||||jjs;t|j EdH\}|j t kr;t j ||jd|j|||d|jj|||}|jjrgd}|dur[|||f}|jjj||fdSt|j EdH} |dur}|jj|||| | S)Nrencoding) _get_preparedr'NO _execute_sendSHOULD _send_preparer4rtr(r2rW FATAL_ERROReerror_from_result _encoding_send_query_preparedrmaybe_add_to_cache result_queueappendvalidate) r@rrrpreprXresultkeyqueuedrrArArBr,s,     zBaseCursor._maybe_prepare_gencCs|jj||Sr`)r4rget)r@rrrArArBrWszBaseCursor._get_preparedrccsR||EdH|||}|j||dd|j||_t|jEdHdS)z2Generator to send the query for `Cursor.stream()`.NT)rforce_extended)rrrr2set_single_row_moder=r*)r@rrrrrArArB_stream_send_gen\s  zBaseCursor._stream_send_genrccst|jEdH}|durdS|j}|tkr*||_|jj||d|r(||_|S|t ks2|t krI|r>t|jEdH}|s4|t krGt ddS| |S)N) set_loadersz1the operation in stream() didn't produce a result)r)r2rWrfrHr0 set_pgresultr~r1re COMMAND_OKrProgrammingError_raise_for_result)r@rrhrWrArArB_stream_fetchone_genks*  zBaseCursor._stream_fetchone_genccsP|jr td||jr|j|urd|_t||_|j EdHdS)zGenerator to start the processing of a query. It is implemented as generator because it may send additional queries, such as `begin`. the cursor is closedN) rOrInterfaceErrorr>r=rr+r0r4r)r@rrArArBrs  zBaseCursor._start_query statementccs|jjr td|EdH|r!t|j}||||j}| |}|j |ddt |j EdH}t |dkr@td||d||_|ddS)zd7_q5dS)Nr)rrGrLrrwrKr)r@r first_batchrhrArArB_set_results_from_pipeline#s    z%BaseCursor._set_results_from_pipelinerXcCsZ|jjr |jjjt|jj||j|jd|jjj ddS|jj||j|jddS)N)r) r4rtrrrr2 send_preparerrr)r@rXrrArArBr9s zBaseCursor._send_preparec Csj|dur|j}n|r tnt}|jjr'|jjjt|jj ||j |j |ddS|jj ||j |j |ddS)N)rr) r6rr5r4rtrrrr2send_query_preparedrr)r@rXrrrrArArBrGs     zBaseCursor._send_query_preparedcCsj|jrtd|j}|std|j}|tkrdS|tkr'tj||j d|t kr0t dtd)Nrzno result availablerrz*the last operation didn't produce a result) rOrrrHrrWrerrrrr)r@rhrWrArArB_check_result_for_fetch^s    z"BaseCursor._check_result_for_fetchcCsJ|j}|tks |tkr dS|tkrtj||jdtdt |j )zV Check that the value returned in a copy() operation is a legit COPY. NrzZcopy() should be used only with COPY ... TO STDOUT or COPY ... FROM STDIN statements, got ) rWrrrrrrrrrVrXrrArArBros zBaseCursor._check_copy_resultvaluemodecCsv||js J|dkr|j|}n|dkr|}ntd|dd|kr1|jjks6tdtd||_dS)Nrelativeabsolutez bad mode: z'. It should be 'relative' or 'absolute'rzposition out of bound)rrHrI ValueErrorr IndexError)r@rrnewposrArArB_scroll~s   zBaseCursor._scrollcCs|jddd|_dS)z3Non-blocking part of closing. Common to sync/async.F)rDTN)r>r<r^rArArB_closes  zBaseCursor._closecCs t|jSr`)r&r2r^rArArBrs zBaseCursor._encoding)Tr`rEN)rr-rEN)D__name__rTrUsplit __slots__rrV__annotations__rrrrCboolr>strr\propertyr3rr8r9rOr rr riintrjrlr rrorsrxr|r~rrrrrrrr#rr r'bytesrrrrrrrrr rFormatrwrrrrrrrrrArArArBr/3s,        $ ) ,    ! .     r/c seZdZdZdZedBddZedddeefd dZd d dddeeeffd dZd e fddZ dee e dee dee d d fddZdCddZed eefddZejdeed d fddZd eefddZ dDd d ddedeedeed eed e f d!d"Zd#d$ded%eed&ed d fd'd(Z dDd d)dedeed eed eefd*d+Zd eefd,d-ZdEd/ed eefd0d1Z d eefd2d3Z!d eefd4d5Z"dFd7ed8e#d d fd9d:Z$e% dDd d;dd?Z(dCd@dAZ)Z*S)GCursorpsycopgrAr3Connection[Row]cCrnr`rAr?rArArBrCrzCursor.__init__Connection[Any] row_factorycCrnr`rAr@r3rrArArBrCsN)rcst||p |j|_dSr`)superrCr _row_factoryrrSrArBrCs rEcCs|Sr`rAr^rArArB __enter__rzCursor.__enter__exc_typeexc_valexc_tbcCs |dSr`)close)r@rrrrArArB__exit__s zCursor.__exit__cCs |dS)zI Close the current cursor and free associated resources. N)rr^rArArBrs z Cursor.closecCr])z9Writable attribute to control how result rows are formed.rr^rArArBrr_zCursor.row_factorycCs||_|jr |||_dSdSr`)rrHr1)r@rrArArBrscCs ||Sr`rr^rArArBr~s zCursor._make_row_makerrrrrrc Csrz&|jj|j|j||||dWdW|S1swYW|Stjy8}z|dd}~ww)z= Execute a query or command to the database. rN)r4lockwaitrr _NO_TRACEBACKwith_traceback)r@rrrrexrArArBr(s   zCursor.executeF)rrrc 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_supportedr4rrtrrrrrrr)r@rrrprrArArB executemanys:    "  & zCursor.executemanyrccs|jjr td|jjzz/|j|j|||dd}|j||r>|j d|j }|Vd}|j||s(Wntj yQ}z| dd}~wwW|jjtkr|jz|j|jddrt |j|jddsiWn ty~Ynwz |j|jddWnUtyYnUwnC|jjtkr|jz|j|jddr |j|jddsWn tyYnwz |j|jddWwtyYwwwWddSWddSWddS1swYdS)zC Iterate row-by-row on a result from the database. z(stream() cannot be used in pipeline moderTrFN)r)r2pipeline_statusrrr4rrrrr0load_rowr1rrtransaction_statusACTIVEcancel Exception)r@rrrrrecrrArArBstreamsl          "z Cursor.streamcCs<|||j|j|j}|dur|jd7_|S)z Return the next record from the current recordset. Return `!None` the recordset is finished. :rtype: Optional[Row], with Row defined by `row_factory` Nr)_fetch_pipelinerr0r rIr1)r@recordrArArBfetchone-s zCursor.fetchonerrqcCs^|||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` ) rrrHr;r0 load_rowsrIminrr1rv)r@rqrecordsrArArB fetchmany<s 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` )rrrHr0rrIrr1)r@rrArArBfetchallRs   zCursor.fetchallc#sZdtdttffdd } |j}|dur"dSjd7_|Vq)NposrEcsj|jSr`)r0r r1)rr^rArBloadcszCursor.__iter__..loadTr)rrrr rrI)r@rrowrAr^rB__iter___s zCursor.__iter__rrrcCs||||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)rr)r@rrrArArBscrollms z Cursor.scrollwriterrr c csz9|jj|j|||Wdn1swYt||d }|VWdn1s4wYWntjyL}z|dd}~ww|ddS)zk Initiate a :sql:`COPY` operation and return an object to manage it. :rtype: Copy Nrr) r4rrrrrrrrw)r@rrr copyrrArArBr!{s  z Cursor.copycCsj|jdur/|js1|jjr3|jj|j|jjjddWddS1s(wYdSdSdSdS)NFTr)rLrHr4rtrrrr^rArArBrs  " zCursor._fetch_pipeline)r3rrr`)r)r)+rrTrUrrrCrrr r!rr BaseExceptionrrrrrsetterrr~rrrr(rr rrrrrrrrrrr CopyWriterrr!r __classcell__rArArrBrs        % ,  rrN)R__doc__ functoolsrrrtypingrrrrrr r r r r rrwarningsr contextlibrrrrrabcrrrrr!rrr$rowsrrr_columnr _compatr!pq.miscr"_queriesr#r$rtr% _encodingsr& _preparingr' generatorsr(r)r*r+pq.abcr,r-r3r.rr5rrVrrrerrrrrfrTransactionStatusr r/rrArArArBsX               h