j%dZddlmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddl mZdd lmZdd lmZmZmZmZdd lmZmZmZmZdd lmZmZe rdd lm Z ddl!m"Z"ddl#m$Z$ejJjLZ&ejJjNZ'ejPjRZ)GddedZ*GddeZ+Gdde+Z,Gdde,Z-y)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<ddd dfdZddZ dd Zdd Zdd Z dd Z dd Z ddZ ddZ ddZxZS)CopyajManage 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 crt||||s t|}||_|j|_y)N)r!)super__init__ LibpqWriterr write_write)selfcursorr!r __class__s T/var/www/html/kelly/kelly-backend/venv/lib/python3.12/site-packages/psycopg/_copy.pyr$z Copy.__init__5s6 / (F ll c&|j|SN)_enterr(s r+ __enter__zCopy.__enter__Cs  r,c&|j|yr.)finish)r(exc_typeexc_valexc_tbs r+__exit__z Copy.__exit__Gs Gr,c#`K|jx}r||jx}ryyw)z5Implement block-by-block iteration on :sql:`COPY TO`.N)readr(datas r+__iter__z Copy.__iter__Qs/iik!d!Jiik!d!)..cT|jj|jS)z Read an unparsed row after a :sql:`COPY TO` operation. Return an empty string when the data is finished. ) connectionwait _read_genr0s r+r9z Copy.readVs ##DNN$455r,c#`K|jx}||jx}yyw)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^s1(v5L(v5r=cT|jj|jS)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()`. )r?r@ _read_row_genr0s r+rCz Copy.read_rowhs"##D$6$6$899r,cb|jj|x}r|j|yy)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&z Copy.writess/>>''/ /4 / KK  0r,cb|jj|x}r|j|yy)z=Write a record to a table after a :sql:`COPY FROM` operation.N)rI write_rowr')r(rowr;s r+rLzCopy.write_row}s->>++C0 04 0 KK  1r,c|jtk(rR|s-|jjx}r|j ||j j |d|_y|sy|jjtk7ry|jj|jj|jy)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_INrIendr'r r3 _finished_pgconntransaction_statusACTIVEr? _try_cancelr@_end_copy_out_gen)r(excr;s r+r3z Copy.finishs ??g %>>--//4/KK% KK  s #!DN||..&8 OO ' ' ) OO !7!7!9 :r,)r) Cursor[Any]r!z bool | Noner z Writer | None)returnr)r4ztype[BaseException] | Noner5BaseException | Noner6zTracebackType | NonerZNone)rZzIterator[Buffer])rZr)rZzIterator[tuple[Any, ...]])rZztuple[Any, ...] | None)rJz Buffer | strrZr\)rMz Sequence[Any]rZr\rXr[rZr\)__name__ __module__ __qualname____doc____annotations__r$r1r7r<r9rErCr&rLr3 __classcell__r*s@r+rr"s J N # $ # # #  #,&%    6 : ;r,rzConnection[Any]c,eZdZdZeddZdddZy)rzG A class to write copy data somewhere (for async connections). cy)zWrite some data to destination.Nr:s r+r&z Writer.writes r,Ncy)z Called when write operations are finished. If operations finished with an error, it will be passed to ``exc``. Nrg)r(rXs r+r3z Writer.finishs r,r;rrZr\r.r])r^r_r`rarr&r3rgr,r+rrs    r,rc.eZdZdZdZddZddZd d dZy) r%z@ An `Writer` to write copy data to a Postgres database. psycopg.copycj||_|j|_|jj|_yr.)r)r?pgconnrS)r(r)s r+r$zLibpqWriter.__init__s'  ++-- r,c Pt|tkr6|jjt |j |t ytdt|tD]A}|jjt |j |||tzt Cy)Nflushr)lenrr?r@rrSrranger(r;is r+r&zLibpqWriter.writesy t9 ' OO t(>'?s3%HC::dll44i@DD )//&&x d'CDC%(5DKK   s/B B#"B#r)rYrir.r])r^r_r`rar$r&r3rgr,r+r%r%s J. )r,r%cFeZdZdZdZdfd ZddZd dZd d fd 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. rkcjt||tt|_d|_d|_y)N)maxsize)r#r$rr_queue_worker _worker_error)r(r)r*s r+r$zQueuedLibpqWriter.__init__s+  %*:%> &* 37r,c |jjx}rS|jjt |j |t |jjx}rRyy#t$r}||_Yd}~yd}~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. roN) rgetr?r@rrSr BaseExceptionr)r(r;exs r+workerzQueuedLibpqWriter.workerso $++//++$+$$WT\\4|%TU++//++$+ $!#D   $sA-A22 B ;BB cd|jst|j|_|jr |jt |t kr|j j|ytdt |t D]'}|j j|||t z)y)Nr) rrrrrqrrputrrrss r+r&zQueuedLibpqWriter.writes|| -DL   $$ $ t9 ' KKOOD !1c$i9 Q_)< =>:r,c|jjd|jrt|jd|_|jr |jt ||y)Nr,)rrrrrr#r3)r(rXr*s r+r3zQueuedLibpqWriter.finishsP  << 4<< DL   $$ $ sr,r)rZr\rir.r]) r^r_r`rar$rr&r3rcrds@r+rrs( J8 $?&  r,rN).ra __future__rabcrrtypesrtypingrrr r r r{r _compatr_acompatrrrr _copy_baserrrr generatorsrrrr)rr?r ExecStatusrPCOPY_OUTTransactionStatusrUrrr%rrgr,r+rs ##9922KK)& --   == ! !   $ $y;8% &y;x S &,)&,)^? ?r,