j#dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddl mZdd lmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZddlm Z erddl!m"Z"ejFjHZ$Gdde defZ%y)z Psycopg Cursor object. ) annotations) TracebackType) TYPE_CHECKINGAnyIterableIteratoroverload)contextmanager)errors)pq)ParamsQuery)CopyWriter)Row RowFactoryRowMaker)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 U/var/www/html/kelly/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 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) r@rAr is_supported _pipelinerB_executemany_gen_pipeline_pipeline_nolock_executemany_gen_no_pipelinerDrErF)rrG params_seqrKrIs r! executemanyzCursor.executemanyds  *((*zz++  ::5*iX "ZZ88: JJOO $ > >$):y!";:JJOO99%YW#!;:!( *##D) ) *sMD A2D -C575D,D 5C> :DD D D D6 D11D6r r>sizec #"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 moderTTFN)first)_pgconnpipeline_statusrDProgrammingErrorr@rArB_stream_send_gen_stream_fetchone_genrangentuples_txload_rowr9rErFtransaction_statusACTIVE _try_cancel Exception) rrGrHr>rUrWresposrIs r!streamz Cursor.streams << ' '$$%OP P ZZ__  ))%T)R!ZZ__T-F-Fu-MNNcN$S[[1"hh//T^^DD 2!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|j|j}|j|jkrG|jj |j|j }|xjdz c_|Sy)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 N_fetch_pipeline_check_result_for_fetch_posr^r_r`r9)rrerecords 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 recordset. `!size` default to `!self.arraysize` if not specified. :rtype: Sequence[Row], with Row defined by `row_factory` ) rjrk arraysizer_ load_rowsrlminr^r9len)rrUrerecordss 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 recordset. :rtype: Sequence[Row], with Row defined by `row_factory` )rjrkr_rqrlr^r9)rrerts r!fetchallzCursor.fetchallsR **,(($$TYY T^^LKK r#c#FK|j|j}|j|jkrd|jj |j|j }|xjdz c_||j|jkrcyyw)Nr ri)rrerows r!__iter__zCursor.__iter__sr **,ii#++%((##DIIt~~>C IINIIii#++%s BB!B!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)rj_scroll)rvaluemodes r!scrollz Cursor.scrolls  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) r@rArB_start_copy_genrrDrErF_select_current_result)r statementrHrcopyrIs r!rz Cursor.copys *  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_returningr8r@rNrArB _fetch_genr+s r!rjzCursor._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)returnr)r/ztype[BaseException] | Noner0zBaseException | Noner1zTracebackType | NonerNone)rr)rr)r%rrr)rz RowMaker[Row]r) rGrrH Params | Noner= bool | Noner>rrr)rGrrRzIterable[Params]rKboolrr) rGrrHrr>rrUintr Iterator[Row])rz Row | None)r)rUrr list[Row])rr)rr)relative)r}rr~strrr)rrrHrrz Writer | NonerzIterator[Copy])__name__ __module__ __qualname__ __slots__r r"r,r2r.propertyr%setterr;rJrSrgrnrurwrzrr rrj __classcell__)r)s@r!rrsJI 88  ) ;J  UYB)B;QB ,&%    !!// ' !% $"      *PU**(8*HL* *B!%. # ...  .  . .` &  "!%' !% '''  '  ''.Mr#rrN)&__doc__ __future__rtypesrtypingrrrrr contextlibr r rDr abcrrrrrrowsrrr_compatrrNr _cursor_baserr rTransactionStatusrbrrr#r!rsh #CC%++$&   $ $xMZ)3. /xMr#