o
    Ch:                     @   s   d dl Z d dlZd dlZd dl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mZ d dlmZ d dlmZmZmZmZ G dd deZG d	d
 d
eZG dd deZG dd deeZG dd deZG dd deZdS )    N)AsyncIteratorIterableMappingOptionalSequenceTupleType)Redis)
ConnectionConnectionPool
EncodableTSSLConnection)AsyncSentinelCommands)ConnectionErrorReadOnlyErrorResponseErrorTimeoutErrorc                   @      e Zd ZdS )MasterNotFoundErrorN__name__
__module____qualname__ r   r   h/var/www/html/myvaluetrips/my_value_trip_new/venv/lib/python3.10/site-packages/redis/asyncio/sentinel.pyr          r   c                   @   r   )SlaveNotFoundErrorNr   r   r   r   r   r      r   r   c                       st   e Zd Z fddZdd Zdd Zdd Zd	d
 Z		dddddede	e
 de	e
 de	e f fddZ  ZS )SentinelManagedConnectionc                    s"   | d| _t jdi | d S )Nconnection_poolr   )popr   super__init__)selfkwargs	__class__r   r   r!      s   z"SentinelManagedConnection.__init__c                 C   sB   d| j j d| j j }| jrd| j d| j }||7 }|d S )N<.z,host=z,port=z)>)r%   r   r   hostport)r"   s	host_infor   r   r   __repr__#   s
   z"SentinelManagedConnection.__repr__c                    s*   |\| _ | _| j| jjddI d H  d S )NF)check_healthretry_socket_connect)r(   r)   connect_check_healthr   check_connection)r"   addressr   r   r   
connect_to*   s   z$SentinelManagedConnection.connect_toc              	      sz   | j rd S | jjr| | j I d H I d H  d S | j 2 z3 d H W }z| |I d H W   S  ty9   Y qw 6 tN)_readerr   	is_masterr2   get_master_addressrotate_slavesr   r   )r"   slaver   r   r   _connect_retry1   s    z(SentinelManagedConnection._connect_retryc                    s   | j | jdd I d H S )Nc                 S   s
   t dS )Nr   )asynciosleep)errorr   r   r   <lambda>A   s   
 z3SentinelManagedConnection.connect.<locals>.<lambda>)retrycall_with_retryr9   r"   r   r   r   connect>   s
   
z!SentinelManagedConnection.connectFNT)disconnect_on_errorpush_requestdisable_decodingtimeoutrB   rC   c                   sN   zt  j||||dI d H W S  ty&   | jjr%|  I d H  td w )N)rD   rE   rB   rC   z"The previous master is now a slave)r    read_responser   r   r5   
disconnectr   )r"   rD   rE   rB   rC   r$   r   r   rF   D   s   z'SentinelManagedConnection.read_response)FN)r   r   r   r!   r,   r2   r9   rA   boolr   floatrF   __classcell__r   r   r$   r   r      s(    r   c                   @   r   )SentinelManagedSSLConnectionNr   r   r   r   r   rK   _   r   rK   c                       s\   e Zd ZdZ fddZdd Z fddZdef fd	d
Zdd Z	de
fddZ  ZS )SentinelConnectionPoolz
    Sentinel backed connection pool.

    If ``check_connection`` flag is set to True, SentinelManagedConnection
    sends a PING command right after establishing the connection.
    c                    sz   | d|ddrtnt|d< |dd| _|dd| _t jdi | t	| | j
d< || _|| _d | _d | _d S )	Nconnection_classsslFr5   Tr0   r   r   )getr   rK   r   r5   r0   r    r!   weakrefproxyconnection_kwargsservice_namesentinel_managermaster_addressslave_rr_counter)r"   rS   rT   r#   r$   r   r   r!   k   s   

zSentinelConnectionPool.__init__c              	   C   s2   d| j j d| j j d| j d| jrdpd d	S )Nr&   r'   z	(service=(masterr8   z))>)r%   r   r   rS   r5   r@   r   r   r   r,   }   s   zSentinelConnectionPool.__repr__c                    s   t    d | _d | _d S r3   )r    resetrU   rV   r@   r$   r   r   rY      s   

zSentinelConnectionPool.reset
connectionc                    s0   | j  p| j o| j|j|jfk}|ot |S r3   )r5   rU   r(   r)   r    owns_connection)r"   rZ   checkr$   r   r   r[      s   z&SentinelConnectionPool.owns_connectionc                    sB   | j | jI d H }| jr| j|kr|| _| jddI d H  |S )NF)inuse_connections)rT   discover_masterrS   r5   rU   rG   )r"   rU   r   r   r   r6      s   
z)SentinelConnectionPool.get_master_addressreturnc                 C  s   | j | jI dH }|r8| jdu rtdt|d | _tt|D ]}| jd t| | _|| j }|V  q#z
|  I dH V  W n	 t	yK   Y nw t
d| j)zRound-robin slave balancerNr      zNo slave found for )rT   discover_slavesrS   rV   randomrandintlenranger6   r   r   )r"   slaves_r8   r   r   r   r7      s   

z$SentinelConnectionPool.rotate_slaves)r   r   r   __doc__r!   r,   rY   r
   r[   r6   r   r7   rJ   r   r   r$   r   rL   c   s    
rL   c                   @   s   e Zd ZdZ			dddZdd Zdd	 Zd
edede	fddZ
defddZdee deeeef  fddZdedeeeef  fddZeefdedee dee fddZeefdedee dee fddZdS )Sentinela~  
    Redis Sentinel cluster client

    >>> from redis.sentinel import Sentinel
    >>> sentinel = Sentinel([('localhost', 26379)], socket_timeout=0.1)
    >>> master = sentinel.master_for('mymaster', socket_timeout=0.1)
    >>> await master.set('foo', 'bar')
    >>> slave = sentinel.slave_for('mymaster', socket_timeout=0.1)
    >>> await slave.get('foo')
    b'bar'

    ``sentinels`` is a list of sentinel nodes. Each node is represented by
    a pair (hostname, port).

    ``min_other_sentinels`` defined a minimum number of peers for a sentinel.
    When querying a sentinel, if it doesn't meet this threshold, responses
    from that sentinel won't be considered valid.

    ``sentinel_kwargs`` is a dictionary of connection arguments used when
    connecting to sentinel instances. Any argument that can be passed to
    a normal Redis connection can be specified here. If ``sentinel_kwargs`` is
    not specified, any socket_timeout and socket_keepalive options specified
    in ``connection_kwargs`` will be used.

    ``connection_kwargs`` are keyword arguments that will be used when
    establishing a connection to a Redis server.
    r   Nc                    sJ   |d u rdd |  D }| _ fdd|D  _| _| _| _d S )Nc                 S   s    i | ]\}}| d r||qS )socket_)
startswith).0kvr   r   r   
<dictcomp>   s
    z%Sentinel.__init__.<locals>.<dictcomp>c                    s&   g | ]\}}t d||d  jqS ))r(   r)   r   )r	   sentinel_kwargs)rl   hostnamer)   r@   r   r   
<listcomp>   s    z%Sentinel.__init__.<locals>.<listcomp>)itemsrp   	sentinelsmin_other_sentinelsrR   _force_master_ip)r"   rt   ru   rp   force_master_iprR   r   r@   r   r!      s   


zSentinel.__init__c                    s   t dd}t dd}|r-t| jj i I dH }|r'|gS |r+dS dS  fdd| jD }tj| I dH }|rD|S t|S )z
        Execute Sentinel command in sentinel nodes.
        once - If set to True, then execute the resulting command on a single
               node at random, rather than across the entire sentinel cluster.
        onceFreturn_responsesNTc                    s"   g | ]}t |j i qS r   )r:   Taskexecute_command)rl   sentinelargsr#   r   r   rr      s    z,Sentinel.execute_command.<locals>.<listcomp>)	rH   r   rb   choicert   r{   r:   gatherall)r"   r~   r#   rx   ry   responsetasks	responsesr   r}   r   r{      s&   zSentinel.execute_commandc                 C   sX   g }| j D ]}||jjd  d|jjd   qd| j d| jj dd| dS )	Nr(   :r)   r&   r'   z(sentinels=[,z])>)rt   appendr   rR   r%   r   join)r"   sentinel_addressesr|   r   r   r   r,      s   

zSentinel.__repr__staterS   r_   c                 C   s2   |d r|d s|d rdS |d | j k rdS dS )Nr5   is_sdownis_odownFznum-other-sentinelsT)ru   )r"   r   rS   r   r   r   check_master_state  s
   zSentinel.check_master_statec           
         s   t  }t| jD ][\}}z	| I dH }W n ttfy4 } z|| d| W Y d}~q	d}~ww ||}|rd| ||rd|| jd | jd< | j|< | j	durX| j	n|d }||d f  S q	d}	t
|dkrudd| }	td	||	 )
z
        Asks sentinel servers for the Redis master's address corresponding
        to the service labeled ``service_name``.

        Returns a pair (address, port) or raises MasterNotFoundError if no
        master is found.
        Nz - r   ipr)    z : z, zNo master found for )list	enumeratert   sentinel_mastersr   r   r   rO   r   rv   rd   r   r   )
r"   rS   collected_errorssentinel_nor|   masterser   r   
error_infor   r   r   r^     s2   

zSentinel.discover_masterrf   c                 C   s:   g }|D ]}|d s|d rq| |d |d f q|S )z1Remove slaves that are in an ODOWN or SDOWN stater   r   r   r)   )r   )r"   rf   slaves_aliver8   r   r   r   filter_slaves6  s   zSentinel.filter_slavesc                    sV   | j D ]$}z
||I dH }W n tttfy   Y qw | |}|r(|  S qg S )z;Returns a list of alive slaves for service ``service_name``N)rt   sentinel_slavesr   r   r   r   )r"   rS   r|   rf   r   r   r   ra   A  s   

zSentinel.discover_slavesredis_classconnection_pool_classc                 K   8   d|d< t | j}|| ||| fi |}||S )a  
        Returns a redis client instance for the ``service_name`` master.
        Sentinel client will detect failover and reconnect Redis clients
        automatically.

        A :py:class:`~redis.sentinel.SentinelConnectionPool` class is
        used to retrieve the master's address before establishing a new
        connection.

        NOTE: If the master's address has changed, any cached connections to
        the old master are closed.

        By default clients will be a :py:class:`~redis.Redis` instance.
        Specify a different class to the ``redis_class`` argument if you
        desire something different.

        The ``connection_pool_class`` specifies the connection pool to
        use.  The :py:class:`~redis.sentinel.SentinelConnectionPool`
        will be used by default.

        All other keyword arguments are merged with any connection_kwargs
        passed to this class and passed to the connection pool as keyword
        arguments to be used to initialize Redis connections.
        Tr5   dictrR   update	from_poolr"   rS   r   r   r#   rR   r   r   r   r   
master_forO  s
   


zSentinel.master_forc                 K   r   )a  
        Returns redis client instance for the ``service_name`` slave(s).

        A SentinelConnectionPool class is used to retrieve the slave's
        address before establishing a new connection.

        By default clients will be a :py:class:`~redis.Redis` instance.
        Specify a different class to the ``redis_class`` argument if you
        desire something different.

        The ``connection_pool_class`` specifies the connection pool to use.
        The SentinelConnectionPool will be used by default.

        All other keyword arguments are merged with any connection_kwargs
        passed to this class and passed to the connection pool as keyword
        arguments to be used to initialize Redis connections.
        Fr5   r   r   r   r   r   	slave_forv  s
   


zSentinel.slave_for)r   NN)r   r   r   rh   r!   r{   r,   r   strrH   r   r^   r   r   r   r   r   r   ra   r	   rL   r   r   r   r   r   r   r   ri      sL    
 #


*ri   )r:   rb   rP   typingr   r   r   r   r   r   r   redis.asyncio.clientr	   redis.asyncio.connectionr
   r   r   r   redis.commandsr   redis.exceptionsr   r   r   r   r   r   r   rK   rL   ri   r   r   r   r   <module>   s    $AG