o "4g&@s0dZddlmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddl mZdd lmZdd lmZmZmZmZdd lmZmZdd lmZmZmZmZe rhdd lm Z ddl!m"Z"ddl#m$Z$ej%j&Z&ej%j'Z'ej(j)Z)GdddedZ*GdddeZ+Gddde+Z,Gddde,Z-dS)z6 Objects to support the COPY protocol (sync version). ) annotations)ABCabstractmethod) TracebackType)AnyIteratorSequence TYPE_CHECKING)pq)errors)Self)BaseCopyMAX_BUFFER_SIZE QUEUE_SIZE PREFER_FLUSH)copy_tocopy_end)spawngatherQueueWorker)Buffer)Cursor) ConnectioncseZdZUdZdZded<dddd2fd d Zd3ddZd4ddZd5ddZd6dd Z d7d"d#Z d8d%d&Z d9d)d*Z d:d-d.Z d;d0d1ZZS)<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)binaryrcursor Cursor[Any]r bool | None Writer | Nonecs.tj||d|st|}||_|j|_dS)N)r)super__init__ LibpqWriterrwrite_write)selfr rr __class__V/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/_copy.pyr%5s  z Copy.__init__returnr cCs ||SN)_enterr)r,r,r- __enter__CszCopy.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonecCs||dSr/)finish)r)r3r5r7r,r,r-__exit__Gsz Copy.__exit__Iterator[Buffer]ccs |}|s dS|Vq)z5Implement block-by-block iteration on :sql:`COPY TO`.TN)readr)datar,r,r-__iter__Qsz Copy.__iter__rcC|j|S)z Read an unparsed row after a :sql:`COPY TO` operation. Return an empty string when the data is finished. ) connectionwait _read_genr1r,r,r-r=Ysz Copy.readIterator[tuple[Any, ...]]ccs |}|dur dS|Vq)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)recordr,r,r-rowsasz Copy.rowstuple[Any, ...] | NonecCrA)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()`. )rBrC _read_row_genr1r,r,r-rFns z Copy.read_rowbuffer Buffer | strcC"|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)rKr?r,r,r-r'ys z Copy.writerow Sequence[Any]cCrM)z=Write a record to a table after a :sql:`COPY FROM` operation.N)rN write_rowr()r)rOr?r,r,r-rQs zCopy.write_rowexccCsn|jtkr|j}|r|||j|d|_dS|s dS|jj t kr(dS|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_INrNendr(rr: _finished_pgconntransaction_statusACTIVErB _try_cancelrC_end_copy_out_gen)r)rRr?r,r,r-r:s       z Copy.finish)r r!rr"rr#)r.r )r3r4r5r6r7r8r.r9)r.r<)r.r)r.rE)r.rI)rKrLr.r9)rOrPr.r9rRr6r.r9)__name__ __module__ __qualname____doc____annotations__r%r2r;r@r=rHrFr'rQr: __classcell__r,r,r*r-r"s      rzConnection[Any]c@s*eZdZdZed ddZddd d ZdS)rzG A class to write copy data somewhere (for async connections). r?rr.r9cCdS)zWrite some data to destination.Nr,r>r,r,r-r'sz Writer.writeNrRr6cCrc)z Called when write operations are finished. If operations finished with an error, it will be passed to ``exc``. Nr,r)rRr,r,r-r:sz Writer.finishr?rr.r9r/r\)r]r^r_r`rr'r:r,r,r,r-rs  rc@s4eZdZdZdZdddZdd d ZddddZd S)r&z@ An `Writer` to write copy data to a Postgres database. psycopg.copyr r!cCs||_|j|_|jj|_dSr/)r rBpgconnrWr)r r,r,r-r%szLibpqWriter.__init__r?rr.r9c Csft|tkr|jt|j|tddStdt|tD]}|jt|j|||ttdqdS)Nflushr)lenrrBrCrrWrranger)r?ir,r,r-r's zLibpqWriter.writeNrRr6cCst|rdt|jd|}||jjd}nd}z |jt|j|}Wntj y2|s/YdSw|g|j _ dS)Nzerror from Python: z - replace) typer_encoderW _encodingrBrCre QueryCanceledr _results)r)rRmsgbmsgresr,r,r-r:szLibpqWriter.finishr r!rer/r\)r]r^r_r`r%r'r:r,r,r,r-r&s   r&csJeZdZdZdZdfdd Zdd d Zdd dZddfdd ZZS)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. rfr r!cs(t|ttd|_d|_d|_dS)N)maxsize)r$r%rr_queue_worker _worker_errorrhr*r,r-r%s   zQueuedLibpqWriter.__init__r.r9c Cs^z |j}|s WdS|jt|j|tdqty.}z ||_WYd}~dSd}~ww)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. TriN) r|getrBrCrrWr BaseExceptionr~)r)r?exr,r,r-workers zQueuedLibpqWriter.workerr?rcCsl|js t|j|_|jr|jt|tkr|j|dStdt|tD]}|j|||tq%dS)Nr) r}rrr~rkrr|putrlrmr,r,r-r' s  zQueuedLibpqWriter.writeNrRr6cs>|jd|jrt|jd|_|jr|jt|dS)N)r|rr}rr~r$r:rdr*r,r-r:s  zQueuedLibpqWriter.finishry)r.r9rer/r\) r]r^r_r`r%rr'r:rbr,r,r*r-rzs  rzN).r` __future__rabcrrtypesrtypingrrrr r r rs_compatr _copy_baserrrr generatorsrr_acompatrrrrrr rrBr ExecStatusrTCOPY_OUTTransactionStatusrYrrr&rzr,r,r,r-s.        /