hZ$UdZddlmZddlZddlmZddlmZmZddl m Z ddl m Z dd l mZmZdd lmZmZmZdd lmZdd lmZmZmZdd lmZer&ddlmZddlmZm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(dZ)de*d<e jVjXZ,e jVjZZ-e j\j^Z/e j`jbZ1ejddZ3GddZ4Gdde4Z5Gdde4Z6y)z commands pipeline management ) annotationsN) TracebackType) TYPE_CHECKINGAny)errors)pq)PipelineCommandPQGen)DequeSelf TypeAlias)connection_summary) fetch_manypipeline_communicatesend) capabilities)PGresult)KeyPrepare) Connection) BaseCursor)BaseConnection)AsyncConnectionzEtuple[BaseCursor[Any, Any], tuple[Key, Prepare, bytes] | None] | Noner PendingResultpsycopgceZdZUded<ded<ddZddZeddZeddZ dd Z dd Z dd Z dd Z dd ZddZddZddZy) BasePipelinezDeque[PipelineCommand] command_queuezDeque[PendingResult] result_queuec||_|j|_tt|_tt |_d|_y)Nr)_connpgconnr r rrr level)selfconns K/var/www/tech_portal/venv/lib/python3.12/site-packages/psycopg/_pipeline.py__init__zBasePipeline.__init__/s; kk "?35!-02 c|jjd|jj}t|jj }d|d|dt |ddS)N.< z at 0xx>) __class__ __module__ __qualname__rr"r#id)r%clsinfos r'__repr__zBasePipeline.__repr__6s]**+1T^^-H-H,IJ!$**"3"343%qfRXaL22r)cTtj|jjSN)r PipelineStatusr#pipeline_statusr%s r'statuszBasePipeline.status;s  !r$r#enter_pipeline_modertransaction_statusACTIVE _sync_genr;s r' _enter_genzBasePipeline._enter_genDsm!!- ::? KK + + -   4;;#A#AV#K~~' ' ' a  (sA=B?BBcV|xjdzc_|jdk(r:|jjtk7r |jj yyy#t j $r5}|rtjd||n|jdYd}~yd}~wwxYw)Nrrzerror ignored exiting %r: %s) r$r#r<BADexit_pipeline_modeeOperationalErrorloggerwarningwith_traceback)r%excexc2s r'_exitzBasePipeline._exitRs a ::?t{{11S8 4 ..0 9?%% 4NN#A4N--d33O  4sA B(3+B##B(c#K|j|jEd{|jdEd{y77w)NFflush _enqueue_sync_communicate_gen _fetch_genr;s r'rEzBasePipeline._sync_gen`sA ((***???/// +/s!$A AA AA A c#K |j|jEd{|jdEd{y77#|jdEd{7wxYww)z_ Exit current pipeline by sending a Sync and fetch back all remaining results. NTrSrUr;s r' _exit_genzBasePipeline._exit_genesZ 3    ,,. . .T2 2 2 / 2tT2 2 2sIA+#A AA A+A A+A A+ A(!A$"A((A+c#Kt|j|jEd{}d}|D]/}|jj } |j ||1||y7?#t j$r}||}Yd}~Ud}~wwxYww)zqCommunicate with pipeline to send commands and possibly fetch results, which are then processed. N)rr#rr popleft_process_resultsrJError)r%fetched exceptionresultsqueuedrOs r'rWzBasePipeline._communicate_genos2$++t?Q?QRR  $G&&..0F $%%fg6 $  O !S 77 $$ #I $s9$B A&%B  A(B (B ;B?B B  B c#K|jsy|r7|jjt|jEd{d}|jrZt |jEd{x}sn:|jj } |j |||jrZ||y7q7H#tj$r}||}Yd}~2d}~wwxYww)a^Fetch available results from the connection and process them with pipeline queued items. If 'flush' is True, a PQsendFlushRequest() is issued in order to make sure results can be fetched. Otherwise, the caller may emit a PQpipelineSync() call to ensure the output buffer gets flushed before fetching. N) r r#send_flush_requestrrr\r]rJr^)r%rTr`rarbrOs r'rXzBasePipeline._fetch_gens    KK * * ,DKK( ( ( +5dkk+B BCGC&&..0F $%%fg6  O ! )!C77 $$ #I $sTACB6*C/B80!CB:$ C1C8C:C CCCCc|k|\}|jtk(r+tj||jj |jt k(rtjdy|\}}|r.|\}}}|jjj|||||j||j|y)arProcess a results set fetched from the current pipeline. This matches 'results' with its respective element in the pipeline queue. For commands (None value in the pipeline queue), results are checked directly. For prepare statement creation requests, update the cache. Otherwise, results are attached to their respective cursor. N)encodingzpipeline aborted) r< FATAL_ERRORrJerror_from_resultr# _encodingPIPELINE_ABORTEDPipelineAbortedr" _preparedvalidate_check_results _set_results) r%rbraresultcursorprepinfokeyprepnames r'r]zBasePipeline._process_resultss >IV}} +))&4;;;P;PQQ"22''(:;;3 & FH"*T4 &&//T4I  ! !' *    (r)c|jj|jj|jjdy)z#Enqueue a PQpipelineSync() command.N)rappendr# pipeline_syncr r;s r'rVzBasePipeline._enqueue_syncs4 !!$++";";<   &r)N)r&zBaseConnection[Any]returnNone)rystr)ryzpq.PipelineStatus)rybool)ry PQGen[None])rOBaseException | Noneryrz)rTr|ryr})rbrrazlist[PGresult]ryrzryrz)__name__r1r2__annotations__r(r6propertyr< classmethodr?rFrQrErZrWrXr]rVr)r'rr+sh))&&3 >>++  40 3 >).'r)rc\eZdZUdZdZded<d fd Zd dZd dZ d dZxZ S) Pipelinez(Handler for connection in pipeline mode.rConnection[Any]r"c$t||yr8superr(r%r&r0s r'r(zPipeline.__init__ r)c |jj5|jj|jdddy#1swYyxYw#tj $r}|j dd}~wwxYw)zkSync the pipeline, send any pending command and receive and process all available results. Nr"lockwaitrErJ _NO_TRACEBACKrNr%exs r'syncz Pipeline.syncsb * 2  01 2 2 2 *##D) ) *s4A*A A AAAB*A;;Bc|jj5|jj|jddd|S#1swY|SxYwr8r"rrrFr;s r' __enter__zPipeline.__enter__s> ZZ__ / JJOODOO- . /  / s *A  Ac~ |jj5|jj|jddd|j|y#1swYxYw#t$r5}|rt j d||n|jdYd}~Wd}~wwxYw#|j|wxYwNz error ignored terminating %r: %s r"rrrZ ExceptionrLrMrNrQr%exc_typeexc_valexc_tbrPs r'__exit__zPipeline.__exit__s  2  01 2 JJw  2 2 0A4N))$//O 0 JJw s@A(*AA(A%!A(( B&1+B!B)!B&&B))B<)r&rryrzrryr rztype[BaseException] | Nonerr~rzTracebackType | Noneryrz) rr1r2__doc__rr(rrr __classcell__r0s@r'rrsK2J *  , & %    r)rc\eZdZUdZdZded<d fd Zd dZd dZ d dZxZ S) AsyncPipelinez.Handler for async connection in pipeline mode.rAsyncConnection[Any]r"c$t||yr8rrs r'r(zAsyncPipeline.__init__rr)c\K |jj4d{|jj|jd{dddd{y7F77 #1d{7swYyxYw#tj $r}|j dd}~wwxYwwr8rrs r'rzAsyncPipeline.syncs *zz 8 8jjoodnn&6777 8 8 87 8 8 8 8 *##D) ) *sB,BA%B-A+A'A+ BA) B$B,%B'A+)B+A=1A4 2A=9B<B,=BB)B$$B))B,cK|jj4d{|jj|jd{dddd{|S7G77 #1d{7swY|SxYwwr8rr;s r' __aenter__zAsyncPipeline.__aenter__sk::?? 5 5**//$//"34 4 4 5 5  5 4 5 5 5 5 sVBA%B-A+A'A+ BA)B'A+)B+A>1A4 2A>9BcK |jj4d{|jj|jd{dddd{|j|y7W7(7#1d{7swY*xYw#t$r5}|rt j d||n|jdYd}~fd}~wwxYw#|j|wxYwwrrrs r' __aexit__zAsyncPipeline.__aexit__s zz 8 8jjoodnn&6777 8 8 JJw  87 8 8 8 8 0A4N))$//O 0 JJw sC(BA6B-A<A8A< BA: B$C(6B8A<:B<BB B B C+C C CCC%%C()r&rryrzrrr) rr1r2rrr(rrrrrs@r'rrsK8J *  , & %    r)r)7r __future__rloggingtypesrtypingrrrrJr abcr r _compatr r rpq.miscr generatorsrrr _capabilitiesrpq.abcr _preparingrr connectionr _cursor_baser_connection_baserconnection_asyncrrr ExecStatusrgrj ConnStatusrHTransactionStatusrD getLoggerrLrrrrr)r'rs #%'++'>>' (&(01Lymm'' ==11mm   $ $   9 %M'M'`( |( V% L% r)