2iXdZddlmZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZmZmZmZmZdd lmZmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,ddl-m.Z.erddlm/Z/ddl0m1Z1m2Z2ejfjhZ4ejfjjZ5ejljnZ7ejljpZ8ejljrZ9ejljtZ:ejljvZ;ejljxZejlj~Z?ejljZ@ejjZBGddeeefZCy)z Psycopg BaseCursor object ) annotations) TYPE_CHECKINGAnyGenericNoReturn) ReferenceTyperef)partial)IterableSequence)adapt)errors)pq)ConnectionTypeLoaderParamsPQGenQuery)RowRowMaker)Column)Template)connection_summary)PostgresClientQuery PostgresQuery)Prepare)executefetchsend) capabilities) Transformer)PGconnPGresultceZdZUdjZded<ded<ded<eZded <d4d Zd5d6d Z d7d Z e d8d Z e d9dZ e d:dZe d;dZe ddZd?d@dZdAdZe dBdZdCdZ d?ddd dDdZ dEdZ dEdZddd dFdZ d? dGdZ d?dd dHd ZdId!Zd?dJd"Z d? dKd#Zd$dd% dLd&Z d? dMd'Z!dNd(Z"dOd)Z#d?dPd*Z$dNd+Z%e& dQd,Z'dRd-Z(dd dSd.Z)dTd/Z*dUd0Z+dVd1Z,dWd2Z-e d7d3Z.y)X 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 RowMaker[Row] _make_rowr#_pgconnztype[PostgresQuery] _query_clsc8||_t|_|j|_t j |j|_d|_ d|_ d|_ |jt|jt||j_y)Nr F)_connTEXTformatpgconnr*r AdaptersMapadapters _adapters arraysize_closed _last_query_resetr _loaders_changedr _register_loader_callback)self connections S/var/www/html/secretsanta/venv/lib/python3.12/site-packages/psycopg/_cursor_base.py__init__zBaseCursor.__init__=sv  !(( **:+>+>? )- 4;  ! !3t94 0cpg|_d|_d|_d|_d|_|d|_|rd|_yy)Nr)_resultspgresult_pos_iresult _rowcount_execmany_returning_query)r: reset_querys r<r7zBaseCursor._resetLs?(* )-    04 DK r>c ^|jjd|jj}t|j}|j rd}nB|j r4tj|j jj}nd}d|d|d|dt|dd S) N.closedz no result) __class__ __module__ __qualname__rr*r5rBr ExecStatusstatusnameid)r:clsinforSs r<__repr__zBaseCursor.__repr__Xs**+1T^^-H-H,IJ!$,,/ <<F ]]]]4==#7#78==F F3%r&D64|1==r>c|jS)z$The connection this cursor is using.)r-r:s r<r;zBaseCursor.connectioncszzr>c|jSN)r3rZs r<r2zBaseCursor.adaptershs ~~r>c|jS)z`True` if the cursor is closed.)r5rZs r<rKzBaseCursor.closedls||r>c|j}|rr|js9|jtk(s&|jtk(s|jt k(r-t |jDcgc]}t||c}Sycc}w)z A list of `Column` objects describing the current resultset. `!None` if the current resultset didn't return tuples. N)rBnfieldsrS TUPLES_OK SINGLE_TUPLE TUPLES_CHUNKranger)r:resis r< descriptionzBaseCursor.descriptionqsfmm  KKzzY&zz\)zz\)-23;;-?@F4O@ @As+Bc|jS)zk Number of records affected by the operation that produced the current result set. )rErZs r<rowcountzBaseCursor.rowcounts ~~r>cx|jxr|jjtk(}|r |jSdS)zsIndex of the next row to fetch in the current result set. `!None` if there is no result to fetch. N)rBrSr`rC)r:tupless r< rownumberzBaseCursor.rownumbers2 D4==#7#79#D"tyy,,r>cyr\)r:sizess r< setinputsizeszBaseCursor.setinputsizes r>Ncyr\rm)r:sizecolumns r< setoutputsizezBaseCursor.setoutputsizerpr>c|jt|jdz kr|j|jdzyy)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. r TN)rDlenrA_select_current_resultrZs r<nextsetzBaseCursor.nextsets; ==3t}}-1 1  ' ' (9 :r>cr|jr|jjnd}|r|jSdS)z{ The status tag of the current result set. `!None` if the cursor doesn't have a result available. N)rBcommand_statusdecode)r:msgs r< statusmessagezBaseCursor.statusmessages./3mmdmm**"szz|,,r>ctr\)NotImplementedErrorrZs r<_make_row_makerzBaseCursor._make_row_makers!!r>preparebinaryc#K|j|Ed{|j||}|j|||Ed{|jjr,|jjj Ed{||_|jjj|jEd{y777G7 w)z*Generator implementing `Cursor.execute()`.Nr) _start_query_convert_query_maybe_prepare_genr- _pipeline_communicate_genr6 _prepared maintain_gen)r:queryparamsrrpgqs r< _execute_genzBaseCursor._execute_gens$$U+++!!%0**3*OOO ::  zz++<<> > > ::''44TZZ@@@ ,O > AsFCC -CC ACC=CCC CCCc#LK|jj}|sJ|j|Ed{|sd|_|jJ||_d}|D]d}|r|j ||}||_d}nj||j|dEd{|jEd{f||_ |r|jdEd{|jjj|jEd{y77{7e7B7 w)zY Generator implementing `Cursor.executemany()` with pipelines available. NrTFr)flush)r-rrrErFrrGdumprrr6 _fetch_genrr)r:r params_seq returningpipelinefirstrrs r<_executemany_gen_pipelinez$BaseCursor._executemany_gen_pipelines" ::''x$$U+++DN''///#,  3F))%8!  ..sD.A A A002 2 2 3! ***6 6 6::''44TZZ@@@1 , B 2 7@sX/D$DA.D$ D!D$8D9$D$D 6D$D"D$D$D$ D$"D$c#K|j|Ed{|sd|_|jJ||_d}|D]L}|r|j||}||_d}nj ||j |dEd{N||_|jjj|jEd{y77G7 w)z] Generator implementing `Cursor.executemany()` with pipelines not available. NrTFr) rrErFrrGrrr6r-rr)r:rrrrrrs r<_executemany_gen_no_pipelinez'BaseCursor._executemany_gen_no_pipelines $$U+++DN''///#,   BF))%8!  ..sD.A A A B!::''44TZZ@@@' , B As4CC A.CC?CCCCCc#K|j||\}}|tjur|j||n|tjur{|j |||j jsSt|jEd{\}|jtk(r!tj||j|j||||j j j#|||}|j jr;d}||||f}|j jj$j'||fyt|jEd{} |(|j j j)|||| |j+| |j-| y717Tw)Nrencoding) _get_preparedrNO _execute_sendSHOULD _send_preparer-rrr*rS FATAL_ERROReerror_from_result _encoding_send_query_preparedrmaybe_add_to_cache result_queueappendvalidate_check_results _set_results) r:rrrpreprTresultkeyqueuedresultss r<rzBaseCursor._maybe_prepare_gensy''W5 d 7::    s6  2w~~%""4-zz+++24<<+@ @IV}} 311&4>>RR  % %dC % ? jj""55c4F ::  FtT* JJ - - 4 4dF^ D %T\\22 ? JJ ) )#tT7 C G$ '"3!A&3s&BGGCG/G0AGGcN|jjj||Sr\)r-rget)r:rrs r<rzBaseCursor._get_prepared?s!zz##''W55r>rc#K|j|Ed{|j||}|j||d|dkr td|dk(r|jj n1t jd|jj|||_ t|jEd{y77w)z2Generator to send the query for `Cursor.stream()`.NT)rforce_extendedr zsize must be >= 1)check) rrr ValueErrorr*set_single_row_moder!has_stream_chunkedset_chunked_rows_moder6r )r:rrrrrrs r<_stream_send_genzBaseCursor._stream_send_genDs$$U+++!!%0 3vdC !801 1 QY LL , , .  + +$ 7 LL . .t 4  %%% , &s"C CB*C C C  C c#Kt|jEd{}|y|j}|tk(s |tk(r=||_|j j|||r|j|_ |S|tk(s |tk(rA|r t|jEd{}|r |tk7rtjdy|j|S779w)N) set_loadersz1the operation in stream() didn't produce a result)rr*rSrarbrBr' set_pgresultrr)r` COMMAND_OKrProgrammingError_raise_for_result)r:rrdrSs r<_stream_fetchone_genzBaseCursor._stream_fetchone_genZs*/ *== ; \ !V|%;DM HH ! !#5 ! 9!%!5!5!7J y Fj$8!&t||!44"((G))#. .1 >5s(C&C"BC&*C$+C&21C&$C&c#,K|jrtjd|j|jr|j|ur!d|_t j ||_|jjEd{y7w)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) rKrInterfaceErrorr7r6rr"r'r-r)r:rs r<rzBaseCursor._start_queryusp ;;""#9: : D$4$4E$A#D ((.DH::**,,,sB B B Bc#6K|jjrtjd|j Ed{|st |t r3t|j}|j|||j}|j|}|j|dtt|jEd{x}dk7rtj d|j#|d|j%|y77Jw)z++C"&C ::   JJ . . 5 5LL22KKLL"'-- % "%  u||sf} LL * *  #mm!KK! +  LL # #EKK 0r>c`|j|j}|j|||Sr\)r+r'r)r:rrrs r<rzBaseCursor._convert_querys)oodhh' E6" r>c|stjd|D]G}|jx}tk7s|tk7s#|t k7s-|t k7s7|j|Iy)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 InternalErrorrSr`r EMPTY_QUERY COPY_BOTHr)r:rrdrSs r<rzBaseCursor._check_resultss]//"@A A ,C::%)3j(k)i'&&s+ ,r>ct|jx}tk(r!tj||j|t k(rtj d|tk(s|tk(s |tk(rtjdtjdtj|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: )rSrrrrPIPELINE_ABORTEDPipelineAbortedCOPY_INCOPY_OUTrrrrrRrTr:rrSs r<rzBaseCursor._raise_for_resultsmm #F 3%%ft~~F F ' '##$67 7 w &H"4)8K$$J //:2==;P;U;U:VW r>cv||_|j|x}|_|jj ||d|_|j tk(r|jj|_ n4|j tk7r!|jj}||nd|_ |j|_ y)zL Select one of the results in the cursor as the active one. )r/rNr@)rDrArBr'rrCrSr`ntuplesrErcommand_tuplesrr))r:rer/rdnrowss r<rwz!BaseCursor._select_current_results "mmA..dm c&1 :: "!]]22DN ZZ8 #MM00E&+&7URDN--/r>cD|j!||jdd|jdy|jr=|j }|jj||r|jdyy|D]%}|xj|j xsdz c_'y)Nr)rFrArwextendrEr)r:r first_batchrds r<rzBaseCursor._set_resultss  # # +&DMM!   ' ' *  % %"mm+K MM  )++A.  :#"4"4"99 :r>c|x}sytj||_|j}|jt |j kr#|j|j||_yy)zCallback called when self.adapters.set_loaders is called. Allow to change the loaders after the results have been returned already. N)rr"r'rCrDrvrArw)rVwselfoidloaderr:poss r<r8zBaseCursor._loaders_changed"sb $$T*ii ==3t}}- -  ' ' 6DI .r>c|jjr|jjjjt |j j ||j|j|jjjjdy|j j ||j|jy)N)r) r-rrrr r* send_preparerrr)r:rTrs r<rzBaseCursor._send_prepare4s ::   JJ . . 5 5LL--KK %   JJ - - 4 4T : LL % %dEKKU[[ % Qr>c | |j}n|rtnt}|jjrf|jjj j t|jj||j|j|y|jj||j|j|y)N)rr) r/rr.r-rrrr r*send_query_preparedrr)r:rTrrrs r<rzBaseCursor._send_query_preparedBs >++C"&C ::   JJ . . 5 5LL44JJ"%++"%   LL , ,cjj 3 - r>cB|jrtjd|jx}stjd|j x}t k(r|S|tk(r!tj||j|tk(rtjd|jrd|jjd}n& tj|j }d|d}tjd |#t"$r|d}Y.wxYw) Nrzno result availablerrz (command status: )z - unknownz (result status: z)the last operation didn't produce records)rKrrrBrrSr`rrrrrrzr{rrRrTr)r:rdrSdetail status_names r<_check_result_for_fetchz"BaseCursor._check_result_for_fetchYs ;;""#9: :}}$$$$%:; ;jj FY .J { "%%cDNNC C ' '##$67 7!!-c.@.@.G.G.I-J!L8"$--"7"<"c|j}|dk(r|j|z}n|dk(r|}ntd|dd|cxkr|jkst dt d||_y)Nrelativeabsolutez bad mode: z'. It should be 'relative' or 'absolute'rzposition out of bound)rrCrr IndexError)r:valuemoderdnewposs r<_scrollzBaseCursor._scrolls}**, : YY&F Z Fz$/VWX XF(S[[(45 5)45 5 r>c6|jdd|_y)z3Non-blocking part of closing. Common to sync/async.F)rHTN)r7r5rZs r<_closezBaseCursor._closes  & r>c.|jjSr\)r*rrZs r<rzBaseCursor._encodings||%%%r>)r;r)T)rHboolreturnNone)r str)r r)r zadapt.AdaptersMap)r r)r zlist[Column] | None)r int)r  int | None)rnz Sequence[Any]r r r\)rrrrsr r r )r  bool | None)r z str | None)r r() rrr Params | Nonerrrrr  PQGen[None])rrrzIterable[Params]rrr r)rrrrrrr r)rrrrr ztuple[Prepare, bytes]) rrrrrrrrr r r)rrr zPQGen[PGresult | None])rz Query | Noner r)rrrrr r)rrrrrrr r )rrrrr r)rzlist[PGresult]r r )rr$r r)rer r/zpq.Format | Noner r )rz#ReferenceType[BaseCursor[Any, Any]]rr rz type[Loader]r r )rTbytesrrr r )rTrrrrrr r )r r$)rr$r r )rr rr r r )r r )/__name__rPrQsplit __slots____annotations__rr+r=r7rXpropertyr;r2rKrfrhrkrortrxr}rrrrrrrrrrrrrrrwr classmethodr8rrrrrrrrmr>r<r&r&0sC EG   O&3J#3    >*--   --"!%A $" AAA  A  A A$!A!A(8!AEI!A !AFAA(8AEIA A> $" *# *# *#  *#  *#Z:>6 6+66 6!%& # &&&  &  & &,/6 - 9=##(5# #> %" +1+1 +1  +1  +1\59$1 ,&"04:"7>AKW " RIM -:E .4  &&r>r&N)D__doc__ __future__rtypingrrrrweakrefrr functoolsr collections.abcr r rrrrabcrrrrrrowsrr_columnr_compatrpq.miscr_queriesrr _preparingr generatorsrrr _capabilitiesr!r"pq.abcr#r$Formatr.rrRrrr`rrrrrarbrTransactionStatusACTIVEr&rmr>r<r,s- #88&.=='8,,' ( yy~~   mm'' ]] % % MM # # == ! ! --   MM # # mm'' }})) }})) ==11   $ $g &,-g &r>