R hi$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 j+j,Z,e j+j-Z-e j.j/Z/e j0j1Z1ej2dZ3GddZ4Gdde4Z5Gdde4Z6dS)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<d!d Zd"d Zed#d Zed$dZ d%dZ d&dZ d%dZ d%dZ d%dZd'dZd(dZd)dZd S)* BasePipelinezDeque[PipelineCommand] command_queuezDeque[PendingResult] result_queueconnBaseConnection[Any]returnNonec||_|j|_tt|_tt |_d|_dS)Nr)_connpgconnr r rrr level)selfr!s \/var/www/html/01_SiteInternet/03_Maps/venv/lib/python3.11/site-packages/psycopg/_pipeline.py__init__zBasePipeline.__init__/sC k "?355!-022 strc|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[*JJT^-HJJ!$*"344232222RXX22222r,pq.PipelineStatusc>tj|jjSN)r PipelineStatusr'pipeline_statusr)s r*statuszBasePipeline.status;s !<===r,boolc(tjS)zCReturn `!True` if the psycopg libpq wrapper supports pipeline mode.)r has_pipeline)r8s r* is_supportedzBasePipeline.is_supported?s(***r, PQGen[None]c# Ktjd|jdkr|jn6|js|jjtkr|Ed{V|xjdz c_dS)NT)checkrr) rrDr(r'enter_pipeline_modertransaction_statusACTIVE _sync_genr@s r* _enter_genzBasePipeline._enter_genDs!---- :?? K + + - - - -   (4;#AV#K#K~~'' ' ' ' ' ' ' ' a r,excBaseException | NonecD|xjdzc_|jdkr|jjtkro |jdS#t j$r?}|rtd||n| dYd}~dSd}~wwxYwdSdS)Nrrzerror ignored exiting %r: %s) r(r'rABADexit_pipeline_modeeOperationalErrorloggerwarningwith_traceback)r)rNexc2s r*_exitzBasePipeline._exitRs a :??t{1S88 4 ..00000% 4 4 44NN#A4NNNN--d333ONNNNN  4 ?88sA B4BBc#K||Ed{V|dEd{VdS)NFflush _enqueue_sync_communicate_gen _fetch_genr@s r*rLzBasePipeline._sync_gen`sh ((*********???///////////r,c#K ||Ed{V|dEd{VdS#|dEd{VwxYw)z_ Exit current pipeline by sending a Sync and fetch back all remaining results. NTr[r]r@s r* _exit_genzBasePipeline._exit_genes 3    ,,.. . . . . . . .T22 2 2 2 2 2 2 2 2 2tT22 2 2 2 2 2 2 2 2s .AA.c#Kt|j|jEd{V}d}|D]R}|j} |||3#t j$r}||}Yd}~Kd}~wwxYw||dS)zqCommunicate with pipeline to send commands and possibly fetch results, which are then processed. N)rr'rr popleft_process_resultsrSError)r)fetched exceptionresultsqueuedrNs r*r_zBasePipeline._communicate_genos2$+t?QRRRRRRRR  $ $G&..00F $%%fg66667 $ $ $$ #I $  O ! sAA7)A22A7r\c#K|jsdS|r3|jt|jEd{Vd}|jrut |jEd{V}|snX|j} |||n #tj$r}||}Yd}~nd}~wwxYw|ju||dS)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_requestrrrdrerSrf)r)r\rhrirjrNs r*r`zBasePipeline._fetch_gens   F  ) K * * , , ,DK(( ( ( ( ( ( ( (  $!+DK!8!8888888G &..00F $%%fg66667 $ $ $$ #I $ $  O ! sBB5'B00B5rjrrilist[PGresult]cp|Z|\}|jtkr tj||jj|jt krtjddS|\}}|r(|\}}}|jj ||||| || |dS)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) rA FATAL_ERRORrSerror_from_resultr' _encodingPIPELINE_ABORTEDPipelineAbortedr& _preparedvalidate_check_results _set_results) r)rjriresultcursorprepinfokeyprepnames r*rezBasePipeline._process_resultss >IV} ++)&4;;PQQQQ"222'(:;;;32 & FH J"*T4 &//T4III  ! !' * * *    ( ( ( ( (r,c|j|jj|jddS)z#Enqueue a PQpipelineSync() command.N)rappendr' pipeline_syncr r@s r*r^zBasePipeline._enqueue_syncs; !!$+";<<<   &&&&&r,N)r!r"r#r$)r#r-)r#r;)r#rB)r#rF)rNrOr#r$)r\rBr#rF)rjrrirmr#r$r#r$)__name__r5r6__annotations__r+r:propertyrA classmethodrErMrYrLrbr_r`rer^r,r*rr+s7))))&&&&3333 >>>X>+++[+     4 4 4 40000 3333 @)))).''''''r,rcLeZdZUdZdZded<dfd Zdd Zdd ZddZxZ S)Pipelinez(Handler for connection in pipeline mode.rConnection[Any]r&r!r#r$cJt|dSr=superr+r)r!r4s r*r+zPipeline.__init__! r,c |jj5|j|ddddS#1swxYwYdS#tj$r}|dd}~wwxYw)zkSync the pipeline, send any pending command and receive and process all available results. Nr&lockwaitrLrS _NO_TRACEBACKrWr)exs r*syncz Pipeline.syncs * 2 2  0 0111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 * * *##D)) ) *s9 A-A AA  AA AA>$A99A>r c|jj5|j|dddn #1swxYwY|Sr=r&rrrMr@s r* __enter__zPipeline.__enter__s Z_ / / JOODOO-- . . . / / / / / / / / / / / / / / / s-AA  A exc_typetype[BaseException] | Noneexc_valrOexc_tbTracebackType | Nonec |jj5|j|dddn #1swxYwYnK#t$r>}|rt d||n|dYd}~nd}~wwxYw||dS#||wxYwNz error ignored terminating %r: %s r&rrrb ExceptionrUrVrWrYr)rrrrXs r*__exit__zPipeline.__exit__s  2 2  0 0111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 0 0 0 0A4NNNN))$///ONNNN 0 JJw     DJJw    sQ A-A AA  AA AB5 B4BB5BB55C )r!rr#r$rr#r rrrrOrrr#r$) rr5r6__doc__rr+rrr __classcell__r4s@r*rrs22J****         r,rcLeZdZUdZdZded<dfd Zdd Zdd ZddZxZ S) AsyncPipelinez.Handler for async connection in pipeline mode.rAsyncConnection[Any]r&r!r#r$cJt|dSr=rrs r*r+zAsyncPipeline.__init__rr,c6K |jj4d{V|j|d{Vdddd{VdS#1d{VswxYwYdS#tj$r}|dd}~wwxYwr=rrs r*rzAsyncPipeline.syncs8 *z 8 8 8 8 8 8 8 8joodnn&6&6777777777 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 * * *##D)) ) *s:A/3A A/ A&&A/)A&*A//B>BBr cK|jj4d{V|j|d{Vdddd{Vn#1d{VswxYwY|Sr=rr@s r* __aenter__zAsyncPipeline.__aenter__s:? 5 5 5 5 5 5 5 5*//$//"3"344 4 4 4 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 s3A A$'A$rrrrOrrcK |jj4d{V|j|d{Vdddd{Vn#1d{VswxYwYnK#t$r>}|rt d||n|dYd}~nd}~wwxYw||dS#||wxYwrrrs r* __aexit__zAsyncPipeline.__aexit__s z 8 8 8 8 8 8 8 8joodnn&6&6777777777 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 0 0 0 0A4NNNN))$///ONNNN 0 JJw     DJJw    sRA-3A A- A%%A-(A%)A-,C- B574B0+C0B55CC&)r!rr#r$rrr) rr5r6rrr+rrrrrs@r*rrs88J****         r,r)7r __future__rloggingtypesrtypingrrrrSr abcr r _compatr r rpq.miscr generatorsrrr _capabilitiesrpq.abcr _preparingrr connectionr _cursor_baser_connection_baserconnection_asyncrrr ExecStatusrprs ConnStatusrQTransactionStatusrK getLoggerrUrrrrr,r*rs^ #"""""%%%%%%%%''''''''++++++++++''''''>>>>>>>>>>''''''2      ((((((((&&&&&&((((((000000111111Lm' =1m  $  9 % %N'N'N'N'N'N'N'N'b( ( ( ( ( |( ( ( V% % % % % L% % % % % r,