File: //home/arjun/projects/env/lib/python3.10/site-packages/redis/__pycache__/sentinel.cpython-310.pyc
o
$we_7 � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z m
Z
m Z d dl m
Z
mZmZmZ d dlmZ G dd � d e
�ZG d
d � d e
�ZG d d
� d
e �ZG dd� dee �ZG dd� d�ZG dd� de
�ZG dd� de�ZdS )� N)�Optional)�Redis)�SentinelCommands)�
Connection�ConnectionPool�
SSLConnection)�ConnectionError�
ReadOnlyError�
ResponseError� TimeoutError)� str_if_bytesc @ � e Zd ZdS )�MasterNotFoundErrorN��__name__�
__module__� __qualname__� r r �G/home/arjun/projects/env/lib/python3.10/site-packages/redis/sentinel.pyr � r c @ r
)�SlaveNotFoundErrorNr r r r r r r r c sj e Zd Z� fdd�Zdd� Z� fdd�Zdd� Zd d
� Z dd d d �d
ee dee f� fdd�Z
� Z S )�SentinelManagedConnectionc s" |� d�| _t� jdi |�� d S )N�connection_poolr )�popr �super�__init__)�self�kwargs�� __class__r r r s z"SentinelManagedConnection.__init__c C sD | j }t| �j� d|j� d�}| jr d| j� d| j� �}|| }|S )N� <service=z%s>z,host=z,port=)r �typer � service_name�host�port)r �pool�s� host_infor r r �__repr__ s z"SentinelManagedConnection.__repr__c sH |\| _ | _t� �� | jjr | �d� t| �� �dkr"t d��d S d S )N�PING�PONGz PING failed)
r# r$ r �connectr �check_connection� send_commandr �
read_responser )r �addressr r r �
connect_to! s
�z$SentinelManagedConnection.connect_toc C s\ | j rd S | jjr| �| j�� � d S | j�� D ]}z| �|�W S ty+ Y qw t��N)�_sockr � is_masterr0 �get_master_address�
rotate_slavesr r )r �slaver r r �_connect_retry) s �z(SentinelManagedConnection._connect_retryc C s | j �| jdd� �S )Nc S s d S r1 r )�errorr r r �<lambda>7 s z3SentinelManagedConnection.connect.<locals>.<lambda>)�retry�call_with_retryr7 �r r r r r+ 6 s z!SentinelManagedConnection.connectF)�disconnect_on_error� push_requestr= r> c s> z
t � j|||d�W S ty | jjr | �� td��� w )N)�disable_decodingr= r> z"The previous master is now a slave)r r. r r r3 �
disconnectr )r r? r= r> r r r r. 9 s � �z'SentinelManagedConnection.read_response)F) r r r r r( r0 r7 r+ r �boolr. �
__classcell__r r r r r s
����r c @ r
)� SentinelManagedSSLConnectionNr r r r r rC R r rC c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SentinelConnectionPoolProxyc C s0 t �|�| _|| _|| _|| _|| _| �� d S r1 )�weakref�ref�connection_pool_refr3 r, r" �sentinel_manager�reset)r r r3 r, r" rH r r r r W s z$SentinelConnectionPoolProxy.__init__c C s d | _ d | _d S r1 )�master_address�slave_rr_counterr< r r r rI f s
z!SentinelConnectionPoolProxy.resetc C sD | j �| j�}| jr | j|kr || _| �� }|d ur |jdd� |S )NF)�inuse_connections)rH �discover_masterr" r3 rJ rG r@ )r rJ r r r r r4 j s z.SentinelConnectionPoolProxy.get_master_addressc c s� � | j �| j�}|r5| jd u rt�dt|�d �| _tt|��D ]}| jd t|� | _|| j }|V q z| �� V W n t yE Y nw t
d| j����)Nr � zNo slave found for ) rH �discover_slavesr" rK �random�randint�len�ranger4 r r )r �slaves�_r6 r r r r5 u s �
�z)SentinelConnectionPoolProxy.rotate_slavesN)r r r r rI r4 r5 r r r r rD V s
rD c s\ e Zd ZdZ� fdd�Zdd� Z� fdd�Zedd � �Z� fd
d �Z d d
� Z
dd� 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|�dd�r tnt�|d< |�dd�| _|�dd�| _t| | j| j||d�| _t� j d i |�� | j| j
d< || _ || _ d S )
N�connection_class�sslFr3 Tr, )r r3 r, r" rH r r )
�getr rC r r3 r, rD �proxyr r �connection_kwargsr" rH )r r" rH r r r r r � s&
���
z SentinelConnectionPool.__init__c C s, | j rdnd}t| �j� d| j� d|� d�S )N�masterr6 r �(�))r3 r! r r" )r �roler r r r( � s z SentinelConnectionPool.__repr__c s t � �� | j�� d S r1 )r rI rZ r< r r r rI � s
z SentinelConnectionPool.resetc C s | j jS r1 )rZ rJ r< r r r rJ � s z%SentinelConnectionPool.master_addressc s4 | j p| j o| j|j|jfk}t� }|o|�|�S r1 )r3 rJ r# r$ r �owns_connection)r �
connection�check�parentr r r r` � s
�z&SentinelConnectionPool.owns_connectionc C �
| j �� S r1 )rZ r4 r< r r r r4 � s
z)SentinelConnectionPool.get_master_addressc C rd )zRound-robin slave balancer)rZ r5 r< r r r r5 � s
z$SentinelConnectionPool.rotate_slaves)
r r r �__doc__r r( rI �propertyrJ r` r4 r5 rB r r r r rV � s
rV c @ sj e Zd ZdZ ddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
e e fdd�Z
e e fdd�ZdS )�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 sD |d u r
dd� |� � D �}|� _� fdd�|D �� _|� _|� _d S )Nc S s i | ] \}}|� d �r||�qS )�socket_)�
startswith)�.0�k�vr r r �
<dictcomp>� s
��z%Sentinel.__init__.<locals>.<dictcomp>c s$ g | ]\}}t ||fi � j���qS r )r �sentinel_kwargs)rj �hostnamer$ r< r r �
<listcomp>� s ��z%Sentinel.__init__.<locals>.<listcomp>)�itemsrn � sentinels�min_other_sentinelsr[ )r rr rs rn r[ r r< r r � s �
�
zSentinel.__init__c O sf t |�dd��}d|�� v r|�d� |r#t�| j�j|i |�� dS | jD ]
}|j|i |�� q&dS )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.
�onceFT)rA rY �keysr rP �choicerr �execute_command)r �argsr rt �sentinelr r r rw � s
�zSentinel.execute_commandc C s@ g }| j D ] }|�d�|jj�� qt| �j� dd�|�� d�S )Nz
{host}:{port}z <sentinels=[�,z]>)rr �append�
format_mapr r[ r! r �join)r �sentinel_addressesry r r r r( s
� zSentinel.__repr__c C s2 |d r |d s |d rdS |d | j k rdS dS )Nr3 �is_sdown�is_odownFznum-other-sentinelsT)rs )r �stater" r r r �check_master_state
s
zSentinel.check_master_statec C s� t � }t| j�D ]N\}}z|�� }W n ttfy0 } z|�|� d|��� W Y d}~qd}~ww |�|�}|rV| �||�rV|| jd | jd< | j|< |d |d f S qd}t |�dkrgdd�
|�� �}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, zNo master found for ) �list� enumeraterr �sentinel_mastersr r r{ rY r� rR r} r ) r r" �collected_errors� sentinel_nory �masters�er� �
error_infor r r rM s( ��
�� zSentinel.discover_masterc C s: g }|D ]}|d s|d rq|� |d |d f� q|S )z1Remove slaves that are in an ODOWN or SDOWN stater� r r� r$ )r{ )r rT � slaves_aliver6 r r r �
filter_slaves/ s zSentinel.filter_slavesc C sN | j D ]!}z|�|�}W n tttfy Y qw | �|�}|r$| S qg S )z;Returns a list of alive slaves for service ``service_name``)rr �sentinel_slavesr r
r r� )r r" ry rT r r r rO 8 s
�
�zSentinel.discover_slavesc K �4 d|d< t | j�}|�|� |�||| fi |���S )a�
Returns a redis client instance for the ``service_name`` master.
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.
Tr3 ��dictr[ �update� from_pool�r r" � redis_class�connection_pool_classr r[ r r r �
master_forD s
�zSentinel.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.
Fr3 r� r� r r r � slave_forh s
�zSentinel.slave_for)r N)r r r re r rw r( r� rM r� rO r rV r� r� r r r r rg � s"
�
�'�rg )rP rE �typingr � redis.clientr �redis.commandsr �redis.connectionr r r �redis.exceptionsr r r
r � redis.utilsr r r r rC rD rV rg r r r r �<module> s >08