
    ,i                        U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 d dl
mZ d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZm Z  ddl!m"Z" e"rd dl#Z#d d	l#m$Z$m%Z%m&Z& ndZ#dZ%dZ$dZ&dd
l'm(Z( ddl)m*Z*m+Z+ ddl!m,Z,m-Z- ej\                  dk\  rd dl m/Z0 nd dl0m/Z0 d dl1m2Z2 d dl3m4Z4 d dl5m6Z6 d dl7m8Z8m9Z9 d dl:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZA d dlBmCZC d dlDmEZEmFZFmGZG ddlHmIZImJZJmKZKmLZLmMZM dZNdZOdZPdZQdZR G d dej                        ZTeTj                  ZVeeeLeMeKf      eWd<   eEreKZXneLZX G d d e      ZY G d! d"e      ZZeeYeZf   Z[ G d# d$      Z\ G d% d&e\      Z] G d' d(e]      Z^ G d) d*      Z_ G d+ d,e\      Z`d-Zad.eeb   fd/Zcd0 Zd eeeefefececeeeeecededefd1      Zgeehed2eif   f   eWd3<    G d4 d5ed67      Zjd8ehd.ejfd9Zk ed:d;<      Zl G d= d;      Zm G d> d?em      Zny)@    N)abstractmethod)chain)MappingProxyType)AnyCallableIterableListMappingOptionalProtocolSetTupleType	TypedDictTypeVarUnion)ParseResultparse_qsunquoteurlparse   )SSL_AVAILABLE)
SSLContext
TLSVersionVerifyFlags)TokenInterface)!AsyncAfterConnectionReleasedEventEventDispatcher)deprecated_argsformat_error_message)      r!   )timeout)Retry)	NoBackoff)DEFAULT_RESP_VERSION)CredentialProvider"UsernamePasswordCredentialProvider)AuthenticationError$AuthenticationWrongNumberOfArgsErrorConnectionError	DataError
RedisErrorResponseErrorTimeoutError)
EncodableT)HIREDIS_AVAILABLEget_lib_versionstr_if_bytes)
BaseParserEncoder_AsyncHiredisParser_AsyncRESP2Parser_AsyncRESP3Parser   *   $s   
   
    c                       e Zd Z e       Zy)	_SentinelN)__name__
__module____qualname__objectsentinel r<   ]/var/www/html/langgraph-service/venv/lib/python3.12/site-packages/redis/asyncio/connection.pyr>   r>   R   s	    xHr<   r>   DefaultParserc                       e Zd ZddZy)ConnectCallbackProtocolc                      y NrD   self
connections     rE   __call__z ConnectCallbackProtocol.__call__a   s    r<   NrM   AbstractConnectionr?   r@   rA   rN   rD   r<   rE   rH   rH   `   s    =r<   rH   c                       e Zd ZddZy)AsyncConnectCallbackProtocolc                    K   y wrJ   rD   rK   s     rE   rN   z%AsyncConnectCallbackProtocol.__call__e   s        NrO   rQ   rD   r<   rE   rS   rS   d   s    Cr<   rS   c            .          e Zd ZdZdZdddddedddedddd	 e       dddedd
ddde	e
ef   dee
   dee   dee   dede	eef   de
de
dedee   dededee
   dee
   dee
   dee
   dee   dee   dee   dee   d ee   d!ee   f,d"Zefd#efd$Zd% Zd& Zed'        Zed(        Z d) Z!d* Z"dee   d+dfd,Z#d- Z$	 dUd/ed0efd1Z%d2 Z&d3 Z'ed4        Z(ed+e
fd5       Z)d6e*d+e
fd7Z+d8 Z,dVd9Z-dWd/ed+dfd:Z.dXd;ed+dfd<Z/d= Z0d> Z1d? Z2d@e3e4   d+dfdAZ5	 dWd@e	e4e
e3e4   f   d/ed+dfdBZ6dCedDed+dfdEZ7dF Z8	 	 dYd.ddGdHedIee   dJedKee   fdLZ9dCe:d+e;e4   fdMZ<dNe3e3e:      d+e;e4   fdOZ=dP Z>dQ Z?dRe@fdSZAdT ZBy)ZrP   z0Manages communication to and from a Redis server)dbusernameclient_namelib_namelib_versioncredential_providerpasswordsocket_timeoutsocket_connect_timeoutredis_connect_funcretry_on_timeoutretry_on_errorhealth_check_intervalnext_health_checklast_active_atencoderssl_contextprotocol_reader_writer_parser_connect_callbacks_buffer_cutoff_lock_socket_read_size__dict__r   NFutf-8stricti   zredis-pyr   )rW   r]   r^   r_   ra   rb   encodingencoding_errorsdecode_responsesparser_classsocket_read_sizerc   rY   rZ   r[   rX   retryr`   encoder_classr\   rh   event_dispatcherrW   r]   r^   r_   ra   rb   rs   rt   ru   rv   rw   rc   rY   rZ   r[   rX   rx   r`   ry   r\   rh   rz   c                D   |s|r|t        d      |t               | _        n|| _        || _        || _        || _        || _        || _        || _        || _	        || _
        ||}|| _        || _        |t        u rg }|rS|j                  t               |j                  t         j"                         |j                  t$        j                         || _        |s|rR|st)        t+               d      | _        nt/        j0                  |      | _        | j,                  j3                  |       nt)        t+               d      | _        || _        d| _         ||||	      | _        || _        d | _        d | _        || _         | jC                  |
       g | _"        d| _#        d | _$        d| _%        	 tM        |      }|dk  s|d	kD  rtU        d
      || _+        y # tN        $ r	 tP        }Y .tR        $ r tU        d      w xY w# dk  s|d	kD  rtU        d
      || _+        w xY w)Nz'username' and 'password' cannot be passed along with 'credential_provider'. Please provide only one of the following arguments: 
1. 'password' and (optional) 'username'
2. 'credential_provider'   r   ip  Fzprotocol must be an integerr   r!   zprotocol must be either 2 or 3),r,   r   _event_dispatcherrW   rY   rZ   r[   r\   r]   rX   r^   r_   ra   SENTINELappendr/   socketr#   asynciorb   r$   r%   rx   copydeepcopyupdate_supported_errorsrc   rd   rf   r`   ri   rj   ro   
set_parserrl   rm   _re_auth_token_should_reconnectint	TypeErrorr&   
ValueErrorr+   rh   )rL   rW   r]   r^   r_   ra   rb   rs   rt   ru   rv   rw   rc   rY   rZ   r[   rX   rx   r`   ry   r\   rh   rz   ps                           rE   __init__zAbstractConnection.__init__   s   4 &9&E+  #%4%6D"%5D"& &#6   ,!)%3"&<# 0X%N!!,/!!&..1!!'"6"67,N"9;2
 "]]51
JJ..~>y{A.DJ%:"(*$X@PQ"47;7;!1%NP"8<!&		%HA 1uA%&FGG$DM  	%$A 	A!"?@@	A 1uA%&FGG$DMs$   1G G>(H *G>>H H	_warningsc                     t        | dd       rA|j                  d| t        |        	 t        j                          | j                          y y # t        $ r Y y w xY w)Nrj   zunclosed Connection )source)getattrwarnResourceWarningr   get_running_loop_closeRuntimeError)rL   r   s     rE   __del__zAbstractConnection.__del__   sb     4D)NN&th/  ((* *   s   $A 	AAc                 n    | j                   r)| j                   j                          dx| _         | _        yy)zR
        Internal method to silently close the connection without waiting
        N)rj   closeri   rL   s    rE   r   zAbstractConnection._close   s.     <<LL *..DL4< r<   c                     dj                  d | j                         D              }d| j                  j                   d| j                  j                   d| dS )N,c              3   0   K   | ]  \  }}| d |   yw)=NrD   ).0kvs      rE   	<genexpr>z.AbstractConnection.__repr__.<locals>.<genexpr>   s     HTQ1QCjHs   <.(z)>)joinrepr_pieces	__class__r@   r?   )rL   	repr_argss     rE   __repr__zAbstractConnection.__repr__   sQ    HHHT5E5E5GHI	4>>,,-Qt~~/F/F.GqSUVVr<   c                      y rJ   rD   r   s    rE   r   zAbstractConnection.repr_pieces      r<   c                 >    | j                   d uxr | j                  d uS rJ   )ri   rj   r   s    rE   is_connectedzAbstractConnection.is_connected  s    ||4'DDLL,DDr<   c                     t        j                  |      }|| j                  vr| j                  j                  |       yy)a^  
        Register a callback to be called when the connection is established either
        initially or reconnected.  This allows listeners to issue commands that
        are ephemeral to the connection, for example pub/sub subscription or
        key tracking.  The callback must be a _method_ and will be kept as
        a weak reference.
        N)weakref
WeakMethodrl   r   )rL   callbackwms      rE   register_connect_callbackz,AbstractConnection.register_connect_callback	  s;     )T,,,##**2. -r<   c                     	 | j                   j                  t        j                  |             y# t        $ r Y yw xY w)z
        De-register a previously registered callback.  It will no-longer receive
        notifications on connection events.  Calling this is not required when the
        listener goes away, since the callbacks are kept as weak methods.
        N)rl   remover   r   r   )rL   r   s     rE   deregister_connect_callbackz.AbstractConnection.deregister_connect_callback  s8    	##**7+=+=h+GH 		s   .1 	==returnc                 4     || j                         | _        y)z
        Creates a new instance of parser_class with socket size:
        _socket_read_size and assigns it to the parser for the connection
        :param parser_class: The required parser class
        )rw   N)ro   rk   )rL   rv   s     rE   r   zAbstractConnection.set_parser   s     $T5K5KLr<   c                 D   K   | j                  d       d{    y7 w)z5Connects to the Redis server if not already connectedTcheck_healthN)connect_check_healthr   s    rE   connectzAbstractConnection.connect(  s     ''T':::     Tr   retry_socket_connectc                    K    j                   ry 	 |r+ j                  j                   fd fd       d {    n j                          d {    	  j                  s j                  |       d {    nIt	        j                   j                        r j                          d {   n j                           j$                  D cg c]  } |       s| c} _         j$                  D ]4  } |       } |       }|st'        j(                  |      s+| d {    6 y 7 7 # t        j
                  $ r  t        j                  t        j                  f$ r t        d      t        $ r}t         j                  |            d }~wt        $ r}t        |      |d }~ww xY w7 A7 # t         $ r  j#                          d {  7    w xY wc c}w 7 w)Nc                  $     j                         S rJ   )_connectr   s   rE   <lambda>z9AbstractConnection.connect_check_health.<locals>.<lambda>4  s    DMMO r<   c                 $    j                         S rJ   
disconnecterrorrL   s    rE   r   z9AbstractConnection.connect_check_health.<locals>.<lambda>4  s    4??;L r<   zTimeout connecting to serverr   )r   rx   call_with_retryr   r   CancelledErrorr   r#   r/   OSErrorr+   _error_message	Exceptionr`   on_connect_check_healthiscoroutinefunctionr-   r   rl   inspectisawaitable)rL   r   r   eexcrefr   tasks   `       rE   r   z'AbstractConnection.connect_check_health,  s     	0#jj00+-L   mmo%%	**222MMM
 2243J3JK 11$777006 372I2I"S3SU3"S** 	CuHD>D++D1

		E &%% 	 4 45 	?=>> 	:!$"5"5a"899 	0!#&C/	0 N 8  	//###	 #T
 s   G.'D7 D3D7 D5D7 !G  9F::8G  2F=3G  G.G'#G''(G.G.&G.+G,,G.3D7 5D7 7A	F7 FF7&F22F77G.:G  =G   G$GG$$	G.c                     d| _         y )NTr   r   s    rE   mark_for_reconnectz%AbstractConnection.mark_for_reconnect[  s
    !%r<   c                     | j                   S rJ   r   r   s    rE   should_reconnectz#AbstractConnection.should_reconnect^  s    %%%r<   c                    K   y wrJ   rD   r   s    rE   r   zAbstractConnection._connecta  s	     rU   c                      y rJ   rD   r   s    rE   _host_errorzAbstractConnection._host_errore  r   r<   	exceptionc                 6    t        | j                         |      S rJ   )r    r   )rL   r   s     rE   r   z!AbstractConnection._error_messagei  s    #D$4$4$6	BBr<   c                     | j                   S rJ   )rh   r   s    rE   get_protocolzAbstractConnection.get_protocoll  s    }}r<   c                 D   K   | j                  d       d{    y7 w)z=Initialize the connection, authenticate and select a databaseTr   N)r   r   s    rE   
on_connectzAbstractConnection.on_connecto  s     ***===r   c                 r	  K   | j                   j                  |        | j                   }d }| j                  s| j                  s| j                  rF| j                  xs  t        | j                  | j                        }|j                          d {   }|r&| j                  dvrt        | j                   t              rK| j                  t               |j                  | j                   _        | j                   j                  |        t        |      dk(  rd|d   g} | j                  d| j                  dg|ddi d {    | j                          d {   }|j!                  d	      t#        | j                        k7  r6|j!                  d
      t#        | j                        k7  rt%        d      |rP | j                  dg|ddi d {    	 | j                          d {   }t)        |      dk7  rt+        d      | j                  dvrt        | j                   t              rK| j                  t               |j                  | j                   _        | j                   j                  |        | j                  d| j                  |       d {    | j                          d {   }| j,                  rV| j                  dd| j,                  |       d {    t)        | j                          d {         dk7  rt%        d      | j.                  r(| j                  ddd| j.                  |       d {    | j0                  r(| j                  ddd| j0                  |       d {    | j2                  r&| j                  d| j2                  |       d {    d | j.                  | j0                  fD        D ]  }	 | j                          d {     | j2                  r0t)        | j                          d {         dk7  rt%        d      y y 7 7 7 7 g7 Q# t&        $ r= | j                  d|d   d       d {  7   | j                          d {  7  }Y w xY w7 7 7 7 7 I7 7 7 # t4        $ r Y w xY w7 w)N)r   2r|   defaultr   HELLOAUTHr   Fs   protoprotozInvalid RESP versionr}   r   OKzInvalid Username or PasswordCLIENTSETNAMEzError setting client nameSETINFOzLIB-NAMEzLIB-VERSELECTc              3   &   K   | ]	  }|s|  y wrJ   rD   )r   sents     rE   r   z=AbstractConnection.on_connect_check_health.<locals>.<genexpr>  s     M4$Ms   zInvalid Database)rk   r   r\   rX   r]   r(   get_credentials_asyncrh   
isinstancer7   r   r8   EXCEPTION_CLASSESlensend_commandread_responsegetr   r+   r*   r3   r)   rY   rZ   r[   rW   r.   )rL   r   parser	auth_argscred_providerresponseauth_response_s           rE   r   z*AbstractConnection.on_connect_check_healths  sd    %	##(( T5dmmT]]S  ,AACCI h6$,,(9: 12171I1I.''-9~"&	!5	 $$##1:IN   "//11H||H%T]]);;AT]]#A$ &&<== #$##FKYKUKKK;&*&8&8&: : M*d2)*HII ]](*$,,(9: 12171I1I.''-##GT]]#VVV!//11H ##  )	 $    $"4"4"6674?%&ABB ==##) $    ##  ) $    77##HdggL#QQQ NDMM43C3C#DM 	A((***	 77$"4"4"6674?%&899 @ { D 2 L !;7 ;
 ''	"E'RRR&*&8&8&: : :;$ W1 7
 R
 +   7sT  BR7P8B5R7P;R7P> A:R7QR7 Q 3Q4Q 8B-R7%R&R7=R>2R70R1R7RAR7R3R7R1R79R":&R7!R&4R$5R&9&R7R5 R7;R7>R7R7Q #R*Q-+RRR	R7RR7R7R7R7R7R7"R7$R&&	R2/R71R22R7nowaitc                 R  K   	 t        | j                        4 d{    | j                  j                          | j                  s	 ddd      d{    y	 | j
                  j                          |s"| j
                  j                          d{    d| _	        d| _        ddd      d{    y7 7 e7 '# t        $ r Y 0w xY w# d| _	        d| _        w xY w7 -# 1 d{  7  sw Y   yxY w# t        j                  $ r t        d| j                         dw xY ww)z!Disconnects from the Redis serverNz#Timed out closing connection after )async_timeoutr_   rk   on_disconnectr   rj   r   wait_closedr   ri   r   r/   )rL   r   s     rE   r   zAbstractConnection.disconnect  s    	$T%@%@A ( (**,((( ( (
(LL&&( ""ll66888 $(DL#'DL( ( ( ( 9  $(DL#'DL( ( ( ( ## 	5d6Q6Q5RS	s   D'C7 B8C7 (C"	C7 B:C7 D'9B>B<B>C"'C7 2C 3C7 7D'8C7 :C7 <B>>	C
C	C

CCC" C7 "C4(C+)C40C7 3D'4C7 7-D$$D'c                    K   | j                  dd       d{    t        | j                          d{         dk7  rt        d      y7 47 w)z Send PING, expect PONG in returnPINGFr   NPONGz#Bad response from PING health check)r   r3   r   r+   r   s    rE   
_send_pingzAbstractConnection._send_ping  sR     U;;;d00223v=!"GHH > 	<2s   AAAAAAc                 @   K   | j                          d{    y7 w)z Function to call when PING failsNr   rL   r   s     rE   _ping_failedzAbstractConnection._ping_failed       oo   c                    K   | j                   rit        j                         j                         | j                  kD  r9| j
                  j                  | j                  | j                         d{    yyy7 w)z3Check the health of the connection with a PING/PONGN)	rc   r   r   timerd   rx   r   r  r
  r   s    rE   r   zAbstractConnection.check_health   sf      &&((*//1D4J4JJ**,,T__d>O>OPPP K ' Qs   A/A;1A92A;commandc                    K   | j                   j                  |       | j                   j                          d {    y 7 wrJ   )rj   
writelinesdrain)rL   r  s     rE   _send_packed_commandz'AbstractConnection._send_packed_command  s.     (ll  """s   9AAAc                   K   | j                   s| j                  d       d {    |r| j                          d {    	 t        |t              r|j                         }t        |t              r|g}| j                  r8t        j                  | j                  |      | j                         d {    y | j                  j                  |       | j                  j                          d {    y 7 7 7 G7 # t        j                  $ r( | j                  d       d {  7   t        d      d t         $ r{}| j                  d       d {  7   t#        |j$                        dk(  rd|j$                  d   }}n|j$                  d   }|j$                  d   }t'        d	| d
| d      |d }~wt(        $ r | j                  d       d {  7    w xY ww)NFr   Tr   zTimeout writing to socketr|   UNKNOWNr   zError z while writing to socket. r   )r   r   r   r   strencodebytesr^   r   wait_forr  rj   r  r  r/   r   r   r   argsr+   BaseException)rL   r  r   r   err_noerrmsgs         rE   send_packed_commandz&AbstractConnection.send_packed_command  s       +++???##%%%	'3'!..*'5)")""&&--g68K8K   ''0ll((*** @%
 +## 	F///...:;E 		///...166{a!*AFF1I! :6(!D  	
 ///...	s   "G%C9G%C;G%A1D 5C=6D :G%;8D 3C?4D 8G%;G%=D ?D (G")D,*G"F9EA F99!G"GG""G%r  kwargsc                    K   | j                   | j                  | |j                  dd             d{    y7 w)z+Pack and send a command to the Redis serverr   Tr   N)r  pack_commandr   )rL   r  r   s      rE   r   zAbstractConnection.send_command5  s@     &&Dt$6::nd3S ' 
 	
 	
s   4><>c                   K   	 | j                   j                          d{   S 7 # t        $ rK}| j                  d       d{  7   | j	                         }t        d| d|j                         d}~ww xY ww)z8Poll the socket to see if there's data that can be read.NTr  Error while reading from z: )rk   can_read_destructiver   r   r   r+   r  )rL   r   
host_errors      rE   r%  z'AbstractConnection.can_read_destructive;  su     	V::<<<< 	V///...))+J!$=j\AFF8"TUU	Vs<   A?( &( A?( 	A<A7A	0A77A<<A?)disconnect_on_errorpush_requestdisable_decodingr#   r'  r(  c                  K   ||n| j                   }| j                         }	 |X| j                  dv rJt        |      4 d{    | j                  j                  ||       d{   }ddd      d{    n|It        |      4 d{    | j                  j                  |       d{   }ddd      d{    nX| j                  dv r&| j                  j                  ||       d{   }n$| j                  j                  |       d{   }| j                  r6t        j                         j                         | j                  z   }	|	| _        t#        t$              r|d|S 7 57 7 # 1 d{  7  sw Y   rxY w7 7 7 # 1 d{  7  sw Y   xY w7 7 # t        j                  $ r0 |Y y|r| j                  d       d{  7   t        d|       t        $ r=}|r| j                  d       d{  7   t        d| d	|j                         d}~wt        $ r |r| j                  d       d{  7    w xY ww)
z0Read the response from a previously sent commandN)3r!   )r)  r(  )r)  Tr  zTimeout reading from r$  z : )r^   r   rh   r  rk   r   r   r/   r   r   r+   r  r  rc   r   r  rd   r   r.   )
rL   r)  r#   r'  r(  read_timeoutr&  r   r   	next_times
             rE   r   z AbstractConnection.read_responseD  sj     #*"5w4;N;N%%'
%	'DMMX,E(6  %)\\%?%?)9 &@ &  H   )(6  %)\\%?%?)9 &@ &  H   (*!%!;!;%5L "< "  "&!;!;%5 "< " . %%002779D<V<VVI%.D"h.$W    
    
 ## 	E""ooTo222!6zlCDD 	W"ooTo222!$=j\QVVH"UVV 	 #ooTo222	s%  !I&F6 E8F6 !F(E;)F-F6 8E>9F6 FF6  F3F4F8F6 F3F6 7F28$F6 F4F6 !AI&8F6 ;F>F6 FF
FF6 FF6 F/#F&$F/+F6 4F6 6I#I&I#$G'%I# H8H H88#I#II##I&c           	         g }t        |d   t              rJ t        |d   t              r1t        |d   j	                         j                               |dd z   }n)d|d   v r"t        |d   j                               |dd z   }t        j                  t        t        t        |            j	                         t        f      }| j                  }t        | j                  j                  |      D ]  }t        |      }t        |      |kD  s||kD  st        |t              rat        j                  |t        t        |      j	                         t        f      }|j!                  |       |j!                  |       t        }t        j                  |t        t        |      j	                         t        |t        f      } |j!                  |       |S )z2Pack a series of arguments into the Redis protocolr   r|   N    )r   floatr  tupler  split	SYM_EMPTYr   SYM_STARr   SYM_CRLFrm   maprf   
memoryview
SYM_DOLLARr   )rL   r  outputbuffbuffer_cutoffarg
arg_lengths          rE   r"  zAbstractConnection.pack_command~  s    d1gu---d1gs#a)//12T!"X=DT!W_a)DH4D~~xSY)>)>)@(KL++t||**D1 	C SJD	M)-c:. ~~:s:'='='?J d#c" ~~"J..0  	!	4 	dr<   commandsc                    g }g }d}| j                   }|D ]  } | j                  | D ]  }t        |      }||kD  s||kD  st        |t              r*|r$|j                  t        j                  |             d}g }||kD  st        |t              r|j                  |       y|j                  |       ||z  }  |r$|j                  t        j                  |             |S )z.Pack multiple commands into the Redis protocolr   )rm   r"  r   r   r7  r   r3  r   )	rL   r>  r9  piecesbuffer_lengthr;  cmdchunkchunklens	            rE   pack_commandsz AbstractConnection.pack_commands  s      ++ 	.C***C0 .u:!M1-/!%4innV&<=$%MFm+z%/LMM%(MM%(!X-M!.	.& MM)..01r<   c                 F    t        | j                  j                        dk(  S )zCheck if the socket is emptyr   )r   ri   _bufferr   s    rE   _socket_is_emptyz#AbstractConnection._socket_is_empty  s    4<<''(A--r<   c                    K   | j                         s,| j                  d       d {    | j                         s+y y 7 w)NT)r(  )rH  r   r   s    rE   process_invalidation_messagesz0AbstractConnection.process_invalidation_messages  s:     '')$$$$777 '')7s   &AA AAtokenc                     || _         y rJ   )r   )rL   rK  s     rE   set_re_auth_tokenz$AbstractConnection.set_re_auth_token  s
    #r<   c                   K   | j                   l| j                  d| j                   j                  d      | j                   j                                d {    | j	                          d {    d | _         y y 7 %7 wNr   oid)r   r   try_get	get_valuer   r   s    rE   re_authzAbstractConnection.re_auth  s     *####++E2##--/  
 $$&&&"&D +
 's$   AB A<B .A>/B >B )TTr   NT)F)FN)Cr?   r@   rA   __doc__	__slots__r   rF   r2   r5   r   r  r   r   r0  boollistr>   r   r4   r$   ConnectCallbackTr'   r   r   warningsr   r   r   r   r   r   propertyr   r   r   r   r   r   r   r   r   r   r  r   r   r   r   r   r  r
  r   r   r  r  r  r   r%  r   r0   r	   r"  rE  rH  rJ  r   rM  rS  rD   r<   rE   rP   rP   k   s   :I@  "&*.26!&19'!&)6 %'(%)",%4%6"&!%9='.<@"#6:1X% #s(OX% 3-	X%
 !X% !)X% X% dIo.X% X% X% X% :&X% X%  %X% c]X%  3-!X%" c]#X%$ 3-%X%& 'X%( %%56)X%* G}+X%, &&89-X%. 3-/X%0 #?31X%t (0   /W   E E
/	MtJ'7 MD M;
 GK- -?C-^&&   S  C C# C>j:$ j:$ j:Xt  .I Q#(5/ #d #
 QU'UC%89'IM'	'R
 
s 
t 
V "'#'8
 %)',88 %8
 "8 tn8t,* ,e ,\hx
/C&D e <.8$~ $'r<   rP   c                        e Zd ZdZddddddded	eeef   d
edee	eeee
f   f      def
 fdZd Zde	fdZd ZdefdZ xZS )
Connectionz4Manages TCP communication to and from a Redis server	localhosti  FNr   )hostportsocket_keepalivesocket_keepalive_optionssocket_typer`  ra  rb  rc  rd  c                    || _         t        |      | _        || _        |xs i | _        || _        t        |   di | y NrD   )r`  r   ra  rb  rc  rd  superr   )rL   r`  ra  rb  rc  rd  r   r   s          rE   r   zConnection.__init__  sD     	I	 0(@(FB%&"6"r<   c                     d| j                   fd| j                  fd| j                  fg}| j                  r|j	                  d| j                  f       |S )Nr`  ra  rW   rY   )r`  ra  rW   rY   r   rL   r@  s     rE   r   zConnection.repr_pieces  sM    499%		':T477OLMM=$*:*:;<r<   r   c                 4    | j                   | j                  dS )Nr`  ra  rk  r   s    rE   _connection_argumentsz Connection._connection_arguments  s    		49955r<   c                   K   t        | j                        4 d{    t        j                  di | j	                          d{   \  }}ddd      d{    | _        | _        |j                  j                  d      }|r|j                  t        j                  t        j                  d       	 | j                  rs|j                  t        j                  t        j                  d       | j                   j#                         D ]&  \  }}|j                  t        j$                  ||       ( yyy7 7 7 # 1 d{  7  sw Y   xY w# t&        t(        f$ r |j+                           w xY ww)zCreate a TCP socket connectionNr   r|   rD   )r  r_   r   open_connectionrl  ri   rj   	transportget_extra_info
setsockoptr   IPPROTO_TCPTCP_NODELAYrb  
SOL_SOCKETSO_KEEPALIVErc  itemsSOL_TCPr   r   r   )rL   readerwritersockr   r   s         rE   r   zConnection._connect  sE     !<!<= 	 	#*#:#: $,,.$ NFF	 	 ..x8OOF..0B0BAF((OOF$5$5v7J7JAN $ = = C C E >11=> )	 		 	 	 	  Y'  	sp   E;D;E;'ED>EE;E AE;:A>E 8E;>E E;EE	EE;!E88E;c                 8    | j                    d| j                   S )N:rk  r   s    rE   r   zConnection._host_error  s    ))Adii[))r<   )r?   r@   rA   rV  r  r   r   rX  r   r
   r  r   r   rl  r   r   __classcell__r   s   @rE   r^  r^    s    :
   $!&NR# # CHo	#
 # #+73c5j8I3I+J"K# #"6w 60*S *r<   r^  c                   \    e Zd ZdZ	 	 	 	 	 	 	 	 	 	 ddee   dee   deeej                  f   dee	d      dee	d      dee   d	ee   d
e
dee   dee   f fdZdef fdZed        Zed        Zed        Zed        Zed        Zed        Zed        Zed        Zed        Z xZS )SSLConnectionzManages SSL connections to and from the Redis server(s).
    This class extends the Connection class, adding SSL functionality, and making
    use of ssl.SSLContext (https://docs.python.org/3/library/ssl.html#ssl.SSLContext)
    ssl_keyfilessl_certfilessl_cert_reqsssl_include_verify_flagsssl.VerifyFlagsssl_exclude_verify_flagsssl_ca_certsssl_ca_datassl_check_hostnamessl_min_versionssl_ciphersc                 z    t         st        d      t        |||||||||	|

      | _        t	        |   di | y )N$Python wasn't built with SSL support)
keyfilecertfile	cert_reqsinclude_verify_flagsexclude_verify_flagsca_certsca_datacheck_hostnamemin_versionciphersrD   )r   r-   RedisSSLContextrg   rg  r   )rL   r  r  r  r  r  r  r  r  r  r  r   r   s               rE   r   zSSLConnection.__init__  sQ     CDD,;!#!9!9!-'-
 	"6"r<   r   c                 ^    t         |          }| j                  j                         |d<   |S )Nssl)rg  rl  rg   r   )rL   r   r   s     rE   rl  z#SSLConnection._connection_arguments>  s-    .0((,,.ur<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.keyfileC      '''r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.certfileG      (((r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.cert_reqsK  s    )))r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  z"SSLConnection.include_verify_flagsO      444r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  z"SSLConnection.exclude_verify_flagsS  r  r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.ca_certsW  r  r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.ca_data[  r  r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.check_hostname_  s    ...r<   c                 .    | j                   j                  S rJ   )rg   r  r   s    rE   r  zSSLConnection.min_versionc  s    +++r<   )
NNrequiredNNNNTNN)r?   r@   rA   rV  r   r  r   r  
VerifyModer	   rX  r   r   r
   rl  r\  r  r  r  r  r  r  r  r  r  r}  r~  s   @rE   r  r    s    &*&*4>FJFJ&*%)#'04%)#c]# sm# S#..01	#
 #+40A+B"C# #+40A+B"C# sm# c]# !# "*-# c]#>w 
 ( ( ) ) * * 5 5 5 5 ) ) ( ( / / , ,r<   r  c                       e Zd ZdZ	 	 	 	 	 	 	 	 	 	 ddee   dee   deeeej                  f      dee	d      dee	d      d	ee   d
ee   de
dee   dee   fdZdefdZy)r  )r  r  r  r  r  r  r  contextr  r  r  Nr  r  r  r  r  r  r  r  r  r  r  c                    t         st        d      || _        || _        |t        j
                  }nWt        |t              rGt        j
                  t        j                  t        j                  d}||vrt        d|       ||   }|| _
        || _        || _        || _        || _        | j                  t        j
                  k7  r|nd| _        |	| _        |
| _        d | _        y )Nr  )noneoptionalr  z+Invalid SSL Certificate Requirements Flag: F)r   r-   r  r  r  	CERT_NONEr   r  CERT_OPTIONALCERT_REQUIREDr  r  r  r  r  r  r  r  r  )rL   r  r  r  r  r  r  r  r  r  r  	CERT_REQSs               rE   r   zRedisSSLContext.__init__w  s     CDD I	3'----I
 	) A)M  "),I"$8!$8! "nn=N5 	 '-1r<   r   c                    | j                   sdt        j                         }| j                  |_        | j                  |_        | j                  r&| j                  D ]  }|xj                  |z  c_         | j                  r'| j                  D ]  }|xj                  | z  c_         | j                  r3| j                  r'|j                  | j                  | j                         | j                  s| j                  r'|j                  | j                  | j                         | j                  | j                  |_        | j"                  |j%                  | j"                         || _         | j                   S )N)r  r  )cafilecadata)r  r  create_default_contextr  r  verify_moder  verify_flagsr  r  r  load_cert_chainr  r  load_verify_locationsr  minimum_versionr  set_ciphers)rL   r  flags      rE   r   zRedisSSLContext.get  s(   ||002G%)%8%8G""&..G(( 55 1D((D0(1(( 55 2D((TE1(2}}'''U}}--T]]4<<-X+*.*:*:'||'##DLL1"DL||r<   )
NNNNNNNFNN)r?   r@   rA   rW  r   r  r   r  r  r	   rX  r   r   r   r   rD   r<   rE   r  r  h  s    I  "&"&:>BFBF"&!%$,0!%)2#)2 3-)2 E#s~~"567	)2
 't,='>?)2 't,='>?)2 3-)2 #)2 )2 j))2 #)2VZ r<   r  c                   f     e Zd ZdZdddef fdZdeeeeee	f   f      fdZ
d Zdefd	Z xZS )
UnixDomainSocketConnectionz4Manages UDS communication to and from a Redis server pathr  c                2    || _         t        |   di | y rf  )r  rg  r   )rL   r  r   r   s      rE   r   z#UnixDomainSocketConnection.__init__  s    	"6"r<   r   c                     d| j                   fd| j                  fg}| j                  r|j                  d| j                  f       |S )Nr  rW   rY   )r  rW   rY   r   ri  s     rE   r   z&UnixDomainSocketConnection.repr_pieces  sB    499%dgg7MM=$*:*:;<r<   c                 D  K   t        | j                        4 d {    t        j                  | j                         d {   \  }}d d d       d {    | _        | _        | j                          d {    y 7 f7 @7 /# 1 d {  7  sw Y   ?xY w7  w)Nr  )r  r_   r   open_unix_connectionr  ri   rj   r   )rL   rx  ry  s      rE   r   z#UnixDomainSocketConnection._connect  s      !<!<= 	P 	P#*#?#?TYY#OONFF	P 	Poo		PO	P 	P 	P 	P 	 sb   B BB $B	BB	B B%B =B>B B	B 	BBBB c                     | j                   S rJ   r  r   s    rE   r   z&UnixDomainSocketConnection._host_error  s    yyr<   )r?   r@   rA   rV  r  r   r   r   r   r   r   r   r   r}  r~  s   @rE   r  r    sI    :&( # #XeCsCx,@&AB  S r<   r  )0FFALSENNOr   c                 v    | | dk(  ry t        | t              r| j                         t        v ryt	        |       S )Nr  F)r   r  upperFALSE_STRINGSrX  )values    rE   to_boolr    s4    }%%++-="@;r<   c                    | j                  dd      j                  dd      }g }|j                  d      D ]O  }|j                         }t        t        |      st        d|       |j                  t        t        |             Q |S )N[r  ]r   zInvalid ssl verify flag: )replacer2  striphasattrr   r   r   r   )r  verify_flags_strr  r  s       rE   parse_ssl_verify_flagsr    s     }}S"-55c2>L &&s+ 8zz|{D)8?@@GK67	8
 r<   )rW   r^   r_   rb  ra   max_connectionsrc   r  r  r  r#   .URL_QUERY_ARGUMENT_PARSERSc                   Z    e Zd ZU eed<   eed<   ee   ed<   eed<   eed<   eed<   eed<   y)	ConnectKwargsrX   r]   connection_classr`  ra  rW   r  N)r?   r@   rA   r  __annotations__r   rP   r   rD   r<   rE   r  r    s,    MM-..
I
IG
Ir<   r  F)totalurlc                    t        |       }i }t        |j                        j                         D ]N  \  }}|s	t	        |      dkD  st        |d         }t        j                  |      }|r	  ||      ||<   J|||<   P |j                  rt        |j                        |d<   |j                  rt        |j                        |d<   |j                  dk(  r/|j                  rt        |j                        |d<   t        |d<   |S |j                  d	v r|j                  rt        |j                        |d
<   |j                   rt#        |j                         |d<   |j                  r6d|vr2	 t#        t        |j                        j%                  dd            |d<   |j                  dk(  r	t(        |d<   |S d}t        d| d      # t        t        f$ r t        d| d      w xY w# t&        t        f$ r Y ^w xY w)Nr   zInvalid value for 'z' in connection URL.rX   r]   unixr  r  )redisredissr`  ra  rW   /r  r  zredis://, rediss://, unix://z5Redis URL must specify one of the following schemes ())r   r   queryrv  r   r   r  r   r   r   rX   r]   schemer  r  hostnamera  r   r  AttributeErrorr  )r  parsedr   name
value_listr  r   valid_schemess           rE   	parse_urlr    s   "3-FF$V\\288: 
%j#j/A-JqM*E/33D9FW#)%=F4L  %t
% $V__5z$V__5z }};;$V[[1F6N%?!"2 M/ 
-	-??$V__5F6N;; -F6N ;;4v-"76;;#7#?#?R#HIt ==H$)6F%& M 7CM?RST
 	
G ":. W$':4&@T%UVVW6 #J/ s   /G/1G- G*-G?>G?_CPConnectionPool)boundc                       e Zd ZdZedee   dedefd       Ze	dfdee
   dee   fd	Zd
 Zd ZdefdZ edgdd      d!d       Zd Zd Zd Zde
fdZde
fdZd"defdZd Zd#dZd$dZdefdZdefd Z y)%r  a  
    Create a connection pool. ``If max_connections`` is set, then this
    object raises :py:class:`~redis.ConnectionError` when the pool's
    limit is reached.

    By default, TCP connections are created unless ``connection_class``
    is specified. Use :py:class:`~redis.UnixDomainSocketConnection` for
    unix sockets.
    :py:class:`~redis.SSLConnection` can be used for SSL enabled connections.

    Any additional keyword arguments are passed to the constructor of
    ``connection_class``.
    clsr  r   c                 J    t        |      }|j                  |        | di |S )a  
        Return a connection pool configured from the given URL.

        For example::

            redis://[[username]:[password]]@localhost:6379/0
            rediss://[[username]:[password]]@localhost:6379/0
            unix://[username@]/path/to/socket.sock?db=0[&password=password]

        Three URL schemes are supported:

        - `redis://` creates a TCP socket connection. See more at:
          <https://www.iana.org/assignments/uri-schemes/prov/redis>
        - `rediss://` creates a SSL wrapped TCP socket connection. See more at:
          <https://www.iana.org/assignments/uri-schemes/prov/rediss>
        - ``unix://``: creates a Unix Domain Socket connection.

        The username, password, hostname, path and all querystring values
        are passed through urllib.parse.unquote in order to replace any
        percent-encoded values with their corresponding characters.

        There are several ways to specify a database number. The first value
        found will be used:

        1. A ``db`` querystring option, e.g. redis://localhost?db=0

        2. If using the redis:// or rediss:// schemes, the path argument
               of the url, e.g. redis://localhost/0

        3. A ``db`` keyword argument to this function.

        If none of these options are specified, the default db=0 is used.

        All querystring options are cast to their appropriate Python types.
        Boolean arguments can be specified with string values "True"/"False"
        or "Yes"/"No". Values that cannot be properly cast cause a
        ``ValueError`` to be raised. Once parsed, the querystring arguments
        and keyword arguments are passed to the ``ConnectionPool``'s
        class initializer. In the case of conflicting arguments, querystring
        arguments always win.
        rD   )r  update)r  r  r   url_optionss       rE   from_urlzConnectionPool.from_urlL  s&    V  nk"}V}r<   Nr  r  c                    |xs d}t        |t              r|dk  rt        d      || _        || _        || _        g | _        t               | _        | j                  j                  dt              | _        t        j                         | _        | j                  j                  dd       | _        | j                  t!               | _        y y )Nl        r   z,"max_connections" must be a positive integerry   rz   )r   r   r   r  connection_kwargsr  _available_connectionsset_in_use_connectionsr   r5   ry   r   Lockrn   r~   r   )rL   r  r  r  s       rE   r   zConnectionPool.__init__{  s     *2U/3/?Q3FKLL 0!2.@B#<?E !3377Q\\^
!%!7!7!;!;<NPT!U!!)%4%6D" *r<   c                 T   dj                  | j                  j                         D cg c]  \  }}| d|  c}}      }d| j                  j                   d| j                  j
                   d| j                  j                   d| j                  j
                   d| dS c c}}w )Nr   r   r   r   z(<r   z)>)>)r   r  rv  r   r@   r?   r  )rL   r   r   conn_kwargss       rE   r   zConnectionPool.__repr__  s    hht7M7M7S7S7UVtq!1#Qqc
VW))*!DNN,C,C+D&&112!D4I4I4R4R3S}D"	
  Ws   B$
c                 D    g | _         t        j                         | _        y rJ   )r  r   WeakSetr  r   s    rE   resetzConnectionPool.reset  s    &(##*??#4 r<   c                 b    | j                   xs" t        | j                        | j                  k  S )z;Return True if a connection can be retrieved from the pool.)r  r   r  r  r   s    rE   can_get_connectionz!ConnectionPool.can_get_connection  s1     '' D4++,t/C/CC	
r<   *)Use get_connection() without args instead5.3.0args_to_warnreasonversionc                 6  K   | j                   4 d {    	 | j                         }	 | j                  |       d {    d d d       d {    |S 7 A7 # t        $ r | j	                  |       d {  7    w xY w7 2# 1 d {  7  sw Y   S xY wwrJ   )rn   get_available_connectionensure_connectionr  release)rL   command_namekeysoptionsrM   s        rE   get_connectionzConnectionPool.get_connection  s      :: 	 	:668J,,Z888		 	 	 9  ll:...	 	 	 	 sq   BABBAAABBBAA>6A97A>>BBB	B
BBc                    	 | j                   j                         }| j                  j                  |       |S # t        $ rA t        | j                        | j
                  k\  rt        d      d| j                         }Y fw xY w)zCGet a connection from the pool, without making sure it is connectedzToo many connectionsN)	r  pop
IndexErrorr   r  r  r+   make_connectionaddrK   s     rE   r  z'ConnectionPool.get_available_connection  s}    	04488:J
 	  $$Z0  	04++,0D0DD%&<=4G--/J	0s   9 ABBc                     | j                   }| j                  |j                  dd      |j                  dd      |j                  dd            S )z,Return an encoder based on encoding settingsrs   rq   rt   rr   ru   F)rs   rt   ru   )r  ry   r   )rL   r   s     rE   get_encoderzConnectionPool.get_encoder  sQ    ''!!ZZ
G4"JJ'8(C#ZZ(:EB " 
 	
r<   c                 :     | j                   di | j                  S )z=Create a new connection.  Can be overridden by child classes.rD   )r  r  r   s    rE   r  zConnectionPool.make_connection  s    $t$$>t'='=>>r<   rM   c                 p  K   |j                          d{    	 |j                          d{   rt        d      dy7 *7 # t        t        t        f$ rZ |j                          d{  7   |j                          d{  7   |j                          d{  7  rt        d      dY yw xY ww)z8Ensure that the connection object is connected and validNzConnection has datazConnection not ready)r   r%  r+   r/   r   r   rK   s     rE   r  z ConnectionPool.ensure_connection  s       """
	H44666%&;<$F 7 	# 7w7 	H'')))$$&&&44666%&<=4G 7	Hsf   B6AB6A AA  B6A 'B3,A/-B3BB3B!B30B62B33B6c                 (  K   | j                   j                  |       |j                         r|j                          d{    | j                  j                  |       | j                  j                  t        |             d{    y7 L7 w)z(Releases the connection back to the poolN)	r  r   r   r   r  r   r~   dispatch_asyncr   rK   s     rE   r  zConnectionPool.release  s      	  ''
3&&('')))##**:6$$33-j9
 	
 	
 *	
s$   ?BBABB	BBinuse_connectionsc                    K   |r!t        | j                  | j                        }n| j                  }t        j                  d |D        ddi d{   }t        d |D        d      }|r|y7 w)z
        Disconnects connections in the pool

        If ``inuse_connections`` is True, disconnect connections that are
        current in use, potentially by other tasks. Otherwise only disconnect
        connections that are idle in the pool.
        c              3   <   K   | ]  }|j                           y wrJ   r   )r   rM   s     rE   r   z,ConnectionPool.disconnect.<locals>.<genexpr>  s     D*j##%Ds   return_exceptionsTNc              3   B   K   | ]  }t        |t              s|  y wrJ   )r   r  )r   rs     rE   r   z,ConnectionPool.disconnect.<locals>.<genexpr>  s     D!z!]'CADs   )r   r  r  r   gathernext)rL   r   connectionsrespr   s        rE   r   zConnectionPool.disconnect  s|      8=++T-E-E9K 55K^^DD
"
 
 DtDdKI 
s   AA0A.A0c                    K   | j                   4 d{    | j                  D ]  }|j                           ddd      d{    y7 67 # 1 d{  7  sw Y   yxY ww)z<
        Mark all active connections for reconnect.
        N)rn   r  r   )rL   conns     rE   'update_active_connections_for_reconnectz6ConnectionPool.update_active_connections_for_reconnect   s_      :: 	* 	*00 *'')*	* 	* 	* 	* 	* 	* 	*sC   A#A
A#"AA#AA#A#A AA A#c                 @   K   | j                          d{    y7 w)z-Close the pool, disconnecting all connectionsNr   r   s    rE   aclosezConnectionPool.aclose  r  r  c                 d    | j                   D ]	  }||_         | j                  D ]	  }||_         y rJ   )r  rx   r  )rL   rx   r+  s      rE   	set_retryzConnectionPool.set_retry  s:    // 	DDJ	,, 	DDJ	r<   rK  c                    K    j                   4 d {     j                  D ]W  j                  j                  fd fd       d {    j                  j                  fd fd       d {    Y  j                  D ]  j                          d d d       d {    y 7 7 e7 =7 # 1 d {  7  sw Y   y xY ww)Nc                  d     j                  dj                  d      j                               S rO  )r   rQ  rR  )r+  rK  s   rE   r   z1ConnectionPool.re_auth_callback.<locals>.<lambda>  s)    D--e 4eoo6G r<   c                 &    j                  |       S rJ   _mockr   s    rE   r   z1ConnectionPool.re_auth_callback.<locals>.<lambda>  s    $**U"3 r<   c                  $     j                         S rJ   )r   )r+  s   rE   r   z1ConnectionPool.re_auth_callback.<locals>.<lambda>  s    D..0 r<   c                 &    j                  |       S rJ   r4  r   s    rE   r   z1ConnectionPool.re_auth_callback.<locals>.<lambda>  s    

5@Q r<   )rn   r  rx   r   r  rM  )rL   rK  r+  s   ``@rE   re_auth_callbackzConnectionPool.re_auth_callback  s     :: 	. 	.33 	jj00 4	   jj0002Q  	 00 .&&u-.	. 	. 	.	. 	. 	. 	.sh   CB4C6B<B6)B<:B8;(B<#C.B:/C6B<8B<:C<CCC
Cr   c                    K   yw)z
        Dummy functions, needs to be passed as error callback to retry object.
        :param error:
        :return:
        NrD   r	  s     rE   r5  zConnectionPool._mock!  s      	rU   rJ   rU  rT  )rx   r$   r   N)!r?   r@   rA   rV  classmethodr   r  r  r  r^  rP   r   r   r   r   r  rX  r  r   r  r  r  r  r  r  r   r,  r.  r0  r   r8  r-   r5  rD   r<   rE   r  r  =  s     ,d3i ,c , , ,` 6@)-7127 "#7,
5
D 
 U:



	
?H2D H 
(: 
$ ,* .N . r<   c            
            e Zd ZdZddeej                  fdedee	   de
e   de
ej                     f fdZ ed	gd
d      d fd	       Zdef fdZ xZS )BlockingConnectionPoola  
    A blocking connection pool::

        >>> from redis.asyncio import Redis, BlockingConnectionPool
        >>> client = Redis.from_pool(BlockingConnectionPool())

    It performs the same function as the default
    :py:class:`~redis.asyncio.ConnectionPool` implementation, in that,
    it maintains a pool of reusable connections that can be shared by
    multiple async redis clients.

    The difference is that, in the event that a client tries to get a
    connection from the pool when all of connections are in use, rather than
    raising a :py:class:`~redis.ConnectionError` (as the default
    :py:class:`~redis.asyncio.ConnectionPool` implementation does), it
    blocks the current `Task` for a specified number of seconds until
    a connection becomes available.

    Use ``max_connections`` to increase / decrease the pool size::

        >>> pool = BlockingConnectionPool(max_connections=10)

    Use ``timeout`` to tell it either how many seconds to wait for a connection
    to become available, or to block forever:

        >>> # Block forever.
        >>> pool = BlockingConnectionPool(timeout=None)

        >>> # Raise a ``ConnectionError`` after five seconds if a connection is
        >>> # not available.
        >>> pool = BlockingConnectionPool(timeout=5)
    2      r  r#   r  queue_classc                 j    t        |   d||d| t        j                         | _        || _        y )N)r  r  rD   )rg  r   r   	Condition
_conditionr#   )rL   r  r#   r  r?  r  r   s         rE   r   zBlockingConnectionPool.__init__L  s@     	 	
-+	
  	

 "++-r<   r  r  r  r	  c                 l  K   	 | j                   4 d{    t        | j                        4 d{    | j                   j                  | j                         d{    t
        |          }ddd      d{    ddd      d{    	 | j                         d{    |S 7 7 }7 R7 6# 1 d{  7  sw Y   FxY w7 =# 1 d{  7  sw Y   MxY w# t        j                  $ r}t        d      |d}~ww xY w7 a# t        $ r | j                         d{  7    w xY ww)z@Gets a connection from the pool, blocking until one is availableNzNo connection available.)rB  r  r#   r  r  rg  r  r   r/   r+   r  r  r  )rL   r  r  r  rM   errr   s         rE   r  z%BlockingConnectionPool.get_connection\  s/    	G D D(6 D D//2243J3JKKK!&!A!CJD DD D	((444DDKD D D DD D D D ## 	G!"<=3F	G
 5 	,,z***	s   D4C# B/C# CB1C)B7 B3!B73C>B5?CC# CC# D (D
)D .D4/C# 1C3B75C7C		=C >C		CC# C CC C# #D6DDD4
D D1)D,*D11D4rM   c                    K   | j                   4 d{    t        | 	  |       d{    | j                   j                          ddd      d{    y7 F7 17 	# 1 d{  7  sw Y   yxY ww)z)Releases the connection back to the pool.N)rB  rg  r  notify)rL   rM   r   s     rE   r  zBlockingConnectionPool.releases  sc     ?? 	% 	%'/*---OO""$	% 	% 	%-	% 	% 	% 	%sT   A6AA6A!AA!
A6AA6A!A6!A3'A*(A3/A6rJ   )r?   r@   rA   rV  r^  r   	LifoQueuer   r   r0  r   rP   Queuer   r   r  r  r}  r~  s   @rE   r<  r<  *  s    F  "#%5?+2+<+< % 12	
 '--(  U:

$%(: % %r<   r<  )or   r   enumr   r   sysr[  r   abcr   	itertoolsr   typesr   typingr   r   r   r	   r
   r   r   r   r   r   r   r   r   urllib.parser   r   r   r   utilsr   r  r   r   r   
auth.tokenr   eventr   r   r   r    version_infor#   r  redis.asyncio.retryr$   redis.backoffr%   redis.connectionr&   redis.credentialsr'   r(   redis.exceptionsr)   r*   r+   r,   r-   r.   r/   redis.typingr0   redis.utilsr1   r2   r3   _parsersr4   r5   r6   r7   r8   r4  r8  r5  SYM_LFr3  Enumr>   rC   r   r  rF   rH   rS   rZ  rP   r^  r  r  r  r  rX  r  r  r   r0  r  r  rB   r  r  r  r  r<  rD   r<   rE   <module>r^     s@        
     "    B A !77
CJJK ' F 9 z!06 % # 1 T   $ H H  
				   E+->@SSTU U'M%M>h >D8 D 02NNO r	' r	'j6*# 6*rL,J L,^N Nb!3 0 /htn  CS"'##!$%$:$:C GC#v+)>$>? "IU 23 2= 2j e+,j jZM%^ M%r<   