2il+$dZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZdd lmZmZddl m!Z!ddl"m#Z#erddl$m%Z%ejLjNZ'Gdde#defZ(y)z Psycopg Cursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyoverload)contextmanager)IterableIterator)errors)pq)ParamsQueryQueryNoTemplate)CopyWriter)Row RowFactoryRowMaker)SelfTemplate)Pipeline) BaseCursor) ConnectionceZdZdZdZed#dZe d$dZdd d%fdZd&dZ d'd Zd(d Ze d)d Z e jd*d Z d+d Z e d,ddd d-dZ eddd d.dZ d,ddd d/dZ dd d0dZ d,ddd d1dZd2dZd3dZd4dZd5d6dZd7dZd&dZd8dZd9d:dZe d,dd  d;d!Zd(d"ZxZS)<CursorpsycopgcyNr)self connections M/var/www/html/secretsanta/venv/lib/python3.12/site-packages/psycopg/cursor.py__init__zCursor.__init__$s58cyr r)r!r" row_factorys r#r$zCursor.__init__'s r%N)r'cNt|||xs |j|_yr )superr$r' _row_factory)r!r"r' __class__s r#r$zCursor.__init__,s% $'A:+A+Ar%c|Sr rr!s r# __enter__zCursor.__enter__2 r%c$|jyr )close)r!exc_typeexc_valexc_tbs r#__exit__zCursor.__exit__5s r%c$|jy)zI Close the current cursor and free associated resources. N)_closer-s r#r1z Cursor.close=s  r%c|jS)z9Writable attribute to control how result rows are formed.r*r-s r#r'zCursor.row_factoryCs   r%cF||_|jr|||_yyr )r*pgresult _make_row)r!r's r#r'zCursor.row_factoryHs"' ==(.DN r%c$|j|Sr r9r-s r#_make_row_makerzCursor._make_row_makerNs  &&r%preparebinarycyr r)r!queryparamsr@rAs r#executezCursor.executeQsr%cyr r)r!rCr@rAs r#rEzCursor.execute[sr%c  |jj5|jj|j||||ddd|S#1swY|SxYw#tj $r}|j dd}~wwxYw)z= Execute a query or command to the database. r?N)_connlockwait _execute_gene _NO_TRACEBACKwith_traceback)r!rCrDr@rAexs r#rEzCursor.executeds} *  %%eVWV%T       *##D) ) *s4A/AAAAAB1BBF) returningc r |jj5tjr|jjr-|jj |j |||n||jj5|jj |j |||dddn,|jj |j|||dddy#1swYxYw#1swYyxYw#tj$r}|jdd}~wwxYw)zI Execute the same command with a sequence of input data. N) rHrIr is_supported _pipelinerJ_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerLrMrN)r!rC params_seqrPrOs r# executemanyzCursor.executemanyxs  * ((*zz++  ::5*iX "ZZ88: JJOO $ > >$):y!"JJOO99%YW#    ( *##D) ) *sMD A2D -C575D,D 5C> :DD D D D6 D11D6r rAsizec #"K|jjrtjd|jj 5 |jj |j||||d}|jj |j|x}rqt|jD]*}|jj||j,d}|jj |j|x}rq|jj t"k(r|jj% |jj |jdr- |jj |jdr- |jj |jddddy#tj$r}|jdd}~wwxYw#t&$rYlwxYw#t&$rYOwxYw#|jj t"k(r|jj% |jj |jdr- |jj |jdr-n#t&$rYnwxYw |jj |jdw#t&$rYwwxYwwxYw#1swYyxYww)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 moderYTFN)first)_pgconnpipeline_statusrLProgrammingErrorrHrIrJ_stream_send_gen_stream_fetchone_genrangentuples_txload_rowr<rMrNtransaction_statusACTIVE _try_cancel Exception) r!rCrDrArZr\resposrOs r#streamz Cursor.streams << ' '$$%OP P ZZ__   ))%T)R!ZZ__T-F-Fu-MNNcN$S[[1E"hh//T^^DDE!E"ZZ__T-F-Fu-MNNcN<<22f<JJ**,"jjood.G.Ge.G.TU #jjood.G.Ge.G.TU (A(A(A(NO5  ?? .''-- .%%<<22f<JJ**,"jjood.G.Ge.G.TU #jjood.G.Ge.G.TU$  (A(A(A(NO$=  sALLC G7L AH$+H LH+G<<HH" H LHL HLHL"8LAJ43L4 K =L?K L+K0/L0 K< 9L;K< <LLL Lc#RK|jr ||jsyyw)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. N)r;nextsetr-s r#resultszCursor.resultss- == ||~  s%'ct|j |cxkrt|jks'ntd|dt|jd|dkrt|j|z}|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!indexs r# set_resultzCursor.set_results DMM""e@c$--.@@s4==/A.BBVW  19 &.E ##E* r%c|j|j}|j|jkrG|jj |j|j }|xjdz c_|Sy)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_posrcrdrer<r!rjrecords r#fetchonezCursor.fetchonesa **, 99s{{ "XX&&tyy$..AF IINIMr%cB|j|j}|s |j}|jj |j t |j |z|j|j}|xj t|z c_|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` ) rxry arraysizerd load_rowsrzminrcr<rq)r!rZrjrecordss r# fetchmanyzCursor.fetchmanysz **,>>D(($$ IIs499t+S[[94>>  S\! r%c|j|j}|jj|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` )rxryrdrrzrcr<)r!rjrs r#fetchallzCursor.fetchallsR **,(($$TYY T^^LKK r%c|Sr rr-s r#__iter__zCursor.__iter__r/r%c|j|j}|j|jkrG|jj |j|j }|xjdz c_|Std)Nr zno more records to return)rxryrzrcrdrer< StopIterationr{s r#__next__zCursor.__next__!sh **, 99s{{ "XX&&tyy$..AF IINIM788r%cH|j|j||y)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)rx_scroll)r!valuemodes r#scrollz Cursor.scroll*s  UD!r%writerc#K |jj5|jj|j||t ||5}|dddddd|jdy#1swY#xYw#1swY'xYw#t j $r}|jdd}~wwxYww)zU Initiate a :sql:`COPY` operation and return an object to manage it. rNr) rHrIrJ_start_copy_genrrLrMrNrt)r! statementrDrcopyrOs r#rz Cursor.copy8s *   4 4Y GH$v.$J  ##A&    *##D) ) *sWB>B9BA:B B(B>:B ?BB BB;%B66B;;B>c>|jdur|jsv|jjr_|jj5|jj |jjj ddddyyyy#1swYyxYw)NFT)flush)_execmany_returningr;rHrSrIrJ _fetch_genr-s r#rxzCursor._fetch_pipelinePs  $ $E 1]] $$ M  4 4 ? ?d ? KL M M%# 2 M Ms ABB)r"zConnection[Row])r"Connection[Any]r'RowFactory[Row])r"rr'zRowFactory[Row] | None)returnr)r2ztype[BaseException] | Noner3zBaseException | Noner4zTracebackType | NonerNone)rr)rr)r'rrr)rz RowMaker[Row]r ) rCrrD Params | Noner@ bool | NonerArrr)rCrr@rrArrr) rCrrDrr@rrArrr)rCrrWzIterable[Params]rPboolrr) rCrrDrrArrZintrz Iterator[Row])rzIterator[Self])rurrr)rz Row | None)r)rZrr list[Row])rr)rr)relative)rrrstrrr)rrrDrrz Writer | NonerzIterator[Copy])__name__ __module__ __qualname__ __slots__rr$r.r5r1propertyr'setterr>rErXrlrorvr}rrrrrrrrx __classcell__)r+s@r#rr sJI 88  ) ;J  UYB)B;QB ,&%    !!// '!% $"       $"      !% $"      *PU**(8*HL* *B!%. # ...  .  . .` 4 & 9 "!%' !% '''  '  ''.Mr%rrN))__doc__ __future__rtypesrtypingrrr contextlibrcollections.abcr r r rLr abcrrrrrrrowsrrr_compatrrrSr _cursor_baserr"rTransactionStatusrgrrr%r#rsn #//%.//++#$&   $ $wMZ)3. /wMr%