Ytj tUdZddlmZddlZddlmZmZmZddlm Z ddl m Z ddl m Z dd lmZmZdd lmZdd lmZmZmZdd lmZerdd lmZddlmZmZddlmZddl m!Z!dZ"de#d<e j$j%Z%e j$j&Z&e j'j(Z(e j)j*Z*ej+dZ,GddZ-dS)z commands pipeline management ) annotationsN) TYPE_CHECKINGAny TypeAlias)deque)errors)pq)PipelineCommandPQGen)connection_summary) fetch_manypipeline_communicatesend) capabilities)PGresult)KeyPrepare) BaseCursor)BaseConnectionzEtuple[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)_connpgconnrr rrrlevel)selfrs T/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg/_pipeline_base.py__init__zBasePipeline.__init__,sC k "?355!-022 strc|jjd|jj}t|jj}d|d|dt |ddS)N.< z at 0xx>) __class__ __module__ __qualname__r r"r#id)r%clsinfos r&__repr__zBasePipeline.__repr__3s[*JJT^-HJJ!$*"344232222RXX22222r(pq.PipelineStatusc>tj|jjS)N)r PipelineStatusr#pipeline_statusr%s r&statuszBasePipeline.status8s !<===r(boolc(tjS)zCReturn `!True` if the psycopg libpq wrapper supports pipeline mode.)r has_pipeline)r4s r& is_supportedzBasePipeline.is_supported<s(***r( PQGen[None]c# Ktjd|jdkr|jn6|js|jjtkr|Ed{V|xjdz c_dS)NT)checkrr) rr?r$r#enter_pipeline_modertransaction_statusACTIVE _sync_genr;s r& _enter_genzBasePipeline._enter_genAs!---- :?? 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#r<BADexit_pipeline_modeeOperationalErrorloggerwarningwith_traceback)r%rIexc2s r&_exitzBasePipeline._exitOs 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&rGzBasePipeline._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. NTrVrXr;s r& _exit_genzBasePipeline._exit_genbs 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#rrpopleft_process_resultsrNError)r%fetched exceptionresultsqueuedrIs r&rZzBasePipeline._communicate_genls2$+t?QRRRRRRRR  $ $G&..00F $%%fg66667 $ $ $$ #I $  O ! sAA7)A22A7rWc#K|jsdS|r3|jt|jEd{Vd}|jrut |jEd{Vx}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) rr#send_flush_requestrrr_r`rNra)r%rWrcrdrerIs r&r[zBasePipeline._fetch_gen|s   F  ) K * * , , ,DK(( ( ( ( ( ( ( (  $+5dk+B+B B B B B B BCG &..00F $%%fg66667 $ $ $$ #I $ $  O ! sBB5'B00B5rerrdlist[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) r< FATAL_ERRORrNerror_from_resultr# _encodingPIPELINE_ABORTEDPipelineAbortedr" _preparedvalidate_check_results _set_results) r%rerdresultcursorprepinfokeyprepnames r&r`zBasePipeline._process_resultss >IV} ++)&4;;PQQQQ"222'(:;;;32 & FH J"*T4 &//T4III  ! !' * * *    ( ( ( ( (r(c|j|jj|jddS)z#Enqueue a PQpipelineSync() command.N)rappendr# pipeline_syncrr;s r&rYzBasePipeline._enqueue_syncs; !!$+";<<<   &&&&&r(N)rrrr )rr))rr7)rr=)rrA)rIrJrr )rWr=rrA)rerrdrhrr )rr )__name__r1r2__annotations__r'r6propertyr< classmethodr@rHrTrGr]rZr[r`rYr(r&rr(s6))))&&&&3333 >>>X>+++[+     4 4 4 40000 3333 >)))).''''''r(r).__doc__ __future__rloggingtypingrrr collectionsrr rNr abcr r pq.miscr generatorsrrr _capabilitiesrpq.abcr _preparingrr _cursor_baser_connection_baserrr~ ExecStatusrkrn ConnStatusrLTransactionStatusrF getLoggerrPrrr(r&rs #"""""0000000000''''''''''''''>>>>>>>>>>''''''1      ((((((((((((((000000Lm' =1m  $  9 % %M'M'M'M'M'M'M'M'M'M'r(