
    j,g                        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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jH                  d	k\  rd d
l m%Z& nd d
l&m%Z& d dl'm(Z( d dl)m*Z* d dl+m,Z, d dl-m.Z.m/Z/ d dl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 d dl8m9Z9 d dl:m;Z;m<Z<m=Z= ddl>m?Z?m@Z@mAZAmBZBmCZC dZDdZEdZFdZGdZH G d dej                        ZJeJj                  ZLeeeBeCeAf      eMd<   e;reAZNneBZN G d de      ZO G d de      ZPeeOePf   ZQ G d d       ZR G d! d"eR      ZS G d# d$eS      ZT G d% d&      ZU G d' d(eR      ZVd)ZWd*eeX   fd+ZY eeZe[e[eYeYeZeZeYe[d,	      Z\ee]ed-e^f   f   eMd.<    G d/ d0ed12      Z_d3e]d*e_fd4Z` ed5d67      Za G d8 d6      Zb G d9 d:eb      Zcy);    N)abstractmethod)chain)MappingProxyType)AnyCallableIterableListMappingOptionalProtocolSetTupleType	TypedDictTypeVarUnion)ParseResultparse_qsunquoteurlparse   )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 r4   r/var/www/trellinator.diamondhoofcare.com/public_html/venv/lib/python3.12/site-packages/redis/asyncio/connection.pyr6   r6   F   s	    xHr4   r6   DefaultParserc                       e Zd ZddZy)ConnectCallbackProtocolc                      y Nr<   self
connections     r=   __call__z ConnectCallbackProtocol.__call__U   s    r4   NrE   AbstractConnectionr7   r8   r9   rF   r<   r4   r=   r@   r@   T   s    =r4   r@   c                       e Zd ZddZy)AsyncConnectCallbackProtocolc                    K   y wrB   r<   rC   s     r=   rF   z%AsyncConnectCallbackProtocol.__call__Y   s        NrG   rI   r<   r4   r=   rK   rK   X   s    Cr4   rK   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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   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#ed-        Z$ed*e
fd.       Z%d/e&d*e
fd0Z'dKd1Z(dLd2ed*dfd3Z)d4 Z*d5 Z+d6 Z,d7e-e.   d*dfd8Z/	 dMd7e	e.e
e-e.   f   d:ed*dfd;Z0d<ed=ed*dfd>Z1d? Z2	 	 dNd9dd@dAedBee   dCedDee   fdEZ3d<e4d*e5e.   fdFZ6dGe-e-e4      d*e5e.   fdHZ7dI Z8dJ Z9y)OrH   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   )rO   rU   rV   rW   rY   rZ   encodingencoding_errorsdecode_responsesparser_classsocket_read_sizer[   rQ   rR   rS   rP   retryrX   encoder_classrT   r`   rO   rU   rV   rW   rY   rZ   rk   rl   rm   rn   ro   r[   rQ   rR   rS   rP   rp   rX   rq   rT   r`   c                   |s|r|t        d      || _        || _        || _        || _        || _        || _        || _        || _        ||}|| _	        || _
        |t        u rg }|rS|j                  t               |j                  t        j                         |j                  t         j                         || _        |s|rR|st%        t'               d      | _        nt+        j,                  |      | _        | j(                  j/                  |       nt%        t'               d      | _        || _        d| _         ||||	      | _        || _        d | _        d | _        || _        | j?                  |
       g | _         d| _!        	 tE        |      }|dk  s|dkD  rtM        d	      || _'        y # tF        $ r	 tH        }Y .tJ        $ r tM        d      w xY w# dk  s|dkD  rtM        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  zprotocol must be an integerr   r   zprotocol must be either 2 or 3)(r$   rO   rQ   rR   rS   rT   rU   rP   rV   rW   rY   SENTINELappendr'   socketr   asynciorZ   r   r   rp   copydeepcopyupdate_supported_errorsr[   r\   r^   rX   ra   rb   rg   
set_parserrd   re   int	TypeErrorr   
ValueErrorr#   r`   )rD   rO   rU   rV   rW   rY   rZ   rk   rl   rm   rn   ro   r[   rQ   rR   rS   rP   rp   rX   rq   rT   r`   ps                          r=   __init__zAbstractConnection.__init__   s   2 &9&E+  & &#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"		%HA 1uA%&FGG$DM  	%$A 	A!"?@@	A 1uA%&FGG$DMs$   
F2 2GG GG G8	_warningsc                     t        | dd       rA|j                  d| t        |        	 t        j                          | j                          y y # t        $ r Y y w xY w)Nrb   zunclosed Connection )source)getattrwarnResourceWarningrx   get_running_loop_closeRuntimeError)rD   r   s     r=   __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)rb   closera   rD   s    r=   r   zAbstractConnection._close   s.     <<LL *..DL4< r4   c                     dj                  d | j                         D              }d| j                  j                   d| j                  j                   d| dS )N,c              3   0   K   | ]  \  }}| d |   yw)=Nr<   ).0kvs      r=   	<genexpr>z.AbstractConnection.__repr__.<locals>.<genexpr>   s     HTQ1QCjHs   <.()>)joinrepr_pieces	__class__r8   r7   )rD   	repr_argss     r=   __repr__zAbstractConnection.__repr__   sQ    HHHT5E5E5GHI	4>>,,-Qt~~/F/F.GqSUVVr4   c                      y rB   r<   r   s    r=   r   zAbstractConnection.repr_pieces       r4   c                 >    | j                   d uxr | j                  d uS rB   )ra   rb   r   s    r=   is_connectedzAbstractConnection.is_connected   s    ||4'DDLL,DDr4   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
WeakMethodrd   rv   )rD   callbackwms      r=   register_connect_callbackz,AbstractConnection.register_connect_callback   s;     )T,,,##**2. -r4   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)rd   remover   r   r   )rD   r   s     r=   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
        )ro   N)rg   rc   )rD   rn   s     r=   r|   zAbstractConnection.set_parser  s     $T5K5KLr4   c                    K    j                   ry	  j                  j                   fd fd       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 # 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 ?7 	# t        $ r  j!                          d{  7    w xY wc c}w 7 w)z5Connects to the Redis server if not already connectedNc                  $     j                         S rB   )_connectr   s   r=   <lambda>z,AbstractConnection.connect.<locals>.<lambda>  s     r4   c                 $    j                         S rB   
disconnect)errorrD   s    r=   r   z,AbstractConnection.connect.<locals>.<lambda>  s    t7H r4   zTimeout connecting to server)r   rp   call_with_retryrx   CancelledErrorrw   r   r'   OSErrorr#   _error_message	ExceptionrX   
on_connectiscoroutinefunctionr%   r   rd   inspectisawaitable)rD   eexcrefr   tasks   `     r=   connectzAbstractConnection.connect  s    	0**,,')H  	**oo'''
 2243J3JK 11$777006 372I2I"S3SU3"S** 	CuHD>D++D1

		A %% 	 4 45 	?=>> 	:!$"5"5a"899 	0!#&C/	0 ( 8  	//###	 #T
 s   G%D DD F! F8F! FF! +G9GG
(G3G	GGGD A	F!E;;FFFGF! F! !G=G >G	Gc                    K   y wrB   r<   r   s    r=   r   zAbstractConnection._connect?  s	     rM   c                      y rB   r<   r   s    r=   _host_errorzAbstractConnection._host_errorC  r   r4   	exceptionc                 6    t        | j                         |      S rB   )r   r   )rD   r   s     r=   r   z!AbstractConnection._error_messageG  s    #D$4$4$6	BBr4   c                 B	  K   | j                   j                  |        | j                   }d}| j                  s| j                  s| j                  r>| j                  xs  t        | j                  | j                        }|j                         }|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{    | j                          d{   }|j!                  d      t#        | j                        k7  r4|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,                  rT| 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 Z7 D# t&        $ r= | j                  d|d   d       d{  7   | j                          d{  7  }Y w xY w7 7 7 7 7 B7 7 7 # t4        $ r Y w xY w7 w)z=Initialize the connection, authenticate and select a databaseN)r   2rs   defaultr   HELLOAUTHs   protoprotozInvalid RESP versioncheck_healthFrt   r   OKzInvalid Username or PasswordCLIENTSETNAMEzError setting client nameSETINFOzLIB-NAMEzLIB-VERSELECTc              3   &   K   | ]	  }|s|  y wrB   r<   )r   sents     r=   r   z0AbstractConnection.on_connect.<locals>.<genexpr>  s     M4$Ms   zInvalid Database)rc   r   rT   rP   rU   r    get_credentialsr`   
isinstancer/   r|   r0   EXCEPTION_CLASSESlensend_commandread_responsegetr}   r#   r"   r+   r!   rQ   rR   rS   rO   r&   )rD   parser	auth_argscred_providerresponseauth_response_s          r=   r   zAbstractConnection.on_connectJ  s    %	##(( T5dmmT]]S  &557I h6$,,(9: 12171I1I.''-9~"&	!5	#$##GT]]FOYOOO!//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]];;;!//11H ##Hi9I9IJJJ$"4"4"6674?%&ABB ==##HiT]]SSS##HiDDTDTUUU77##Hdgg666 NDMM43C3C#DM 	A((***	 77$"4"4"6674?%&899 @ y P1 L !;7 ;
 ''	"E'RRR&*&8&8&: : :;$ <1 K6
 TU 7
 +   7sA  D:R<P#=RP&A:RP)RP/ (P,)P/ -B+RQ8R0Q;10R!Q>"R>R?ARR1R4R5/R$R
%&RRR R$&R
RR&R)R,P/ /#Q5QQ5+Q.,Q51R4Q55R;R>RRRR
RR	RRRR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_timeoutrW   rc   on_disconnectr   rb   r   wait_closedr   ra   rx   r'   )rD   r   s     r=   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   r+   r   r#   r   s    r=   
_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   )rD   r   s     r=   _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)	r[   rx   r   timer\   rp   r   r   r   r   s    r=   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rB   )rb   
writelinesdrain)rD   r   s     r=   _send_packed_commandz'AbstractConnection._send_packed_command  s.     (ll  """s   9AAATr   c                   K   | j                   s| j                          d {    n|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)
NTr   zTimeout writing to socketrs   UNKNOWNr   zError z while writing to socket. r   )r   r   r   r   strencodebytesrV   rx   wait_forr   rb   r   r   r'   r   r   r   argsr#   BaseException)rD   r   r   r   err_noerrmsgs         r=   send_packed_commandz&AbstractConnection.send_packed_command  s       ,,.  ##%%%	'3'!..*'5)")""&&--g68K8K   ''0ll((*** !%
 +## 	F///...:;E 		///...166{a!*AFF1I! :6(!D  	
 ///...	s    G$C8G$C:G$A1D  4C<5D  9G$:8D  2C>3D  7G$:G$<D  >D   (G!(D+)G!F8EA F88!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   )rD   r  r  s      r=   r   zAbstractConnection.send_command  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: )rc   can_read_destructiver   r   r   r#   r  )rD   r   
host_errors      r=   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                         }	 |^| j                  dv rPt        sJt	        |      4 d{    | j
                  j                  ||       d{   }ddd      d{    n|It	        |      4 d{    | j
                  j                  |       d{   }ddd      d{    n^| j                  dv r,t        s&| j
                  j                  ||       d{   }n$| j
                  j                  |       d{   }| j                  r6t        j                         j!                         | j                  z   }	|	| _        t%        t&              r|d|S 7 ;7 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 : )rV   r   r`   r)   r   rc   r   rx   r'   r   r   r#   r  r  r[   r   r   r\   r   r&   )
rD   r  r   r  r  read_timeoutr  r   r   	next_times
             r=   r   z AbstractConnection.read_response  sq     #*"5w4;N;N%%'
)	(MMX-)(6  %)\\%?%?)9 &@ &  H   )(6  %)\\%?%?)9 &@ &  H   (*3D!%!;!;%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%  !I2%G 	F
G !F.F/F3G >F
?G F"G  F)9F%:F)>G 	F'
9G F>$G (G )G -AI2G F
G FFFG %F)'G )F;/F20F;7G  G I/I2I/0G31I/I#H&$ I#I/'I*(I//I2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   rs   N    )r   floatr   tupler   split	SYM_EMPTYr   SYM_STARr   SYM_CRLFre   mapr^   
memoryview
SYM_DOLLARrv   )rD   r  outputbuffbuffer_cutoffarg
arg_lengths          r=   r  zAbstractConnection.pack_commandD  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r4   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   )re   r  r   r   r  rv   r  r   )	rD   r$  r  piecesbuffer_lengthr!  cmdchunkchunklens	            r=   pack_commandsz AbstractConnection.pack_commandsr  s      ++ 	.C***C0 .u:!M1-/!%4innV&<=$%MFm+z%/LMM%(MM%(!X-M!.	.& MM)..01r4   c                 F    t        | j                  j                        dk(  S )zCheck if the socket is emptyr   )r   ra   _bufferr   s    r=   _socket_is_emptyz#AbstractConnection._socket_is_empty  s    4<<''(A--r4   c                    K   | j                         s,| j                  d       d {    | j                         s+y y 7 w)NT)r  )r.  r   r   s    r=   process_invalidation_messagesz0AbstractConnection.process_invalidation_messages  s:     '')$$$$777 '')7s   &AA AAr   N)FT)FN):r7   r8   r9   __doc__	__slots__ru   r>   r*   r-   r   r   r}   r   r  boollistr6   r   r,   r   ConnectCallbackTr   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<   r4   r=   rH   rH   _   sb   :I@  "&*.26!&19'!&)6 %'(%)",%4%6"&!%9='.<@"#/P% #s(OP% 3-	P%
 !P% !)P% P% dIo.P% P% P% P% :&P% P%  %P% c]P%  3-!P%" c]#P%$ 3-%P%& 'P%( %%56)P%* G}+P%, &&89-P%. 3-/P%d (0   /W   E E
/	MtJ'7 MD M)V   S  C C# CU:nt  .I Q#(5/ #d #
 QU'UC%89'IM'	'R
 
s 
t 
V "'#'<
 %)',<< %<
 "< tn<|,* ,e ,\hx
/C&D e <.8r4   rH   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=  r>  r?  r@  rA  c                    || _         t        |      | _        || _        |xs i | _        || _        t        |   di | y Nr<   )r=  r}   r>  r?  r@  rA  superr   )rD   r=  r>  r?  r@  rA  r  r   s          r=   r   zConnection.__init__  sD     	I	 0(@(FB%&"6"r4   c                     d| j                   fd| j                  fd| j                  fg}| j                  r|j	                  d| j                  f       |S )Nr=  r>  rO   rQ   )r=  r>  rO   rQ   rv   rD   r&  s     r=   r   zConnection.repr_pieces  sM    499%		':T477OLMM=$*:*:;<r4   r   c                 4    | j                   | j                  dS )Nr=  r>  rH  r   s    r=   _connection_argumentsz Connection._connection_arguments  s    		49955r4   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 connectionNrw   rs   r<   )r   rW   rx   open_connectionrI  ra   rb   	transportget_extra_info
setsockoptrw   IPPROTO_TCPTCP_NODELAYr?  
SOL_SOCKETSO_KEEPALIVEr@  itemsSOL_TCPr   r~   r   )rD   readerwritersockr   r   s         r=   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:rH  r   s    r=   r   zConnection._host_error  s    ))Adii[))r4   )r7   r8   r9   r3  r   r   r}   r5  r   r
   r   r   r   rI  r   r   __classcell__r   s   @r=   r;  r;    s    :
   $!&NR# # CHo	#
 # #+73c5j8I3I+J"K# #"6w 60*S *r4   r;  c                       e Zd ZdZ	 	 	 	 	 	 	 	 ddee   dee   dedee   dee   dedeej                     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 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_ca_certsssl_ca_datassl_check_hostnamessl_min_versionssl_ciphersc	           
      T    t        ||||||||      | _        t        
|   di |	 y )N)keyfilecertfile	cert_reqsca_certsca_datacheck_hostnamemin_versionciphersr<   )RedisSSLContextr_   rD  r   )rD   r^  r_  r`  ra  rb  rc  rd  re  r  r   s             r=   r   zSSLConnection.__init__  s<     -<!#!-'	-
 	"6"r4   r   c                 ^    t         |          }| j                  j                         |d<   |S )Nssl)rD  rI  r_   r   )rD   r  r   s     r=   rI  z#SSLConnection._connection_arguments  s-    .0((,,.ur4   c                 .    | j                   j                  S rB   )r_   rg  r   s    r=   rg  zSSLConnection.keyfile      '''r4   c                 .    | j                   j                  S rB   )r_   rh  r   s    r=   rh  zSSLConnection.certfile      (((r4   c                 .    | j                   j                  S rB   )r_   ri  r   s    r=   ri  zSSLConnection.cert_reqs  s    )))r4   c                 .    | j                   j                  S rB   )r_   rj  r   s    r=   rj  zSSLConnection.ca_certs  ru  r4   c                 .    | j                   j                  S rB   )r_   rk  r   s    r=   rk  zSSLConnection.ca_data  rs  r4   c                 .    | j                   j                  S rB   )r_   rl  r   s    r=   rl  zSSLConnection.check_hostname	  s    ...r4   c                 .    | j                   j                  S rB   )r_   rm  r   s    r=   rm  zSSLConnection.min_version  s    +++r4   )NNrequiredNNFNN)r7   r8   r9   r3  r   r   r5  rq  
TLSVersionr   r
   rI  r9  rg  rh  ri  rj  rk  rl  rm  rZ  r[  s   @r=   r]  r]    s*    &*&*'&*%)#(48%)#c]# sm# 	#
 sm# c]# !# "#..1# c]#0w 
 ( ( ) ) * * ) ) ( ( / / , ,r4   r]  c                       e Zd ZdZ	 	 	 	 	 	 	 	 ddee   dee   dee   dee   dee   ded	eej                     d
ee   fdZ	dej                  fdZy)ro  )	rg  rh  ri  rj  rk  contextrl  rm  rn  Nrg  rh  ri  rj  rk  rl  rm  rn  c	                 \   || _         || _        |t        j                  | _        n\t        |t              rLt        j                  t        j                  t        j                  d}	||	vrt        d|       |	|   | _        || _
        || _        || _        || _        || _        d | _        y )N)noneoptionalr{  z+Invalid SSL Certificate Requirements Flag: )rg  rh  rq  	CERT_NONEri  r   r   CERT_OPTIONALCERT_REQUIREDr%   rj  rk  rl  rm  rn  r~  )
rD   rg  rh  ri  rj  rk  rl  rm  rn  	CERT_REQSs
             r=   r   zRedisSSLContext.__init__  s       ]]DN	3'----I
 	) A)M  'y1DN ,&15r4   r   c                 0   | j                   st        j                         }| j                  |_        | j                  |_        | j                  r3| j                  r'|j                  | j                  | j                         | j                  s| j                  r'|j                  | j                  | j                         | j                  | j                  |_        | j                  |j                  | j                         || _         | j                   S )N)rh  rg  )cafilecadata)r~  rq  create_default_contextrl  ri  verify_moderh  rg  load_cert_chainrj  rk  load_verify_locationsrm  minimum_versionrn  set_ciphers)rD   r~  s     r=   r   zRedisSSLContext.get@  s    ||002G%)%8%8G""&..G}}'''U}}--T]]4<<-X+*.*:*:'||'##DLL1"DL||r4   )NNNNNFNN)r7   r8   r9   r4  r   r   r5  rq  r|  r   
SSLContextr   r<   r4   r=   ro  ro    s    
I "&"&#'"&!%$04!%6#6 3-6 C=	6
 3-6 #6 6 cnn-6 #6BS^^ r4   ro  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 rC  )r  rD  r   )rD   r  r  r   s      r=   r   z#UnixDomainSocketConnection.__init__T  s    	"6"r4   r   c                     d| j                   fd| j                  fg}| j                  r|j                  d| j                  f       |S )Nr  rO   rQ   )r  rO   rQ   rv   rF  s     r=   r   z&UnixDomainSocketConnection.repr_piecesX  sB    499%dgg7MM=$*:*:;<r4   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   rW   rx   open_unix_connectionr  ra   rb   r   )rD   rU  rV  s      r=   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 rB   r  r   s    r=   r   z&UnixDomainSocketConnection._host_errore  s    yyr4   )r7   r8   r9   r3  r   r   r   r   r   r}   r   r   r   rZ  r[  s   @r=   r  r  Q  sI    :&( # #XeCsCx,@&AB  S r4   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_STRINGSr5  )values    r=   to_boolr  l  s4    }%%++-="@;r4   )	rO   rV   rW   r?  rY   max_connectionsr[   rc  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)	ConnectKwargsrP   rU   connection_classr=  r>  rO   r  N)r7   r8   r9   r   __annotations__r   rH   r}   r<   r4   r=   r  r    s,    MM-..
I
IG
Ir4   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.rP   rU   unixr  r  )redisredissr=  r>  rO   /r  r  zredis://, rediss://, unix://z5Redis URL must specify one of the following schemes ())r   r   queryrS  r   r   r  r   r~   r   rP   rU   schemer  r  hostnamer>  r}   replaceAttributeErrorr]  )r  parsedr  name
value_listr  r   valid_schemess           r=   	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%?!"0 M- 
-	-??$V__5F6N;; -F6N ;;4v-"76;;#7#?#?R#HIt ==H$)6F%& M 7CM?RST
 	
E ":. 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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dZd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.

    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.
        r<   )r  update)r  r  r  url_optionss       r=   from_urlzConnectionPool.from_url  s&    V  nk"}V}r4   Nr  r  c                     |xs d}t        |t              r|dk  rt        d      || _        || _        || _        g | _        t               | _        | j                  j                  dt              | _        y )Nl        r   z,"max_connections" must be a positive integerrq   )r   r}   r   r  connection_kwargsr  _available_connectionsset_in_use_connectionsr   r-   rq   )rD   r  r  r  s       r=   r   zConnectionPool.__init__  sp     *2U/3/?Q3FKLL 0!2.@B#<?E !3377Qr4   c           
          d| j                   j                   d| j                   j                   d | j                  di | j                  dS )Nr   r   r   r   r<   )r   r8   r7   r  r  r   s    r=   r   zConnectionPool.__repr__  sQ    ))*!DNN,C,C+D%%%?(>(>?B"F	
r4   c                 D    g | _         t        j                         | _        y rB   )r  r   WeakSetr  r   s    r=   resetzConnectionPool.reset  s    &(##*??#4 r4   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    r=   can_get_connectionz!ConnectionPool.can_get_connection  s1     '' D4++,t/C/CC	
r4   c                    K   | j                         }	 | j                  |       d{    |S 7 # t        $ r | j                  |       d{  7    w xY ww)z(Get a connected connection from the poolN)get_available_connectionensure_connectionr  release)rD   command_namekeysoptionsrE   s        r=   get_connectionzConnectionPool.get_connection$  s]     224
	((444
  5 	,,z***	s1   A1 /1 A1 AAAA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addrC   s     r=   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 settingsrk   ri   rl   rj   rm   F)rk   rl   rm   )r  rq   r   )rD   r  s     r=   get_encoderzConnectionPool.get_encoder:  sQ    ''!!ZZ
G4"JJ'8(C#ZZ(:EB " 
 	
r4   c                 :     | j                   di | j                  S )z=Create a new connection.  Can be overridden by child classes.r<   )r  r  r   s    r=   r  zConnectionPool.make_connectionC  s    $t$$>t'='=>>r4   rE   c                 f  K   |j                          d{    	 |j                          d{   rt        d      dy7 *7 # 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   rC   s     r=   r  z ConnectionPool.ensure_connectionG  s       """
	H44666%&;<$F 7 	# 7) 	H'')))$$&&&44666%&<=4G 7	Hsf   B1AB1A AA  B1A "B.'A*(B. BB.BB.+B1-B..B1c                 x   K   | j                   j                  |       | j                  j                  |       yw)z(Releases the connection back to the poolN)r  r   r  rv   rC   s     r=   r  zConnectionPool.releaseW  s0      	  ''
3##**:6s   8: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rB   r   )r   rE   s     r=   r   z,ConnectionPool.disconnect.<locals>.<genexpr>m  s     D*j##%Ds   return_exceptionsTNc              3   B   K   | ]  }t        |t              s|  y wrB   )r   r  )r   rs     r=   r   z,ConnectionPool.disconnect.<locals>.<genexpr>p  s     D!z!]'CADs   )r   r  r  rx   gathernext)rD   r  connectionsrespr   s        r=   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                          d{    y7 w)z-Close the pool, disconnecting all connectionsNr   r   s    r=   aclosezConnectionPool.acloset  r   r   c                 d    | j                   D ]	  }||_         | j                  D ]	  }||_         y rB   )r  rp   r  )rD   rp   conns      r=   	set_retryzConnectionPool.set_retryx  s:    // 	DDJ	,, 	DDJ	r4   r2  r1  )rp   r   r   N)r7   r8   r9   r3  classmethodr   r  r   r  r;  rH   r   r}   r   r   r  r5  r  r  r  r  r  r  r  r   r  r  r<   r4   r=   r  r    s     ,d3i ,c , , ,` 6@)-R12R "#R$
5
D 
		
?H2D H 7(: 7$ , r4   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 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  r<   )rD  r   rx   	Condition
_conditionr   )rD   r  r   r  r  r  r   s         r=   r   zBlockingConnectionPool.__init__  s@     	 	
-+	
  	

 "++-r4   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.)r  r   r   r   r  rD  r  rx   r'   r#   r  r  r  )rD   r  r  r  rE   errr   s         r=   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rE   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)r  rD  r  notify)rD   rE   r   s     r=   r  zBlockingConnectionPool.release  sc     ?? 	% 	%'/*---OO""$	% 	% 	%-	% 	% 	% 	%sT   A6AA6A!AA!
A6AA6A!A6!A3'A*(A3/A6)r7   r8   r9   r3  r;  rx   	LifoQueuer}   r   r   rH   Queuer   r  r  rZ  r[  s   @r=   r  r    sr    F  "!#5?+2+<+< # 12	
 '--( $%(: % %r4   r  )drx   ry   enumr   rw   rq  sysr8  r   abcr   	itertoolsr   typesr   typingr   r   r   r	   r
   r   r   r   r   r   r   r   r   urllib.parser   r   r   r   utilsr   version_infor   r   redis.asyncio.retryr   redis.backoffr   redis.connectionr   redis.credentialsr   r    redis.exceptionsr!   r"   r#   r$   r%   r&   r'   redis.typingr(   redis.utilsr)   r*   r+   _parsersr,   r-   r.   r/   r0   r  r  r  SYM_LFr  Enumr6   r;   ru   r  r>   r@   rK   r7  rH   r;  r]  ro  r  r  r5  r  r}   r  r  r   r:   r  r  r  r  r  r<   r4   r=   <module>r     s        
 
     "    B A ( z!06 % # 1 T   $ H H  
				   E+->@SSTU U'M%M>h >D8 D 02NNO w8 w8t6*# 6*r=,J =,@< <~!3 0 /htn  CS"'##!$%
C GC#v+)>$>? IU 13 1= 1h e+,x xvH%^ H%r4   