o
    Ch:                     @   s   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 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ZG dd de
ZG dd deZdS )    N)Optional)Redis)SentinelCommands)
ConnectionConnectionPoolSSLConnection)ConnectionErrorReadOnlyErrorResponseErrorTimeoutErrorc                   @      e Zd ZdS )MasterNotFoundErrorN__name__
__module____qualname__ r   r   `/var/www/html/myvaluetrips/my_value_trip_new/venv/lib/python3.10/site-packages/redis/sentinel.pyr          r   c                   @   r   )SlaveNotFoundErrorNr   r   r   r   r   r      r   r   c                       sf   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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   sR   | j }dt| j dt| j d|j d}| jr'd| j d| j }|| }|S )N<.	(service=z%s)>z,host=z,port=)r   typer   r   service_namehostport)r   pools	host_infor   r   r   __repr__   s   z"SentinelManagedConnection.__repr__c                 C   s"   |\| _ | _| j| jjdd 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              	   C   s\   | j rd S | jjr| | j  d S | j D ]}z| |W   S  ty+   Y qw tN)_sockr   	is_masterr/   get_master_addressrotate_slavesr   r   )r   slaver   r   r   _connect_retry0   s   z(SentinelManagedConnection._connect_retryc                 C   s   | j | jdd S )Nc                 S   s   d S r0   r   )errorr   r   r   <lambda>>   s    z3SentinelManagedConnection.connect.<locals>.<lambda>)retrycall_with_retryr6   r   r   r   r   connect=   s   z!SentinelManagedConnection.connectF)disconnect_on_errorpush_requestr=   r>   c                   s>   z
t  j|||dW S  ty   | jjr|   td w )N)disable_decodingr=   r>   z"The previous master is now a slave)r   read_responser	   r   r2   
disconnectr   )r   r?   r=   r>   r   r   r   r@   @   s   z'SentinelManagedConnection.read_response)F)r   r   r   r   r)   r/   r6   r<   r   boolr@   __classcell__r   r   r   r   r      s    r   c                   @   r   )SentinelManagedSSLConnectionNr   r   r   r   r   rD   Y   r   rD   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
SentinelConnectionPoolProxyc                 C   s0   t || _|| _|| _|| _|| _|   d S r0   )weakrefrefconnection_pool_refr2   r-   r#   sentinel_managerreset)r   r   r2   r-   r#   rI   r   r   r   r   ^   s   z$SentinelConnectionPoolProxy.__init__c                 C   s   d | _ d | _d S r0   )master_addressslave_rr_counterr;   r   r   r   rJ   m   s   
z!SentinelConnectionPoolProxy.resetc                 C   sD   | j | j}| jr | j|kr || _|  }|d ur |jdd |S )NF)inuse_connections)rI   discover_masterr#   r2   rK   rH   rA   )r   rK   r   r   r   r   r3   q   s   z.SentinelConnectionPoolProxy.get_master_addressc                 c   s    | j | j}|r5| jd u rtdt|d | _tt|D ]}| jd t| | _|| j }|V  q z|  V  W n	 t	yE   Y nw t
d| j)Nr      zNo slave found for )rI   discover_slavesr#   rL   randomrandintlenranger3   r   r   )r   slaves_r5   r   r   r   r4   |   s   

z)SentinelConnectionPoolProxy.rotate_slavesN)r   r   r   r   rJ   r3   r4   r   r   r   r   rE   ]   s
    rE   c                       s\   e Zd ZdZ fddZdd Z fddZedd	 Z fd
dZ	dd Z
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                    s   | d|ddrtnt|d< |dd| _|dd| _t| | j| j||d| _t j	d	i | | j| j
d< || _|| _d S )
Nconnection_classsslFr2   Tr-   )r   r2   r-   r#   rI   r   r   )getr   rD   r   r2   r-   rE   proxyr   r   connection_kwargsr#   rI   )r   r#   rI   r   r   r   r   r      s&   

zSentinelConnectionPool.__init__c              	   C   s:   | j rdnd}dt| j dt| j d| j d| d	S )Nmasterr5   r   r    r!   (z))>)r2   r"   r   r   r#   )r   roler   r   r   r)      s   zSentinelConnectionPool.__repr__c                    s   t    | j  d S r0   )r   rJ   r[   r;   r   r   r   rJ      s   
zSentinelConnectionPool.resetc                 C   s   | j jS r0   )r[   rK   r;   r   r   r   rK      s   z%SentinelConnectionPool.master_addressc                    s4   | j  p| j o| j|j|jfk}t }|o||S r0   )r2   rK   r$   r%   r   owns_connection)r   
connectioncheckparentr   r   r   r`      s
   z&SentinelConnectionPool.owns_connectionc                 C   
   | j  S r0   )r[   r3   r;   r   r   r   r3      s   
z)SentinelConnectionPool.get_master_addressc                 C   rd   )zRound-robin slave balancer)r[   r4   r;   r   r   r   r4      s   
z$SentinelConnectionPool.rotate_slaves)r   r   r   __doc__r   r)   rJ   propertyrK   r`   r3   r4   rC   r   r   r   r   rW      s    
rW   c                   @   sl   e Zd ZdZ			dddZdd Zdd	 Zd
d Zdd Zdd Z	dd Z
eefddZeefddZdS )Sentinelar  
    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)
    >>> master.set('foo', 'bar')
    >>> slave = sentinel.slave_for('mymaster', socket_timeout=0.1)
    >>> 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 ||fi  jqS r   )r   sentinel_kwargs)rj   hostnamer%   r;   r   r   
<listcomp>   s    z%Sentinel.__init__.<locals>.<listcomp>)itemsrn   	sentinelsmin_other_sentinelsr\   _force_master_ip)r   rr   rs   rn   force_master_ipr\   r   r;   r   r      s   


zSentinel.__init__c                 O   s   t |dd}t |dd}|r)t| jj|i |}|r#|gS |r'dS dS g }| jD ]}||j|i | q.|r@|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_responsesT)rB   r   rQ   choicerr   execute_commandappendall)r   argsr   rv   rw   response	responsessentinelr   r   r   ry      s   
zSentinel.execute_commandc                 C   sN   g }| j D ]}|d|jj qdt| j dt| j dd| dS )Nz{host}:{port}r   r    z(sentinels=[,z])>)	rr   rz   
format_mapr   r\   r"   r   r   join)r   sentinel_addressesr   r   r   r   r)     s   
zSentinel.__repr__c                 C   s2   |d r|d s|d rdS |d | j k rdS dS )Nr2   is_sdownis_odownFznum-other-sentinelsT)rs   )r   stater#   r   r   r   check_master_state&  s
   zSentinel.check_master_statec           
      C   s   t  }t| jD ]X\}}z| }W n ttfy0 } z|| d| W Y d}~qd}~ww ||}|r`| ||r`|| jd | jd< | j|< | j	durT| j	n|d }||d f  S qd}	t
|dkrqd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.
        z - Nr   ipr%    z : z, zNo master found for )list	enumeraterr   sentinel_mastersr   r   rz   rZ   r   rt   rS   r   r   )
r   r#   collected_errorssentinel_nor   masterser   r   
error_infor   r   r   rN   .  s0   

zSentinel.discover_masterc                 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%   )rz   )r   rU   slaves_aliver5   r   r   r   filter_slavesQ  s   zSentinel.filter_slavesc                 C   sN   | j D ]!}z||}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``)rr   sentinel_slavesr   r
   r   r   )r   r#   r   rU   r   r   r   rP   Z  s   

zSentinel.discover_slavesc                 K   4   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.
        Tr2   dictr\   update	from_poolr   r#   redis_classconnection_pool_classr   r\   r   r   r   
master_forf  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.
        Fr2   r   r   r   r   r   	slave_for  s   

zSentinel.slave_for)r   NN)r   r   r   re   r   ry   r)   r   rN   r   rP   r   rW   r   r   r   r   r   r   rg      s$    
#	
)rg   )rQ   rF   typingr   redis.clientr   redis.commandsr   redis.connectionr   r   r   redis.exceptionsr   r	   r
   r   r   r   r   rD   rE   rW   rg   r   r   r   r   <module>   s    A0=