*Jf\( UdZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZdd lmZmZdd lmZmZdd lmZdd lmZdd lmZmZddlm Z m!Z!m"Z"e rddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ede dee eee,fffZ-e e.d<ej/j0Z0ej/j1Z1ej2j3Z3ej4j5Z5ej6dZ7GddZ8Gdde8Z9Gdde8Z:dS)z commands pipeline management N) TracebackType)AnyListOptionalUnionTupleType TYPE_CHECKING) TypeAlias)pq)errors)PipelineCommandPQGen)DequeSelf)connection_summary)pgconn_encoding)KeyPrepare)pipeline_communicate fetch_manysend)PGresult) BaseCursor)BaseConnection Connection)AsyncConnectionzBaseCursor[Any, Any] PendingResultpsycopgceZdZUeeed<eeed<dZee ed<ddZ de fd Z e dejfd Zede fd Zede fd Zdedfd ZdeeddfdZdedfdZdedfdZdedfdZde dedfdZdededddfdZddZdS) BasePipeline command_queue result_queueN _is_supportedconnBaseConnection[Any]returnc||_|j|_tt|_tt |_d|_dS)Nr)_connpgconnrrr#rr$level)selfr&s T/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/_pipeline.py__init__zBasePipeline.__init__.sC k "?355!-022 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__5s[*JJT^-HJJ!$*"344232222RXX22222r0c>tj|jjSN)r PipelineStatusr+pipeline_statusr-s r.statuszBasePipeline.status:s !<===r0cptj| t_tjS)zCReturn `!True` if the psycopg libpq wrapper supports pipeline mode.)r"r%_not_supported_reasonr;s r. is_supportedzBasePipeline.is_supported>s-  % --0-F-F-H-H)HL &))r0ctjdkrdtjdStjdkrdtjdSdS)zReturn the reason why the pipeline mode is not supported. Return an empty string if pipeline mode is supported. i"zlibpq too old z+; v14 or greater required for pipeline modez libpq too old: module built for )r version__build_version__rFs r.rEz"BasePipeline._not_supported_reasonEsk :<<& ====   & ( (=23G===  rr0c#ZK|s)tjd||jdkr|jn6|js|jjtkr| Ed{V|xjdz c_dS)Nzpipeline mode not supported: rr ) rGeNotSupportedErrorrEr,r+enter_pipeline_moder#transaction_statusACTIVE _sync_genrBs r. _enter_genzBasePipeline._enter_gen[s  "" %N0J0J0L0LNN  :?? K + + - - - -   (4;#AV#K#K~~'' ' ' ' ' ' ' ' a r0exccD|xjdzc_|jdkr|jjtkro |jdS#t j$r?}|rtd||n| dYd}~dSd}~wwxYwdSdS)Nr rzerror ignored exiting %r: %s) r,r+rCBADexit_pipeline_moderMOperationalErrorloggerwarningwith_traceback)r-rTexc2s r._exitzBasePipeline._exitls a :??t{1S88 4 ..00000% 4 4 44NN#A4NNNN--d333ONNNNN  4 ?88sA B4BBc#K||Ed{V|dEd{VdS)NFflush _enqueue_sync_communicate_gen _fetch_genrBs r.rRzBasePipeline._sync_genzsh ((*********???///////////r0c#K ||Ed{V|dEd{VdS#|dEd{VwxYw)z_ Exit current pipeline by sending a Sync and fetch back all remaining results. NTr_rarBs r. _exit_genzBasePipeline._exit_gens 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+r#r$popleft_process_resultsrMError)r-fetched exceptionresultsqueuedrTs r.rczBasePipeline._communicate_gens2$+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_requestrrrhrirMrj)r-r`rlrmrnrTs r.rdzBasePipeline._fetch_gens   F  ) K * * , , ,DK(( ( ( ( ( ( ( (  $!+DK!8!8888888G &..00F $%%fg66667 $ $ $$ #I $ $  O ! sBB5'B00B5rnrmrcV|b|\}|jtkr(tj|t |j|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) rC FATAL_ERRORrMerror_from_resultrr+PIPELINE_ABORTEDPipelineAbortedr* _preparedvalidate_set_results_from_pipeline) r-rnrmresultcursorprepinfokeyprepnames r.rizBasePipeline._process_resultss >IV} ++)&?4;;W;WXXXX"222'(:;;;32 & FH J"*T4 &//T4III  - -g 6 6 6 6 6r0c|j|jj|jddS)z#Enqueue a PQpipelineSync() command.N)r#appendr+ pipeline_syncr$rBs r.rbzBasePipeline._enqueue_syncs; !!$+";<<<   &&&&&r0)r&r'r(Nr(N)__name__r8r9rr__annotations__rr%rboolr/strr=propertyr r@rC classmethodrGrErrS BaseExceptionr]rRrfrcrdrrirbr0r.r"r")s)))) &&&&$(M8D>(((3#3333 >)>>>X>*T***[* c[*E$K" 4-0 4T 4 4 4 405;0000 35;3333%+ 4E$K@7#7.2:.>7 77770''''''r0r"ceZdZUdZdZded<dfd Zdd Zdefd Zd e e e d e e d e e ddfdZ xZS)Pipelinez(Handler for connection in pipeline mode.r Connection[Any]r*r&r(NcJt|dSr?superr/r-r&r7s r.r/zPipeline.__init__! r0c |jj5|j|ddddS#1swxYwYdS#tj$r}|dd}~wwxYw)zkSync the pipeline, send any pending command and receive and process all available results. Nr*lockwaitrRrM _NO_TRACEBACKr[r-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>c|jj5|j|dddn #1swxYwY|Sr?r*rrrSrBs r. __enter__zPipeline.__enter__s Z_ / / JOODOO-- . . . / / / / / / / / / / / / / / / s-AA  A exc_typeexc_valexc_tbc |jj5|j|dddn #1swxYwYnK#t$r>}|rt d||n|dYd}~nd}~wwxYw||dS#||wxYwNz error ignored terminating %r: %s r*rrrf ExceptionrYrZr[r]r-rrrr\s 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(Nr)rr8r9__doc__rr/rrrrr rrr __classcell__r7s@r.rrs22J ****4  4 ./ -( '           r0rceZdZUdZdZded<dfd Zdd Zdefd Zd e e e d e e d e e ddfdZ xZS) AsyncPipelinez.Handler for async connection in pipeline mode.r AsyncConnection[Any]r*r&r(NcJt|dSr?rrs r.r/zAsyncPipeline.__init__rr0c6K |jj4d{V|j|d{Vdddd{VdS#1d{VswxYwYdS#tj$r}|dd}~wwxYwr?rrs r.rzAsyncPipeline.sync s8 *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>BBcK|jj4d{V|j|d{Vdddd{Vn#1d{VswxYwY|Sr?rrBs 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$rrrcK |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(Nr)rr8r9rrr/rrrrr rrrrrs@r.rrs88J !!!!****$  4 ./ -( '           r0r);rloggingtypesrtypingrrrrrr r typing_extensionsr rIr rrMabcrr_compatrrpq.miscr _encodingsr _preparingrr generatorsrrrpq.abcrr{r connectionrrconnection_asyncrbytesrr ExecStatusrsru ConnStatusrVTransactionStatusrQ getLoggerrYr"rrrr0r.rs IIIIIIIIIIIIIIIIII'''''''''''''' ''''''''''''$$$$$$$$>>>>>>>>>>2      """"""66666666111111!%&sGU7J1K(LL MM ym' =1m  $  9 % %k'k'k'k'k'k'k'k'\( ( ( ( ( |( ( ( V% % % % % L% % % % % r0