
    jj                         d dl Z d dlmZ d dlmZ d dlmZmZmZm	Z	 	 d dl
Z
 ee
j                            d          d                    dk    Zes ed          n# e$ r dZY nw xY w	 d dlZd	Zn# e$ r dZY nw xY w	 d dlZd	Zn# e$ r dZY nw xY wd d
lmZ 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 Zd Zd"dZd#dZ d Z!d Z"dede#defdZ$dededefd Z%d! Z&dS )$    N)contextmanagerwraps)AnyDictMappingUnion.   z"hiredis package should be >= 3.0.0FT)metadatac                 *    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   )Redis)redis.clientr   from_url)urlkwargsr   s      [/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/redis/utils.pyr   r   !   s0     #"""""5>#(((((    c              #   b   K   |                                  }|V  |                                 d S N)pipelineexecute)	redis_objps     r   r   r   -   s1      A
GGGIIKKKKKr   valuereturnc                 ^    t          | t                    r|                     dd          n| S )Nutf-8replace)errors)
isinstancebytesdecoder   s    r   str_if_bytesr%   4   s/    3=eU3K3KVWY///QVr   c                 :    t          t          |                     S r   )strr%   r$   s    r   safe_strr(   :   s    |E""###r   dictsc                  >    i }| D ]}|                     |           |S )za
    Merge all provided dicts into 1 dict.
    *dicts : `dict`
        dictionaries to merge
    )update)r)   mergedds      r   
dict_merger.   >   s2     F  aMr   c                 8    t                               | |          S r   )dictfromkeys)key_listcallbacks     r   list_keys_to_dictr4   L   s    ==8,,,r   c                     t                      }|                                D ]}|D ]}|                    |           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_resultr>   P   s[     UUFZZ\\   	 	EJJu	 <<r       c                     dd l }d|  d}|r	|d| dz  }|r	|d| dz  }|                    |t          |           d S )Nr   zCall to deprecated r
   z ()z -- Deprecated since version )category
stacklevel)warningswarnDeprecationWarning)namereasonversionrD   rE   msgs         r   warn_deprecatedrL   b   sr    OOO
'
'
'
'C ~F~~~ :9w9999MM# 2zMJJJJJr   c                       fd}|S )z5
    Decorator to mark a function as deprecated.
    c                 D     t                      fd            }|S )Nc                  H    t          pj        d            | i |S )Nr   )rD   )rL   __name__)argsr   funcrH   rI   rJ   s     r   wrapperz7deprecated_function.<locals>.decorator.<locals>.wrappers   s6    D1DM67qQQQQ4((((r   r   )rR   rS   rH   rI   rJ   s   ` r   	decoratorz&deprecated_function.<locals>.decoratorr   sC    	t	) 	) 	) 	) 	) 	) 	) 
	) r    )rI   rJ   rH   rT   s   ``` r   deprecated_functionrV   m   s0    
       r   c                  R   dt           j        j        j                                        vr|t          j        d          } |                     t           j                   t          j                    }|                    t           j                   | 	                    |           dS dS )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
loggerDictkeys	getLoggersetLevelINFOStreamHandler
addHandler)loggerhandlers     r   _set_info_loggerre   }   s    
 gl2=BBDDDD"?33%%%'))&&&'""""" EDr   c                  `    	 t          j        d          } n# t           j        $ r d} Y nw xY w| S )Nredisz99.99.99)r   rJ   PackageNotFoundError)libvers    r   get_lib_versionrj      sE    !'**(   Ms    ++
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
      zError r   z connecting to z. )rQ   len)rk   rl   s     r   format_error_messagerp      s    > 
3j3333	Y^			!	!G	q)GG*GGGG$Y^A& $ $z $ $~a $ $ $	
r   version1version2c                    t          t          t          |                     d                              }t          t          t          |                    d                              }t	          |          t	          |          k    rGt	          |          t	          |          z
  }t          |          D ]}|                    d           nft	          |          t	          |          k     rFt	          |          t	          |          z
  }t          |          D ]}|                    d           t          |          D ]/\  }}||         ||         k    r dS ||         ||         k     r dS 0dS )z
    Compare two versions.

    :return: -1 if version1 > version2
             0 if both versions are equal
             1 if version1 < version2
    r
   r   rn   )r9   mapintsplitro   rangeappend	enumerate)rq   rr   num_versions1num_versions2diff_ivers           r   compare_versionsr      sp    S(.."5"56677MS(.."5"56677M
=C....=!!C$6$66t 	$ 	$A  ####	$	]		c-00	0	0=!!C$6$66t 	$ 	$A  ####M**  3mA...221a 00011 1 1r   c                     t          | t                    r|                     d          S t          | t                    r| S t	          d          )Nr   z$Key must be either a string or bytes)r!   r"   r#   r'   	TypeError)keys    r   ensure_stringr      sN    #u @zz'"""	C		 @
>???r   )r?   r?   r@   )r?   r?   N)'rY   
contextlibr   	functoolsr   typingr   r   r   r	   hiredisrv   __version__rw   HIREDIS_AVAILABLEImportErrorsslSSL_AVAILABLEcryptographyCRYPTOGRAPHY_AVAILABLE	importlibr   r   r   r'   r"   r%   r(   r.   r4   r>   rL   rV   re   rj   BaseExceptionrp   r   r   rU   r   r   <module>r      s    % % % % % %       , , , , , , , , , , , ,NNN G/55c::1=>>!C @k>???@   JJJMM   MMM#! # # #"#      	) 	) 	)   c5j) c    $ $ $wsCx( T#s(^    - - -  $K K K K    
# 
# 
#  	
S 	
] 	
s 	
 	
 	
 	
s c c    <@ @ @ @ @s5   >A A'&A'+A2 2A<;A< B BB