
    Oǻi;7                     @   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 d dlmZ d dlmZmZmZ erd dlmZ 	 d dlZej:                  j=                  d	      Z e ed          d
kD  xs  e ed          d
k(  xr  e ed         dk\  Z!e!s e"d      	 	 d dl#Z#dZ$	 d dl%Z%dZ&d dl'm(Z( de)deddfdZ*ed        Z+dee)e,f   de)fdZ-d Z.dee)ef   dee)ef   fdZ/d Z0d Z1dGdZ2dHdZ3	 	 	 dGd ee4e)f   d!e)d"e)d#e)d$e f
d%Z5 ed&e'      Z6	 dId(ed)e7de8d*eee)      de8f
d+Z9	 	 	 	 dJd,eee)      d*eee)      d"e)d#e)dee6ge6f   f
d-Z:d. Z;	 dKd/eee)e f      d0e de<fd1Z=d2 Z>d3e)d4e?de)fd5Z@d6e)d7e)de fd8ZAd9 ZB	 	 	 	 dLd:ee   d;ee   d<ee   d=ee   dee   f
d>ZCdMd?ZDd@ ZEdA ZFdB ZGdNdCZHdee6ge6f   fdDZI	 dNd ee4e)f   d!e)d$e fdEZJ	 dId,eee)      dee6ge6f   fdFZKy# e"$ r dZ!Y Tw xY w# e"$ r dZ$Y [w xY w# e"$ r dZ&Y bw xY w)O    N)Callable)contextmanager)wraps)TYPE_CHECKINGAnyDictListMappingOptionalTypeVarUnion)	DataError)
AbsExpiryT
EncodableTExpiryTRedis.         z"hiredis package should be >= 3.2.0FT)metadataurlkwargsreturnr   c                 4    ddl m}  |j                  | fi |S )z
    Returns an active Redis client generated from the given database URL.

    Will attempt to extract the database id from the path url fragment, if
    none is provided.
    r   r   )redis.clientr   from_url)r   r   r   s      I/opt/lhia/marcimex/agent/venv/lib/python3.12/site-packages/redis/utils.pyr   r   /   s     #5>>#(((    c              #   T   K   | j                         }| |j                          y wN)pipelineexecute)	redis_objps     r   r#   r#   ;   s"     A
GIIKs   &(valuec                 L    t        | t              r| j                  dd      S | S )Nutf-8replace)errors)
isinstancebytesdecoder'   s    r   str_if_bytesr0   B   s(    3=eU3KWY/QVr    c                 *    t        t        |             S r"   )strr0   r/   s    r   safe_strr3   H   s    |E"##r    dictsc                  :    i }| D ]  }|j                  |        |S )za
    Merge all provided dicts into 1 dict.
    *dicts : `dict`
        dictionaries to merge
    )update)r4   mergedds      r   
dict_merger9   L   s+     F a Mr    c                 .    t         j                  | |      S r"   )dictfromkeys)key_listcallbacks     r   list_keys_to_dictr?   Z   s    ==8,,r    c                     t               }|j                         D ]  }|D ]  }|j                  |         t        |      S )z
    Merge all items in `res` into a list.

    This command is used when sending a command to multiple nodes
    and the result from each node should be merged into a single list.

    res : 'dict'
    )setvaluesaddlist)commandresresultvr'   s        r   merge_resultrI   ^   sG     UFZZ\  	EJJu	 <r    c                 v    dd l }d|  d}|r	|d| dz  }|r	|d| dz  } |j                  |t        |       y )Nr   zCall to deprecated r    () -- Deprecated since version category
stacklevelwarningswarnDeprecationWarning)namereasonversionrP   rR   msgs         r   warn_deprecatedrY   p   sR    vQ
'CF81~.wiq99HMM# 2zJr    c                       fd}|S )z5
    Decorator to mark a function as deprecated.
    c                      t        j                         rt                fd       }|S t                fd       }|S )Nc                  j   K   t        xs j                  d        | i | d {   S 7 wNr   rP   rY   __name__argsr   funcrU   rV   rW   s     r   async_wrapperz=deprecated_function.<locals>.decorator.<locals>.async_wrapper   s5      5vwSTU!4262222s   )313c                  N    t        xs j                  d        | i |S r]   r_   ra   s     r   wrapperz7deprecated_function.<locals>.decorator.<locals>.wrapper   s+     5vwSTUT,V,,r    inspectiscoroutinefunctionr   )rc   rd   rf   rU   rV   rW   s   `  r   	decoratorz&deprecated_function.<locals>.decorator   sN    &&t,4[3 3 !  4[- - Nr     )rV   rW   rU   rj   s   ``` r   deprecated_functionrl   {   s    
$ r    arg_namefunction_namerV   rW   rP   c                 |    dd l }d| d|  d}|r	|d| dz  }|r	|d| dz  } |j                  |t        |	       y )
Nr   	Call to 'z6' function with deprecated usage of input argument/s ''.rK   rL   rM   r   rN   rQ   )rm   rn   rV   rW   rP   rR   rX   s          r   warn_deprecated_arg_usagerr      sg      M? #''/j	4  F81~.wiq99HMM# 2zJr    C)boundrc   rb   allowed_argsc                    | j                   j                  d| j                   j                   }t        t	        ||            }|j                  |       |j                  dd       |r|D ]  }|j                  |d        |S )z
    Extract arguments from function call that should be checked for deprecation/experimental warnings.
    Excludes 'self' and any explicitly allowed args.
    Nself)__code__co_varnamesco_argcountr;   zipr6   pop)rc   rb   r   ru   	arg_namesfilterable_argsallowed_args          r   _get_filterable_argsr      s|     ))*EDMM,E,EFI3y$/0O6"%' 	3KT2	3r    args_to_warnc                 V      dg g  fddt         dt         ffd}|S )z
    Decorator to mark specified args of a function as deprecated.
    If '*' is in args_to_warn, all arguments will be marked as deprecated.
    *c                     D ]f  }|dk(  r@t        |      dkD  r2t        t        |j                               | j                  d       H||v sMt        || j                  d       h y)z*Check and warn about deprecated arguments.r   r      r^   N)lenrr   rD   keysr`   )rc   r~   argr   rV   rW   s      r   _check_deprecated_argsz/deprecated_args.<locals>._check_deprecated_args   so     	Cczc/2Q6)--/0MM  ')A	r    rc   r   c                      t        j                         rt                fd       }|S t                fd       }|S )Nc                  ^   K   t        | |      } |        | i | d {   S 7 wr"   r   rb   r   r~   r   ru   rc   s      r   rd   z9deprecated_args.<locals>.decorator.<locals>.async_wrapper   s7     "6tT6<"X&t_=!4262222   #-+-c                  B    t        | |      } |        | i |S r"   r   r   s      r   rf   z3deprecated_args.<locals>.decorator.<locals>.wrapper   s-    "6tT6<"X&t_=T,V,,r    rg   )rc   rd   rf   r   ru   s   `  r   rj   z"deprecated_args.<locals>.decorator   sN    &&t,4[3 3
 !  4[- -
 Nr    rs   )r   ru   rV   rW   rj   r   s   ```` @r   deprecated_argsr      s=     u  a ( r    c                  ^   dt         j                  j                  j                  j	                         vryt        j
                  d      } | j                  t         j                         t        j                         }|j                  t         j                         | j                  |       yy)zo
    Set up a logger that log info logs to stdout.
    (This is used by the default push response handler)
    push_responseN)
loggingrootmanager
loggerDictr   	getLoggersetLevelINFOStreamHandler
addHandler)loggerhandlers     r   _set_info_loggerr      sy    
 gll22==BBDD""?3%'')&'" Er    protocolexpected_versionc                 r    | yt        | t              r	 t        |       } | |k(  S | |k(  S # t        $ r Y yw xY w)NF)r,   r2   int
ValueError)r   r   s     r   check_protocol_versionr      sR     (C 	8}H '''8'''  		s   * 	66c                  j    	 t        j                  d      } | S # t         j                  $ r d} Y | S w xY w)Nredisz99.99.99)r   rW   PackageNotFoundError)libvers    r   get_lib_versionr     s?    !!'* M (( Ms    22
host_error	exceptionc                     |j                   sd|  dS t        |j                         dk(  rd|j                   d    d|  dS d|j                   d    d|  d|j                   d    dS )NzError connecting to r   r   zError r   z connecting to z. )rb   r   )r   r   s     r   format_error_messager     s~    >>%j\33	Y^^		!	q)*/*QGG Y^^A&'zl"~~a !$	
r    version1version2c                 <   t        t        t        | j                  d                  }t        t        t        |j                  d                  }t	        |      t	        |      kD  r9t	        |      t	        |      z
  }t        |      D ]  }|j                  d        nOt	        |      t	        |      k  r8t	        |      t	        |      z
  }t        |      D ]  }|j                  d        t        |      D ]  \  }}||   ||   kD  r y||   ||   k  s y y)z
    Compare two versions.

    :return: -1 if version1 > version2
             0 if both versions are equal
             1 if version1 < version2
    r   r   r   )rD   mapr   splitr   rangeappend	enumerate)r   r   num_versions1num_versions2diff_ivers           r   compare_versionsr   !  s    S(.."567MS(.."567M
=C..=!C$66t 	$A  #	$	]	c-0	0=!C$66t 	$A  #	$ M* 3mA..1a 00	 r    c                 ~    t        | t              r| j                  d      S t        | t              r| S t	        d      )Nr)   z$Key must be either a string or bytes)r,   r-   r.   r2   	TypeError)keys    r   ensure_stringr   ?  s6    #uzz'""	C	
>??r    expxexatpxatc                    g }| |j                  d       t        | t        j                        r*|j                  t	        | j                                      |S t        | t              r|j                  |        |S t        | t              r,| j                         r|j                  t	        |              |S t        d      ||j                  d       t        |t        j                        r-|j                  t	        |j                         dz               |S t        |t              r|j                  |       |S t        d      |Ht        |t        j                        rt	        |j                               }|j                  d|g       |S |It        |t        j                        rt	        |j                         dz        }|j                  d|g       |S )NEXz$ex must be datetime.timedelta or intPXi  z$px must be datetime.timedelta or intEXATPXAT)r   r,   datetime	timedeltar   total_secondsr2   isdigitr   	timestampextend)r   r   r   r   exp_optionss        r   extract_expire_flagsr   H  s    %'K	~4 b(,,-s2#3#3#5670 / C r", + C RZZ\s2w'( % BCC	4 b(,,-s2#3#3#5#<=>  C r"  BCC		dH--.t~~'(DFD>*  
	dH--.t~~'$./DFD>*r    c                 4    t        j                  | |dd      S )Nz...T)textwidthplaceholderbreak_long_words)textwrapshorten)txt
max_lengths     r   truncate_textr   m  s    
 r    c                       y)zT
    Fake function for a Retry object if you don't need to handle each failure.
    Nrk   rk   r    r   
dummy_failr   s  s     	r    c                     K   yw)zZ
    Async fake function for a Retry object if you don't need to handle each failure.
    Nrk   rk   r    r   dummy_fail_asyncr   z  s      	s   c                 X      j                   t               fd       }| _          S )z4
    Decorator to mark a class as experimental.
    c                 p    t        j                  j                   dt        d        | g|i | y )NzD is an experimental and may change or be removed in future versions.r   rN   )rR   rS   r`   UserWarning)rw   rb   r   clsoriginal_inits      r   new_initzexperimental.<locals>.new_init  s8    ||n`a 	

 	d,T,V,r    )__init__r   )r   r   r   s   ` @r   experimentalr     s3     LLM
=- - CLJr    c                 J    dd l }d|  d} |j                  |t        |       y )Nr   zCall to experimental method zS. Be aware that the function arguments can change or be removed in future versions.rN   rR   rS   r   )rU   rP   rR   rX   s       r   warn_experimentalr     s1     'tf -3 	3 
 HMM#
Cr    c                  &    dt         dt         fd} | S )z7
    Decorator to mark a function as experimental.
    rc   r   c                      t        j                         rt                fd       }|S t                fd       }|S )Nc                  ^   K   t        j                  d        | i | d {   S 7 wNr   r^   r   r`   rb   r   rc   s     r   rd   z=experimental_method.<locals>.decorator.<locals>.async_wrapper  s+     !$--A>!4262222r   c                  B    t        j                  d        | i |S r   r   r   s     r   rf   z7experimental_method.<locals>.decorator.<locals>.wrapper  s!    !$--A>T,V,,r    rg   )rc   rd   rf   s   `  r   rj   z&experimental_method.<locals>.decorator  sN    &&t,4[3 3 !  4[- - Nr    r   )rj   s    r   experimental_methodr     s    
 a $ r    c                 P    dd l }d| d|  d} |j                  |t        |       y )Nr   rp   z6' method with experimental usage of input argument/s 'rq   rN   r   )rm   rn   rP   rR   rX   s        r   warn_experimental_arg_usager     s:    
  M? #''/j	4  HMM#
Cr    c                 B      dg  fddt         dt         ffd}|S )z
    Decorator to mark specified args of a function as experimental.
    If '*' is in args_to_warn, all arguments will be marked as experimental.
    r   c                     D ]b  }|dk(  r>t        |      dkD  r0t        t        |j                               | j                  d       F||v sKt        || j                  d       d y)z,Check and warn about experimental arguments.r   r      r^   N)r   r   rD   r   r`   )rc   r~   r   r   s      r   _check_experimental_argsz3experimental_args.<locals>._check_experimental_args  sb     	NCczc/2Q6+--/0$--A '+C1M	Nr    rc   r   c                      t        j                         rt                fd       }|S t                fd       }|S )Nc                  x   K   t        | |      }t        |      dkD  r	 |        | i | d {   S 7 wNr   r   r   rb   r   r~   r   rc   s      r   rd   z;experimental_args.<locals>.decorator.<locals>.async_wrapper  sB     "6tT6"J'!+,T?C!4262222s   0:8:c                  \    t        | |      }t        |      dkD  r	 |        | i |S r   r   r   s      r   rf   z5experimental_args.<locals>.decorator.<locals>.wrapper  s8    "6tT6"J'!+,T?CT,V,,r    rg   )rc   rd   rf   r   s   `  r   rj   z$experimental_args.<locals>.decorator  sN    &&t,4[3 3 !  4[- - Nr    r   )r   rj   r   s   ` @r   experimental_argsr    s3     uN a , r    ) r  r   )r  r  Nr"   )NNr  r  )r   )NNNN)d   )r   )Lr   rh   r   r   rR   collections.abcr   
contextlibr   	functoolsr   typingr   r   r   r	   r
   r   r   r   redis.exceptionsr   redis.typingr   r   r   r   r   hiredis__version__r   hiredis_versionr   HIREDIS_AVAILABLEImportErrorsslSSL_AVAILABLEcryptographyCRYPTOGRAPHY_AVAILABLE	importlibr   r2   r   r#   r-   r0   r3   r9   r?   rI   rY   rl   rD   rr   rs   tupler;   r   r   r   boolr   r   BaseExceptionr   r   r   r   r   r   r   r   r   r   r   r  rk   r    r   <module>r     s        $ %  T T T & 8 8" ))//4OOA./!3 OA1$E_Q-?)@A)E  >?? 
M#! 	)# 	) 	) 	)  c5j) c $wsCx( T#s(^ -$K: KD#IKK K 	K
 K( Cx  TX
)-=Ed3i=P	$ )-(,	349%349%3 3 	3
 qc1f3l
# BC
(uS#X'
(;>
(	
(	
S 	
] 	
s 	
s c c <@ ! !%!%	""" :
" :
	"
 
*"J		&DXqc1f- : DD#IDD D )-*49%*qc1f*Y    M  #"#s7   AG6 -H 4H 6H HHHHH