Ytj&dZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZmZmZmZdd lmZmZmZmZdd lmZm Z e rddlm!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_CHECKINGAny)IteratorSequence)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 K/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg/_copy.pyr(z Copy.__init__6sK /// ) ((F l returnrc.||SN)_enterr,s r. __enter__zCopy.__enter__Ds  r/exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec0||dSr2)finish)r,r6r8r:s r.__exit__z Copy.__exit__Hs Gr/Iterator[Buffer]c#nK|x}r|V|x}dSdS)z5Implement block-by-block iteration on :sql:`COPY TO`.N)readr,datas r.__iter__z Copy.__iter__RsOiikk!d JJJiikk!d     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.readWs$ ##DNN$4$4555r/Iterator[tuple[Any, ...]]c#nK|x}|V|x}dSdS)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.rowsz Copy.rows_sE(v5LLL(v55555r/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_rowis&##D$6$6$8$8999r/buffer Buffer | strcl|j|x}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.writetsC>''// /4  KK       r/row Sequence[Any]cl|j|x}r||dSdS)z=Write a record to a table after a :sql:`COPY FROM` operation.N)rU write_rowr+)r,rVrDs r.rYzCopy.write_row~sA>++C00 04  KK       r/excc|jtkrU|s0|jx}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 % % &>--///4&KK%%% 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 |jx}rQ|jt |j|t |jx}OdSdS#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. rwN) rgetrGrHrr`r BaseExceptionr)r,rDexs r.workerzQueuedLibpqWriter.workers $+//+++$ V$$WT\4|%T%T%TUUU+//+++$ V V V V V $ $ $!#D        $sA*A00 B :BB rDrcj|jst|j|_|jr|jt |t kr|j|dStdt |t D],}|j|||t z-dS)Nr) rrrrryrrputrzr{s r.r*zQueuedLibpqWriter.writes| . --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__rabcrrtypesrtypingrrcollections.abcr r r rr _compatr_acompatrrrr _copy_baserrrr generatorsrrrr"rrGr ExecStatusr]COPY_OUTTransactionStatusrbrrr)rror/r.rs2 #"""""########%%%%%%%%........222222222222KKKKKKKKKKKK))))))))'&&&&&& -  = !  $y;y;y;y;y;8% &y;y;y;x     S   &,),),),),)&,),),)^????? ?????r/