Ë
    êëj
  ã                  óÀ  — U d Z ddlmZ ddlZddlmZmZ ddlmZ	 ddlm
Z
 ddlmZmZ dd	lmZmZ dd
lmZ ddlmZmZmZ ddlmZ erddlmZ ddlmZmZ ddlmZ ddl m!Z! dZ"de#d<   e
jH                  jJ                  Z%e
jH                  jL                  Z&e
jN                  jP                  Z(e
jR                  jT                  Z* ejV                  d«      Z, G d„ d«      Z-y)z
commands pipeline management
é    )ÚannotationsN)ÚTYPE_CHECKINGÚAnyé   )Úerrors)Úpq)ÚPipelineCommandÚPQGen)ÚDequeÚ	TypeAlias)Úconnection_summary)Ú
fetch_manyÚpipeline_communicateÚsend)Úcapabilities)ÚPGresult)ÚKeyÚPrepare)Ú
BaseCursor)ÚBaseConnectionzEtuple[BaseCursor[Any, Any], tuple[Key, Prepare, bytes] | None] | Noner   ÚPendingResultÚpsycopgc                  ó–   — e Zd ZU ded<   ded<   dd„Zdd„Zedd„«       Zedd„«       Z	dd	„Z
dd
„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zy)ÚBasePipelinezDeque[PipelineCommand]Úcommand_queuezDeque[PendingResult]Úresult_queuec                óš   — || _         |j                  | _        t        t           «       | _        t        t
           «       | _        d| _        y )Nr   )Ú_connÚpgconnr   r	   r   r   r   Úlevel)ÚselfÚconns     ú]/var/www/html/kelly/kelly-backend/venv/lib/python3.12/site-packages/psycopg/_pipeline_base.pyÚ__init__zBasePipeline.__init__,   s;   € ØˆŒ
Ø—k‘kˆŒÜ"¤?Ñ3Ó5ˆÔÜ!¤-Ñ0Ó2ˆÔØˆ�
ó    c                óÊ   — | j                   j                  › d| j                   j                  › �}t        | j                  j
                  «      }d|› d|› dt        | «      d›d�S )NÚ.Ú<Ú z at 0xÚxÚ>)Ú	__class__Ú
__module__Ú__qualname__r   r   r   Úid)r!   ÚclsÚinfos      r#   Ú__repr__zBasePipeline.__repr__3   s]   € Ø—‘×*Ñ*Ð+¨1¨T¯^©^×-HÑ-HÐ,IÐJˆÜ! $§*¡*×"3Ñ"3Ó4ˆØ�3�%�q˜˜˜f¤R¨£X¨a L°Ð2Ð2r%   c                óT   — t        j                  | j                  j                  «      S )N)r   ÚPipelineStatusr   Úpipeline_status©r!   s    r#   ÚstatuszBasePipeline.status8   s   € ä× Ñ  §¡×!<Ñ!<Ó=Ð=r%   c                ó*   — t        j                  «       S )zCReturn `!True` if the psycopg libpq wrapper supports pipeline mode.)r   Úhas_pipeline)r0   s    r#   Úis_supportedzBasePipeline.is_supported<   s   € ô ×(Ñ(Ó*Ð*r%   c              #  ó<  K  — t        j                  d¬«       | j                  dk(  r| j                  j	                  «        nA| j
                  s| j                  j                  t        k(  r| j                  «       E d {  –—†  | xj                  dz  c_        y 7 Œ­w)NT)Úcheckr   r   )	r   r9   r    r   Úenter_pipeline_moder   Útransaction_statusÚACTIVEÚ	_sync_genr6   s    r#   Ú
_enter_genzBasePipeline._enter_genA   sm   è ø€ Ü×!Ñ!¨Õ-Ø�:‰:˜Š?Ø�K‰K×+Ñ+Õ-Ø×Ò 4§;¡;×#AÑ#AÄVÒ#Kð —~‘~Ó'×'Ð'Ø�
Š
�a‰Ž
ð (ús   ‚A=BÁ?BÂ Bc                óV  — | xj                   dz  c_         | j                   dk(  r:| j                  j                  t        k7  r	 | j                  j	                  «        y y y # t
        j                  $ r5}|rt        j                  d| |«       n|j                  d «      ‚Y d }~y d }~ww xY w)Nr   r   zerror ignored exiting %r: %s)
r    r   r7   ÚBADÚexit_pipeline_modeÚeÚOperationalErrorÚloggerÚwarningÚwith_traceback)r!   ÚexcÚexc2s      r#   Ú_exitzBasePipeline._exitO   sŒ   € Ø�
Š
�a‰�
Ø�:‰:˜Š?˜tŸ{™{×1Ñ1´SÒ8ð	4Ø—‘×.Ñ.Õ0ð  9ˆ?øô ×%Ñ%ò 4ñ Ü—N‘NÐ#AÀ4ÈÕNà×-Ñ-¨dÓ3Ð3ô Oûð4ús   ÁA  Á B(Á3+B#Â#B(c              #  ó˜   K  — | j                  «        | j                  «       E d {  –—†  | j                  d¬«      E d {  –—†  y 7 Œ7 Œ­w)NF©Úflush©Ú_enqueue_syncÚ_communicate_genÚ
_fetch_genr6   s    r#   r@   zBasePipeline._sync_gen]   sA   è ø€ Ø×ÑÔØ×(Ñ(Ó*×*Ð*Ø—?‘?¨�?Ó/×/Ñ/ð 	+øØ/ús!   ‚$A
¦A§A
Á AÁA
ÁA
c              #  óÚ   K  — 	 | j                  «        | j                  «       E d{  –—†  | j                  d¬«      E d{  –—†  y7 Œ7 Œ# | j                  d¬«      E d{  –—†7   w xY w­w)z_
        Exit current pipeline by sending a Sync and fetch back all remaining results.
        NTrN   rP   r6   s    r#   Ú	_exit_genzBasePipeline._exit_genb   sZ   è ø€ ð	3Ø×ÑÔ Ø×,Ñ,Ó.×.Ð.à—‘¨T�Ó2×2Ñ2ð /øà2ù�t—‘¨T�Ó2×2Ò2üsI   ‚A+„#A §A¨A ¬A+ÁA	ÁA+ÁA Á	A+ÁA(Á!A$Á"A(Á(A+c              #  ó  K  — t        | j                  | j                  «      E d{  –—† }d}|D ]/  }| j                  j	                  «       }	 | j                  ||«       Œ1 |�|‚y7 Œ?# t        j                  $ r}|€|}Y d}~ŒUd}~ww xY w­w)zqCommunicate with pipeline to send commands and possibly fetch
        results, which are then processed.
        N)r   r   r   r   ÚpopleftÚ_process_resultsrE   ÚError)r!   ÚfetchedÚ	exceptionÚresultsÚqueuedrJ   s         r#   rR   zBasePipeline._communicate_genl   s’   è ø€ ô 2°$·+±+¸t×?QÑ?QÓR×RˆØˆ	ÛˆGØ×&Ñ&×.Ñ.Ó0ˆFð$Ø×%Ñ% f¨gÕ6ð ð Ð ØˆOð !ð Sùô —7‘7ò $ØÐ$Ø #�Iÿøð$üs9   ‚$B¦A&§%BÁA(ÁBÁ(B	Á;BÁ?BÂB	Â	Bc             #  óÀ  K  — | j                   sy|r7| j                  j                  «        t        | j                  «      E d{  –—†  d}| j                   rZt	        | j                  «      E d{  –—† x}sn:| j                   j                  «       }	 | j                  ||«       | j                   rŒZ|�|‚y7 Œq7 ŒH# t        j                  $ r}|€|}Y d}~Œ2d}~ww xY w­w)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_requestr   r   rW   rX   rE   rY   )r!   rO   r[   r\   r]   rJ   s         r#   rS   zBasePipeline._fetch_gen|   sÖ   è ø€ ð × Ò ØáØ�K‰K×*Ñ*Ô,Ü˜DŸK™KÓ(×(Ð(àˆ	Ø×ÒÜ+5°d·k±kÓ+B× BÐC�GÐCð Ø×&Ñ&×.Ñ.Ó0ˆFð$Ø×%Ñ% f¨gÔ6ð ×Óð Ð ØˆOð !ð )øð !Cùô —7‘7ò $ØÐ$Ø #�Iÿøð$üsT   ‚ACÁB6Á*CÁ/B8Á0!CÂB: Â$CÂ1CÂ8CÂ:CÃCÃCÃCÃCc                óŒ  — |€k|\  }|j                   t        k(  r+t        j                  || j                  j
                  ¬«      ‚|j                   t        k(  rt        j                  d«      ‚y|\  }}|r.|\  }}}|j                  j                  j                  ||||«       |j                  |«       |j                  |«       y)ar  Process 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)Úencodingzpipeline aborted)r7   ÚFATAL_ERRORrE   Úerror_from_resultr   Ú	_encodingÚPIPELINE_ABORTEDÚPipelineAbortedr   Ú	_preparedÚvalidateÚ_check_resultsÚ_set_results)	r!   r]   r\   ÚresultÚcursorÚprepinfoÚkeyÚprepÚnames	            r#   rX   zBasePipeline._process_results›   s±   € ð ˆ>Ø‰IˆVØ�}‰}¤Ò+Ü×)Ñ)¨&¸4¿;¹;×;PÑ;PÔQÐQØ—‘Ô"2Ò2Ü×'Ñ'Ð(:Ó;Ð;ð 3ð  &ÑˆF�HÙØ"*‘��T˜4à—‘×&Ñ&×/Ñ/°°T¸4ÀÔIØ×!Ñ! 'Ô*Ø×Ñ Õ(r%   c                ó˜   — | j                   j                  | j                  j                  «       | j                  j                  d«       y)z#Enqueue a PQpipelineSync() command.N)r   Úappendr   Úpipeline_syncr   r6   s    r#   rQ   zBasePipeline._enqueue_sync²   s4   € à×Ñ×!Ñ! $§+¡+×";Ñ";Ô<Ø×Ñ× Ñ  Õ&r%   N)r"   zBaseConnection[Any]ÚreturnÚNone)rt   Ústr)rt   zpq.PipelineStatus)rt   Úbool)rt   úPQGen[None])rJ   zBaseException | Nonert   ru   )rO   rw   rt   rx   )r]   r   r\   zlist[PGresult]rt   ru   )rt   ru   )Ú__name__r-   r.   Ú__annotations__r$   r2   Úpropertyr7   Úclassmethodr:   rA   rL   r@   rU   rR   rS   rX   rQ   © r%   r#   r   r   (   sh   … Ø)Ó)Ø&Ó&óó3ð
 ò>ó ð>ð ò+ó ð+óó4ó0ó
3óó ó>)ô.'r%   r   ).Ú__doc__Ú
__future__r   ÚloggingÚtypingr   r   Ú r   rE   r   Úabcr	   r
   Ú_compatr   r   Úpq.miscr   Ú
generatorsr   r   r   Ú_capabilitiesr   Úpq.abcr   Ú
_preparingr   r   Ú_cursor_baser   Ú_connection_baser   r   rz   Ú
ExecStatusrb   re   Ú
ConnStatusrC   ÚTransactionStatusr?   Ú	getLoggerrG   r   r}   r%   r#   Ú<module>r�      s­   ðòõ #ã ß %å Ý ß 'ß %Ý 'ß >Ñ >Ý 'áÝ ß(Ý(Ý0ð Lð ˆyó ð �m‰m×'Ñ'€Ø—=‘=×1Ñ1Ð Ø‡m�m×Ñ€à	×	Ñ	×	$Ñ	$€à	ˆ×	Ñ	˜9Ó	%€÷M'ò M'r%   