R h&dZddlmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddl mZdd lmZdd lmZmZmZmZdd lmZmZmZmZdd lmZmZe rdd lm Z ddl!m"Z"ddl#m$Z$ej%j&Z&ej%j'Z'ej(j)Z)GddedZ*GddeZ+Gdde+Z,Gdde,Z-dS)z6 Objects to support the COPY protocol (sync version). ) annotations)ABCabstractmethod) TracebackType) TYPE_CHECKINGAnyIteratorSequence)errors)pq)Self)QueueWorkergatherspawn)MAX_BUFFER_SIZE PREFER_FLUSH QUEUE_SIZEBaseCopy)copy_endcopy_to)Buffer)Cursor) ConnectionceZdZUdZdZded<dddd(fd Zd)dZd*dZd+dZd,dZ d-dZ d.dZ d/d"Z d0d%Z d1d'ZxZS)2CopyajManage 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 cursor Cursor[Any]r! bool | None Writer | Nonect|||st|}||_|j|_dS)N)r!)super__init__ LibpqWriterr write_write)selfr"r!r __class__s X/var/www/html/01_SiteInternet/03_Maps/venv/lib/python3.11/site-packages/psycopg/_copy.pyr(z Copy.__init__5sK /// ) ((F l returnrc.||SN)_enterr,s r. __enter__zCopy.__enter__Cs  r/exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec0||dSr2)finish)r,r6r8r:s r.__exit__z Copy.__exit__Gs Gr/Iterator[Buffer]c#BK |}|sdS|V)z5Implement block-by-block iteration on :sql:`COPY TO`.TN)readr,datas r.__iter__z Copy.__iter__Qs3 99;;D JJJ  r/rcZ|j|S)z Read an unparsed row after a :sql:`COPY TO` operation. Return an empty string when the data is finished. ) connectionwait _read_genr4s r.rBz Copy.readYs$ ##DNN$4$4555r/Iterator[tuple[Any, ...]]c#BK |}|dS|V)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()`. TN)read_row)r,records r.rowsz Copy.rowsas0 ]]__F~LLL  r/tuple[Any, ...] | NonecZ|j|S)a 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()`. )rGrH _read_row_genr4s r.rLz Copy.read_rowns&##D$6$6$8$8999r/buffer Buffer | strcl|j|}|r||dSdS)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,rRrDs r.r*z Copy.writeysC~##F++   KK       r/row Sequence[Any]cl|j|}|r||dSdS)z=Write a record to a table after a :sql:`COPY FROM` operation.N)rU write_rowr+)r,rVrDs r.rYzCopy.write_rowsA~'',,   KK       r/excc|jtkrS|j}|r|||j|d|_dS|sdS|jj tkrdS|j |j |dS)aTerminate 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_INrUendr+r r> _finished_pgconntransaction_statusACTIVErG _try_cancelrH_end_copy_out_gen)r,rZrDs r.r>z Copy.finishs ?g % %>%%''D " D!!! K  s # # #!DNNN |.&88 O ' ' ) ) ) O !7!7!9!9 : : : : :r/)r"r#r!r$r r%)r0r)r6r7r8r9r:r;r0r<)r0r@)r0r)r0rJ)r0rO)rRrSr0r<)rVrWr0r<rZr9r0r<)__name__ __module__ __qualname____doc____annotations__r(r5r?rErBrNrLr*rYr> __classcell__r-s@r.rr"s  JNNN # $ # # # # # # # #6666     : : : :     ;;;;;;;;r/rzConnection[Any]c4eZdZdZed dZd d d ZdS)rzG A class to write copy data somewhere (for async connections). rDrr0r<cdS)zWrite some data to destination.NrCs r.r*z Writer.writes  r/NrZr9cdS)z Called when write operations are finished. If operations finished with an error, it will be passed to ``exc``. Nro)r,rZs r.r>z Writer.finishs r/rDrr0r<r2re)rfrgrhrirr*r>ror/r.rrsW   ^        r/rc0eZdZdZdZddZdd ZdddZd S)r)z@ An `Writer` to write copy data to a Postgres database. psycopg.copyr"r#cN||_|j|_|jj|_dSr2)r"rGpgconnr`)r,r"s r.r(zLibpqWriter.__init__s#  +- r/rDrr0r<c vt|tkr6|jt |j|t dStdt|tD]F}|jt |j|||tzt GdS)Nflushr)lenrrGrHrr`rranger,rDis r.r*zLibpqWriter.writes t99 ' ' O tzLibpqWriter.finishs  HS (>HH3HHC::dl4i@@DDD )/&&x d'C'CDDC%(5DK         s-A==BBr"r#rqr2re)rfrgrhrir(r*r>ror/r.r)r)sf J....    )))))))r/r)cFeZdZdZdZdfd ZddZdd Zddfd ZxZS)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. rsr"r#ct|tt|_d|_d|_dS)N)maxsize)r'r(rr_queue_worker _worker_error)r,r"r-s r.r(zQueuedLibpqWriter.__init__sB    %*:%>%>%> &* 37r/r0r<c |j}|sdS|jt |j|t R#t$r}||_Yd}~dSd}~wwxYw)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. TrwN) rgetrGrHrr`r BaseExceptionr)r,rDexs r.workerzQueuedLibpqWriter.workers $ V{((E$$WT\4|%T%T%TUUU  V  $ $ $!#D        $sA5A A1A,,A1rDrcj|jst|j|_|jr|jt |t kr|j|dStdt |t D],}|j|||t z-dS)Nr) rrrrryrrputrzr{s r.r*zQueuedLibpqWriter.write s| . --DL   %$ $ t99 ' ' KOOD ! ! ! ! !1c$ii99 ? ? Q_)<%< =>>>> ? ?r/NrZr9c|jd|jrt|jd|_|jr|jt |dS)Nr/)rrrrrr'r>)r,rZr-s r.r>zQueuedLibpqWriter.finishsj  < 4< DL   %$ $ sr/r)r0r<rqr2re) rfrgrhrir(rr*r>rkrls@r.rrs J888888$$$$$????&           r/rN).ri __future__rabcrrtypesrtypingrrr r r rr _compatr_acompatrrrr _copy_baserrrr generatorsrrrr"rrGr ExecStatusr]COPY_OUTTransactionStatusrbrrr)rror/r.rs0 #"""""########999999999999222222222222KKKKKKKKKKKK))))))))'&&&&&& -  = !  $A;A;A;A;A;8% &A;A;A;H     S   &,),),),),)&,),),)^BBBBB BBBBBr/