]j&SrSSKJr SSKJrJr SSKJr SSKJ r J r SSK J r J r SSKJr SS KJr SS KJr SS KJrJrJrJr SS KJrJrJrJr SS KJrJ r \ (aSSKJ!r! SSK"J#r# SSK$J%r% \RLRNr'\RLRPr(\RRRTr*"SS\S5r+"SS\5r,"SS\,5r-"SS\-5r.g)z6 Objects to support the COPY protocol (sync version). ) annotations)ABCabstractmethod) TracebackType) TYPE_CHECKINGAny)IteratorSequence)errors)pq)Self)QueueWorkergatherspawn)MAX_BUFFER_SIZE PREFER_FLUSH QUEUE_SIZEBaseCopy)copy_endcopy_to)Buffer)Cursor) Connectionc^\rSrSr%SrSrS\S'SSS.SU4SjjjrSS jrSS jrSS jr SS jr SS jr SSjr SSjr SSjrSSjrSrU=r$)Copy#aBManage an asynchronous :sql:`COPY` operation. :param cursor: the cursor where the operation is performed. :param binary: if `!True`, write binary format. :param writer: the object to write to destination. If not specified, write to the `!cursor` connection. Choosing `!binary` is not necessary if the cursor has executed a :sql:`COPY` operation, because the operation result describes the format too. The parameter is useful when a `!Copy` object is created manually and no operation is performed on the cursor, such as when using ``writer=``\ `~psycopg.copy.FileWriter`. psycopgWriterwriterN)binaryr!cr>[TU]XS9 U(d [U5nX0lURUlg)N)r")super__init__ LibpqWriterr!write_write)selfcursorr"r! __class__s J/home/mande/repo/quber/.venv/lib/python3.13/site-packages/psycopg/_copy.pyr% Copy.__init__6s2 / (F ll c&UR5 U$N)_enterr)s r, __enter__Copy.__enter__Ds  r.c&URU5 gr0)finish)r)exc_typeexc_valexc_tbs r,__exit__ Copy.__exit__Hs Gr.c#v# UR5=n(aUv UR5=n(aMgg7f)z5Implement block-by-block iteration on :sql:`COPY TO`.N)readr)datas r,__iter__ Copy.__iter__Rs1iik!d!Jiik!d!!s399cTURRUR55$)zk Read an unparsed row after a :sql:`COPY TO` operation. Return an empty string when the data is finished. ) connectionwait _read_genr2s r,r= Copy.readWs ##DNN$455r.c#f# UR5=nbUv UR5=nbMgg7f)z Iterate on the result of a :sql:`COPY TO` operation record by record. Note that the records returned will be tuples of unparsed strings or bytes, unless data types are specified using `set_types()`. N)read_row)r)records r,rows Copy.rows_s0(v5L(v5s+11cTURRUR55$)z Read a parsed row of data from a table after a :sql:`COPY TO` operation. Return `!None` when the data is finished. Note that the records returned will be tuples of unparsed strings or bytes, unless data types are specified using `set_types()`. )rCrD _read_row_genr2s r,rH Copy.read_rowis"##D$6$6$899r.clURRU5=n(aURU5 gg)z Write a block of data to a table after a :sql:`COPY FROM` operation. If the :sql:`COPY` is in binary format `!buffer` must be `!bytes`. In text mode it can be either `!bytes` or `!str`. N) formatterr'r()r)bufferr?s r,r' Copy.writets/>>''/ /4 / KK  0r.clURRU5=n(aURU5 gg)z=Write a record to a table after a :sql:`COPY FROM` operation.N)rP write_rowr()r)rowr?s r,rTCopy.write_row~s->>++C0 04 0 KK  1r.cUR[:Xa\U(d2URR5=n(aUR U5 UR R U5 SUlgU(dgURR[:wagURR5 URRUR55 g)zTerminate the copy operation and free the resources allocated. You shouldn't need to call this function yourself: it is usually called by exit. It is available if, despite what is documented, you end up using the `Copy` object outside a block. TN) _directionCOPY_INrPendr(r!r6 _finished_pgconntransaction_statusACTIVErC _try_cancelrD_end_copy_out_gen)r)excr?s r,r6 Copy.finishs ??g %>>--//4/KK% KK  s #!DN||..&8 OO ' ' ) OO !7!7!9 :r.)r[r(r!)r* Cursor[Any]r"z bool | Noner!z Writer | None)returnr)r7ztype[BaseException] | Noner8BaseException | Noner9zTracebackType | NonerdNone)rdzIterator[Buffer])rdr)rdzIterator[tuple[Any, ...]])rdztuple[Any, ...] | None)rQz Buffer | strrdrf)rUz Sequence[Any]rdrfrarerdrf)__name__ __module__ __qualname____firstlineno____doc____annotations__r%r3r:r@r=rJrHr'rTr6__static_attributes__ __classcell__r+s@r,rr#s J N # $ # # #  # #,&%    6 : ;;r.rzConnection[Any]c:\rSrSrSr\SSj5rSS SjjrSrg) r z? A class to write copy data somewhere (for async connections). cg)zWrite some data to destination.Nr>s r,r' Writer.writes r.Ncg)zq Called when write operations are finished. If operations finished with an error, it will be passed to ``exc``. Nrt)r)ras r,r6 Writer.finishs r.rtr?rrdrfr0rg) rhrirjrkrlrr'r6rnrtr.r,r r s%    r.r c>\rSrSrSrSrS SjrS SjrS S SjjrSrg) r&z8 An `Writer` to write copy data to a Postgres database. psycopg.copychXlURUlURRUlgr0)r*rCpgconnr\)r)r*s r,r%LibpqWriter.__init__s%  ++-- r.c H[U5[::a3URR[ UR U[ S95 g[S[U5[5H>nURR[ UR XU[-[ S95 M@ g)Nflushr)lenrrCrDrr\rranger)r?is r,r'LibpqWriter.writesw t9 ' OO t(>'?s3%HC::dll44i@DD )//&&x d'CDC%(5DKK   s /BB.-B.)r\rCr*r*rcrxr0rg) rhrirjrkrlr%r'r6rnrtr.r,r&r&s" J. ))r.r&c\^\rSrSrSrSrS U4SjjrS SjrS SjrS S U4SjjjrSr U=r $)QueuedLibpqWriterz `Writer` using a buffer to queue data to write. `write()` returns immediately, so that the main thread can be CPU-bound formatting messages, while a worker thread can be IO-bound waiting to write on the connection. r{cd>[TU]U5 [[S9UlSUlSUlg)N)maxsize)r$r%rr_queue_worker _worker_error)r)r*r+s r,r%QueuedLibpqWriter.__init__s+  %*:%> &* 37r.c.URR5=n(aVURR[ UR U[ S95 URR5=n(aMUgg![anX lSnAgSnAff=f)zPush data to the server when available from the copy queue. Terminate reading when the queue receives a false-y value, or in case of error. The function is designed to be run in a separate task. rN) rgetrCrDrr\r BaseExceptionr)r)r?exs r,workerQueuedLibpqWriter.workersp $++//++$+$$WT\\4|%TU++//++$++ $!#   $sA4A:: BBBc|UR(d[UR5UlUR(a URe[ U5[ ::aUR RU5 g[S[ U5[ 5H'nUR RXU[ -5 M) g)Nr) rrrrrrrputrrs r,r'QueuedLibpqWriter.writes|| -DL   $$ $ t9 ' KKOOD !1c$i9 _)< =>:r.c>URRS5 UR(a[UR5 SUlUR(a URe[ TU]U5 g)Nr.)rrrrrr$r6)r)rar+s r,r6QueuedLibpqWriter.finishsP  << 4<< DL   $$ $ sr.)rrrr)rdrfrxr0rg) rhrirjrkrlr%rr'r6rnrorps@r,rrs( J8 $?&  r.rN)/rl __future__rabcrrtypesrtypingrrcollections.abcr r r rr _compatr_acompatrrrr _copy_baserrrr generatorsrrrr*rrCr ExecStatusrYCOPY_OUTTransactionStatusr^rr r&rrtr.r,rs ##%.22KK)& --   == ! !   $ $y;8% &y;x S &,)&,)^? ?r.