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)z7 Objects to support the COPY protocol (async version). ) annotations)ABCabstractmethod) TracebackType) TYPE_CHECKINGAny) AsyncIteratorSequence)errors)pq)Self)AQueueAWorkeragatheraspawn)MAX_BUFFER_SIZE PREFER_FLUSH QUEUE_SIZEBaseCopy)copy_endcopy_to)Buffer) AsyncCursor)AsyncConnectionceZdZUdZdZded<dddd(fd Zd)dZd*dZd+dZd,dZ d-dZ d.dZ d/d"Z d0d%Z d1d'ZxZS)2 AsyncCopyajManage 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`. psycopg AsyncWriterwriterN)binaryr cursorAsyncCursor[Any]r! bool | NoneAsyncWriter | Nonect|||st|}||_|j|_dS)N)r!)super__init__AsyncLibpqWriterr write_write)selfr"r!r __class__s Q/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg/_copy_async.pyr(zAsyncCopy.__init__3sK /// .%f--F l returnrc2K||SN)_enterr,s r. __aenter__zAsyncCopy.__aenter__As  r/exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec@K||d{VdSr2)finish)r,r6r8r:s r. __aexit__zAsyncCopy.__aexit__Es2 kk'"""""""""""r/AsyncIterator[Buffer]cK|d{Vx}r#|WV|d{Vx}!dSdS)z5Implement block-by-block iteration on :sql:`COPY TO`.N)readr,datas r. __aiter__zAsyncCopy.__aiter__Osu!YY[[(((((()d JJJJ"YY[[(((((()d     r/rcjK|j|d{VS)z Read an unparsed row after a :sql:`COPY TO` operation. Return an empty string when the data is finished. N) connectionwait _read_genr4s r.rBzAsyncCopy.readTs: _))$..*:*:;;;;;;;;;r/AsyncIterator[tuple[Any, ...]]cK|d{Vx}#|WV|d{Vx}!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.rowszAsyncCopy.rows\sk!% //////0v=LLLL!% //////0v=====r/tuple[Any, ...] | NonecjK|j|d{VS)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()`. N)rGrH _read_row_genr4s r.rLzAsyncCopy.read_rowfs<_))$*<*<*>*>?????????r/buffer Buffer | strc|K|j|x}r||d{VdSdS)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*zAsyncCopy.writeqsY>''// /4 $++d## # # # # # # # # # $ $r/row Sequence[Any]c|K|j|x}r||d{VdSdS)z=Write a record to a table after a :sql:`COPY FROM` operation.N)rU write_rowr+)r,rVrDs r.rYzAsyncCopy.write_row{sW>++C00 04 $++d## # # # # # # # # # $ $r/exccK|jtkra|s6|jx}r||d{V|j|d{Vd|_dS|sdS|jj tkrdS|j d{V|j |d{VdS)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. NT) _directionCOPY_INrUendr+r r> _finished_pgconntransaction_statusACTIVErG _try_cancelrH_end_copy_out_gen)r,rZrDs r.r>zAsyncCopy.finishs ?g % % ,>--///4,++d++++++++++$$S)) ) ) ) ) ) ) )!DNNN |.&88 /--// / / / / / / //&&t'='='?'?@@ @ @ @ @ @ @ @ @ @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.  J #%) # # # # # # # ##### <<<< @ @ @ @$$$$$$$$ AAAAAAAAr/rzAsyncConnection[Any]c4eZdZdZed dZd d d ZdS)rzG A class to write copy data somewhere (for async connections). rDrr0r<c KdS)zWrite some data to destination.NrCs r.r*zAsyncWriter.writes  r/NrZr9c KdS)z Called when write operations are finished. If operations finished with an error, it will be passed to ``exc``. Nro)r,rZs r.r>zAsyncWriter.finishs r/rDrr0r<r2re)rfrgrhrirr*r>ror/r.rrsW   ^        r/rc0eZdZdZdZddZdd ZdddZd S)r)zE An `AsyncWriter` to write copy data to a Postgres database. psycopg.copyr"r#cN||_|j|_|jj|_dSr2)r"rGpgconnr`)r,r"s r.r(zAsyncLibpqWriter.__init__s#  +- r/rDrr0r<c Kt|tkr<|jt |j|t d{VdStdt|tD]L}|jt |j|||tzt d{VMdS)Nflushr)lenrrGrHrr`rranger,rDis r.r*zAsyncLibpqWriter.writes t99 ' '/&&wt|T'V'V'VWW W W W W W W W W W1c$ii99  o** d1q?/B+B&C<  r/NrZr9c<K|r;dt|jd|}||jjd}nd} |jt|j|d{V}|g|j_ dS#tj $r|sYdSwxYw)Nzerror from Python: z - replace) typerhencoder` _encodingrGrHrr"_resultse QueryCanceled)r,rZmsgbmsgress r.r>zAsyncLibpqWriter.finishs  HS (>HH3HHC::dl4i@@DDD ),,XdlD-I-IJJJJJJJJC%(5DK         s3BBBr"r#rqr2re)rfrgrhrir(r*r>ror/r.r)r)sf J....    )))))))r/r)cFeZdZdZdZdfd ZddZdd Zddfd ZxZS)AsyncQueuedLibpqWriterz `AsyncWriter` 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(zAsyncQueuedLibpqWriter.__init__sB    &,Z&@&@&@ '+ 37r/r0r<cFK |jd{Vx}r]|jt |j|t d{V|jd{Vx}[dSdS#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. Nrw) rgetrGrHrr`r BaseExceptionr)r,rDexs r.workerzAsyncQueuedLibpqWriter.workers $!%!2!22222223$ o**DL$lCCC"&!2!22222223$      $ $ $!#D        $sA)r,rZr-s r.r>zAsyncQueuedLibpqWriter.finishskooc""""""""" < $,'' ' ' ' ' ' ' 'DL   %$ $ggnnS!!!!!!!!!!!r/r)r0r<rqr2re) rfrgrhrir(rr*r>rkrls@r.rrs J888888$$$$"EEEE& " " " " " " " " " " "r/rN)/ri __future__rabcrrtypesrtypingrrcollections.abcr r r rr _compatr_acompatrrrr _copy_baserrrr generatorsrrr cursor_asyncrconnection_asyncr ExecStatusr]COPY_OUTTransactionStatusrbrrr)rror/r.rsE #"""""########%%%%%%%%33333333666666666666KKKKKKKKKKKK))))))))2))))))111111 -  = !  ${A{A{A{A{A/0{A{A{A|     #   &,),),),),){,),),)^A"A"A"A"A"-A"A"A"A"A"r/