o Zad-x@sdZddlmZddlmZddlmZmZmZm Z m Z ddlm Z m Z m Z mZmZmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZdd lm Z m!Z"ddl#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3m4Z4m5Z5erddlm6Z6ddl7m8Z8m9Z9ddl:m;Z;ejZ>ej?j@Z@ej?jAZAej?jBZBej?jCZCej?jDZDej?jEZEej?jFZFej?jGZGej?jHZHejIjJZJGdddeee$fZKGdddeKde$fZLdS)z psycopg cursor objects )partial) TracebackType)AnyGenericIterableIteratorList)OptionalNoReturnSequenceTupleTypeTypeVar)overload TYPE_CHECKING)contextmanager)pq)adapt)errors)ConnectionTypeQueryParamsPQGen)CopyWriter)RowRowMaker RowFactoryColumn) 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) _connTEXTformatpgconnr0r AdaptersMapadapters _adapters arraysize_closed _last_query_resetselfr1r?M/var/www/mitschrieb_seite/venv/lib/python3.10/site-packages/psycopg/cursor.py__init__>s zBaseCursor.__init__T reset_queryreturnNcCs:g|_d|_d|_d|_d|_|d|_|rd|_dSdS)Nr)_resultspgresult_pos_iresult _rowcount_execmany_returning_query)r>rBr?r?r@r<Hs zBaseCursor._resetc Csp|jjd|jj}tj|j}|jrd}n|jr$t |jj j }nd}d|d|d|dt |dd S) N.closedz no result) __class__ __module__ __qualname__rmiscconnection_summaryr0r:rF ExecStatusstatusnameid)r>clsinforWr?r?r@__repr__Ts$zBaseCursor.__repr__cC|jS)z$The connection this cursor is using.)r2r>r?r?r@r1_zBaseCursor.connectioncCr]N)r8r^r?r?r@r7dszBaseCursor.adapterscCr])z`True` if the cursor is closed.)r:r^r?r?r@rMhr_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|qSr?r).0ir^r?r@ |sz*BaseCursor.description..N)rFnfieldsrW TUPLES_OK SINGLE_TUPLErange)r>resr?r^r@ descriptionmszBaseCursor.descriptioncCr])z6Number of records affected by the precedent operation.)rIr^r?r?r@rowcountr_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)rFrWrerG)r>tuplesr?r?r@ rownumberszBaseCursor.rownumbersizescCdSr`r?)r>rmr?r?r@ setinputsizeszBaseCursor.setinputsizessizecolumncCrnr`r?)r>rqrrr?r?r@ setoutputsizerpzBaseCursor.setoutputsizecCs,|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. rTN)rHlenrE_select_current_resultr^r?r?r@nextsetszBaseCursor.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)rFcommand_statusdecode)r>msgr?r?r@ statusmessageszBaseCursor.statusmessagecCstr`)NotImplementedErrorr^r?r?r@_make_row_makerzBaseCursor._make_row_makerpreparebinaryqueryparamsrrccs||EdH|||}|j|||dEdH}|jjr(|jjEdHn|dus.J||||_|d||_ |jj D] }|j |EdHqDdS)z*Generator implementing `Cursor.execute()`.Nr~r) _start_query_convert_query_maybe_prepare_genr2 _pipeline_communicate_gen_check_resultsrErur; _preparedget_maintenance_commands _exec_command)r>rrrrpgqresultscmdr?r?r@ _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)r2rrrIrJrrKdumprrr; _fetch_genrrr) r>rrrpipelinefirstrrrr?r?r@_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)rrIrrKrrrrEextendcommand_tuplesrur;r2rrr) r>rrrrrrrrhrr?r?r@_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_preparer2rr&r0rW FATAL_ERROReerror_from_result _encoding_send_query_preparedrmaybe_add_to_cache result_queueappendvalidate) r>rrrpreprXresultkeyqueuedrr?r?r@rs,     zBaseCursor._maybe_prepare_gencCs|jj||Sr`)r2rget)r>rrr?r?r@rHszBaseCursor._get_preparedrccsR||EdH|||}|j||dd|j||_t|jEdHdS)z2Generator to send the query for `Cursor.stream()`.NT)rforce_extended)rrrr0set_single_row_moder;r()r>rrrrr?r?r@_stream_send_genMs  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'r0rWrfrFr. set_pgresultr|r/re COMMAND_OKrProgrammingError_raise_for_result)r>rrhrWr?r?r@_stream_fetchone_gen\s*  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) rMrInterfaceErrorr<r;rr)r.r2r)r>rr?r?r@rws  zBaseCursor._start_query statementccs|jjr td|EdH|r!t|j}||||j}| |}|j |ddt |j EdH}t |dkr@td||d||_|ddS)zrrrrrr?r?r@_start_copy_gens       zBaseCursor._start_copy_genF)rrrc Cs|dur|j}n|r tnt}||_|jjr-|jjjt|j j |j |j |j |j|ddS|s6|j s6|tkrG|j j |j |j |j |j|ddS|j |j dS)z Implement part of execute() before waiting common to sync and async. This is not a generator, but a normal non-blocking function. N) param_formats param_types result_format)r4BINARYr3rKr2r command_queuerrr0send_query_paramsrrformatstypes send_query)r>rrrfmtr?r?r@rs0      zBaseCursor._execute_sendcCst|j}||||Sr`)r!r.r)r>rrrr?r?r@rs  zBaseCursor._convert_queryrcCsD|std|D]}|j}|tkr|tkr|tkr||q dS)z Verify that the results of a query are valid. Verify that the query returned at least one result and that they all represent a valid result from the database. zgot no result from the queryN)r InternalErrorrWrer EMPTY_QUERYr)r>rrhrWr?r?r@rs  zBaseCursor._check_resultsrcCsj|j}|tkrtj||jd|tkrtd|tks$|tks$|t kr)t dt dt |j)zV Raise an appropriate error message for an unexpected database result rpipeline abortedz8COPY cannot be used with this method; use copy() insteadz%unexpected result status from query: )rWrrrrPIPELINE_ABORTEDPipelineAbortedCOPY_INCOPY_OUT COPY_BOTHrrrrVrXr>rrWr?r?r@rs zBaseCursor._raise_for_resultrbr4cCst||_|j|}|_|jj||dd|_|jtkr!|jj|_ n|jt kr3|jj }|dur0|nd|_ | |_ dS)zL Select one of the results in the cursor as the active one. )r4rNrD)rHrErFr.rrGrWrentuplesrIrrr|r/)r>rbr4rhnrowsr?r?r@rus   z!BaseCursor._select_current_resultcCs|||j }|jdur|j||r|ddSdS|jr3|j||r1|ddSdS|D] }|j|jp>d7_q5dS)Nr)rrErJrrurIr)r>r first_batchrhr?r?r@_set_results_from_pipelines    z%BaseCursor._set_results_from_pipelinerXcCsZ|jjr |jjjt|jj||j|jd|jjj ddS|jj||j|jddS)N)r) r2rrrrr0 send_preparerrr)r>rXrr?r?r@r*s zBaseCursor._send_preparec Csj|dur|j}n|r tnt}|jjr'|jjjt|jj ||j |j |ddS|jj ||j |j |ddS)N)rr) r4rr3r2rrrrr0send_query_preparedrr)r>rXrrrr?r?r@r8s     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) rMrrrFrrWrerrrrr)r>rhrWr?r?r@_check_result_for_fetchOs    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 ) rWrrrrrrrrrVrXrr?r?r@r`s 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)rrFrG ValueErrorr IndexError)r>rrnewposr?r?r@_scrollos   zBaseCursor._scrollcCs|jddd|_dS)z3Non-blocking part of closing. Common to sync/async.F)rBTN)r<r:r^r?r?r@_close|s  zBaseCursor._closecCs t|jSr`)r$r0r^r?r?r@rs zBaseCursor._encoding)Tr`rCN)rr+rCN)D__name__rRrSsplit __slots__rrV__annotations__rrrrAboolr<strr\propertyr1rr6r7rMr rr riintrjrlr rrorsrvrzr|rrrrrrrr!rr r%bytesrrrrrrrrr rFormatrurrrrrrrrr?r?r?r@r-0s,        $ ) ,    ! .     r-cs0eZdZdZdZedddZedGd d Zedddd d ee fdd Zdddd d e ee ffdd ZdedefddZ de e e de e de eddfddZdHddZedee fddZejd ee ddfddZdee fdd Z dIddd!ded"ed#e ed$e ed%e edef d&d'Zd(d)d"ed*eed+eddfd,d-Z dIdd.d"ed#e ed%e edee fd/d0Zde e fd1d2ZdJd4ede e fd5d6Z!de e fd7d8Z"dee fd9d:Z#dKdd?Z%e& dIdd@dAed#e edBe e'dee(fdCdDZ)dHdEdFZ*Z+S)LCursorpsycopgr?_Selfz Cursor[Any])boundr> Cursor[Row]r1Connection[Row]cCrnr`r?r=r?r?r@rArpzCursor.__init__Connection[Any] row_factorycCrnr`r?r>r1rr?r?r@rAsN)rcst||p |j|_dSr`)superrAr _row_factoryrrQr?r@rAs rCcCs|Sr`r?r^r?r?r@ __enter__r}zCursor.__enter__exc_typeexc_valexc_tbcCs |dSr`)close)r>rrrr?r?r@__exit__s zCursor.__exit__cCs |dS)zI Close the current cursor and free associated resources. N)rr^r?r?r@rs z Cursor.closecCr])z9Writable attribute to control how result rows are formed.rr^r?r?r@rr_zCursor.row_factorycCs||_|jr |||_dSdSr`)rrFr/)r>rr?r?r@rscCs ||Sr`rr^r?r?r@r|s zCursor._make_row_makerr~rrrrc Csrz&|jj|j|j||||dWdW|S1swYW|Stjy8}z|dd}~ww)z= Execute a query or command to the database. r~N)r2lockwaitrr _NO_TRACEBACKwith_traceback)r>rrrrexr?r?r@r&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_supportedr2rrrrrrrrr)r>rrrprr?r?r@ 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)r0pipeline_statusrrr2rrrrr.load_rowr/rrtransaction_statusACTIVEcancel Exception)r>rrrrrecrr?r?r@streamsl          "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_pipelinerr.r rGr/)r>recordr?r?r@fetchone$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` ) rrrFr9r. load_rowsrGminrr/rt)r>rqrecordsr?r?r@ fetchmany3s 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` )rrrFr.rrGrr/)r>rr?r?r@fetchallIs   zCursor.fetchallc#sZdtdttffdd } |j}|dur"dSjd7_|Vq)NposrCcsj|jSr`)r.r r/)rr^r?r@loadZszCursor.__iter__..loadTr)rrrr rrG)r>rrowr?r^r@__iter__Vs 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>rrr?r?r@scrollds 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 Nr!r) r2rrrrrrrru)r>rrr"copyrr?r?r@r#rs  z Cursor.copycCsj|jdur/|js1|jjr3|jj|j|jjjddWddS1s(wYdSdSdSdS)NFTr)rJrFr2rrrrr^r?r?r@rs " zCursor._fetch_pipeline)r>rr1rrr`)r)r),rrRrSrrrrrArrr rr BaseExceptionrrrrrsetterrr|rrrr&rr rrrrrrrrrr r CopyWriterrr#r __classcell__r?r?rr@rs          % ,  rrN)M__doc__ functoolsrrrtypingrrrrrr r r r r rrr contextlibrrrrrabcrrrrr#rrr&rowsrrr_columnr _queriesr!r"rr# _encodingsr$ _preparingr% generatorsr&r'r(r)pq.abcr*r+r1r,rr3rrVrrrerrrrrfrTransactionStatusrr-rr?r?r?r@sR             \