qj$dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z e rddl!m"Z"ejFjHZ$Gdde defZ%y)z Psycopg Cursor object. ) annotations) TracebackType)AnyIteratorIterable TYPE_CHECKINGoverload)contextmanager)pq)errors)QueryParams)CopyWriter)RowRowMaker RowFactory)Self)Pipeline) BaseCursor) ConnectionceZdZdZdZeddZe ddZdd d fdZd!dZ d"d Zd#d Ze d$d Z e jd%d Z d&d Z d'ddd d(dZ dd d)dZ d'ddd d*dZd+dZd,d-dZd.dZd/dZd0d1dZe d'dd d2dZd#dZxZS)3CursorpsycopgcyNr)self connections O/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/cursor.py__init__zCursor.__init__#s58cyrr)rr row_factorys r!r"zCursor.__init__&s r#N)r%cNt|||xs |j|_yr)superr"r% _row_factory)rr r% __class__s r!r"zCursor.__init__+s% $'A:+A+Ar#c|Srrrs r! __enter__zCursor.__enter__1s r#c$|jyr)close)rexc_typeexc_valexc_tbs r!__exit__zCursor.__exit__4s r#c$|jy)zI Close the current cursor and free associated resources. N)_closer+s r!r.z Cursor.close<s  r#c|jS)z9Writable attribute to control how result rows are formed.r(r+s r!r%zCursor.row_factoryBs   r#cF||_|jr|||_yyr)r(pgresult _make_row)rr%s r!r%zCursor.row_factoryGs"' ==(.DN r#c$|j|Srr6r+s r!_make_row_makerzCursor._make_row_makerMs  &&r#preparebinaryc  |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)rqueryparamsr=r>exs r!executezCursor.executePsx * %%eVWV%T!  !   *##D) ) *s4A/AAAAAB1BBF) returningc & tjr|jj5|jj}|r,|jj |j |||ddds|jj5|jj5|jj |j |||ddddddy|jj5|jj |j|||dddyy#1swYxYw#1swYkxYw#1swYyxYw#1swYyxYw#tj$r}|jdd}~wwxYw)zI Execute the same command with a sequence of input data. N) r is_supportedr@rA _pipelinerB_executemany_gen_pipelinepipeline_executemany_gen_no_pipelinerDrErF)rrG params_seqrKprIs r! executemanyzCursor.executemanyds'  *$$&ZZ__ ,,A  ::5*iX%,,.   ::5*iX1@.. ZZ__JJOO99%YW%_ %_1@.. %_ *##D) ) *s*E'AD71$E'E,-EE!E'*E'-E-E'7E<E'E EEE'E'E$ E'$E''F:F  Fr r>sizec #K|jjrtjd|jj 5 |jj |j||||d}|jj |j|rkt|D]0}|jj||j}|n|2d}|jj |j|rk|jjt k(r|jj# |jj |jdr- |jj |jdr- |jj |jddddy#tj$r}|jdd}~wwxYw#t$$rYlwxYw#t$$rYOwxYw#|jjt 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 moderUTNF)first)_pgconnpipeline_statusrDProgrammingErrorr@rArB_stream_send_gen_stream_fetchone_genrange_txload_rowr9rErFtransaction_statusACTIVE _try_cancel Exception) rrGrHr>rVrXposrecrIs r!streamz Cursor.streams| << ' '$$%OP P ZZ__  ))%T)Rjjood&?&?&FG$T{"hh//T^^D;!! + "E jjood&?&?&FG<<22f<JJ**,"jjood.G.Ge.G.TU #jjood.G.Ge.G.TU (A(A(A(NO;_?? .''-- .%%<<22f<JJ**,"jjood.G.Ge.G.TU #jjood.G.Ge.G.TU$  (A(A(A(NO$=!_sALK;CG 7K;AG<+H  LG9#G44G99H< HK;HK; HK;HK;8K8AJ,+K8, J8 5K87J8 8K8<+K('K8( K4 1K83K4 4K88K;;LLc|j|j|jj|j|j }||xjdz c_|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` r )_fetch_pipeline_check_result_for_fetchr_r`_posr9)rrecords r!fetchonezCursor.fetchonesR  $$&""499dnn=   IINI r#cr|j|j|jsJ|s |j}|jj |j t|j |z|jj|j}|xj t|z c_|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` ) rirjr8 arraysizer_ load_rowsrkminntuplesr9len)rrVrecordss r! fetchmanyzCursor.fetchmanys  $$&}}}>>D(($$ IIs499t+T]]-B-BCT^^  S\! r#c"|j|j|jsJ|jj |j |jj |j}|jj |_|S)z Return all the remaining records from the current recordset. :rtype: Sequence[Row], with Row defined by `row_factory` )rirjr8r_rprkrrr9)rrts r!fetchallzCursor.fetchallsh  $$&}}}(($$TYY 0E0Et~~VMM)) r#c#Kjjdfd } |j}|yxjdz c_|/w)NcPjj|jSr)r_r`r9)rers r!loadzCursor.__iter__..loads88$$S$..9 9r#r )reintreturn Row | None)rirjrk)rrzrows` r!__iter__zCursor.__iter__sU  $$& :tyy/C{ IINII sAAcH|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)ri_scroll)rvaluemodes r!scrollz Cursor.scrolls  UD!r#writerc#K |jj5|jj|j||dddt ||5}|ddd|jdy#1swY5xYw#1swY'xYw#t j $r}|jdd}~wwxYww)zU Initiate a :sql:`COPY` operation and return an object to manage it. Nrr) r@rArB_start_copy_genrrDrErF_select_current_result)r statementrHrcopyrIs r!rz Cursor.copys *  4 4Y GH!d6*d + ##A&!+* *##D) ) *sWB>B,A:BB B(B>:B?BB BB;%B66B;;B>c>|jdur|jsv|jjr_|jj5|jj |jjj ddddyyyy#1swYyxYw)NFT)flush)_execmany_returningr8r@rNrArB _fetch_genr+s r!rizCursor._fetch_pipelinesu  $ $E 1]] $$  4 4 ? ?d ? KL!%# 2!s ABB)r zConnection[Row])r Connection[Any]r%RowFactory[Row])r rr%zRowFactory[Row] | None)r|r)r/ztype[BaseException] | Noner0zBaseException | Noner1zTracebackType | Noner|None)r|r)r|r)r%rr|r)r|z RowMaker[Row]r) rGrrH Params | Noner= bool | Noner>rr|r)rGrrRzIterable[Params]rKboolr|r) rGrrHrr>rrVr{r| Iterator[Row])r|r})r)rVr{r| list[Row])r|r)r|r)relative)rr{rstrr|r)rrrHrrz Writer | Noner|zIterator[Copy])__name__ __module__ __qualname__ __slots__r r"r,r2r.propertyr%setterr;rJrTrgrmrurwrrr rri __classcell__)r)s@r!rrsJI 88  ) ;J  UYB)B;QB ,&%    !!// ' !% $"      *PU**(8*HL* *B!%1 # 111  1  1 1f (   "!%' !% '''  '  ''.Mr#rrN)&__doc__ __future__rtypesrtypingrrrrr contextlibr r r rDabcrrrrrrowsrrr_compatrrNr _cursor_baserr rTransactionStatusrbrrr#r!rsh #CC%++$&   $ $AMZ)3. /AMr#