U gE9@sddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZddlmZGdd d e ZGd d d e ZGd d d e ZGdddee ZGdddZGddde ZGdddeZdS)N)Optional)Redis)SentinelCommands) ConnectionConnectionPool SSLConnection)ConnectionError ReadOnlyError ResponseError TimeoutError) str_if_bytesc@s eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rr2/tmp/pip-unpacked-wheel-f3sx1i9r/redis/sentinel.pyr sr c@s eZdZdS)SlaveNotFoundErrorNrrrrrrsrcsfeZdZfddZddZfddZddZd d Zdd d d ee ee d fd dZ Z S)SentinelManagedConnectionc s|d|_tjf|dS)Nconnection_pool)poprsuper__init__)selfkwargs __class__rrrs z"SentinelManagedConnection.__init__cCsR|j}dt|jdt|jd|jd}|jrNd|jd|j}||}|S)N<. (service=z%s)>z,host=z,port=)rtyperr service_namehostport)rpoolsZ host_inforrr__repr__s$z"SentinelManagedConnection.__repr__csD|\|_|_t|jjr@|dt|dkr@t ddS)NZPINGZPONGz PING failed) r#r$rconnectrcheck_connectionZ send_commandr read_responser)raddressrrr connect_to$s    z$SentinelManagedConnection.connect_toc Csf|jr dS|jjr$||jn>|jD].}z||WStk rZYq.Yq.Xq.tdSN)_sockr is_masterr,get_master_address rotate_slavesrr)rslaverrr_connect_retry,s z(SentinelManagedConnection._connect_retrycCs|j|jddS)NcSsdSr-r)errorrrr:z3SentinelManagedConnection.connect..)retryZcall_with_retryr3rrrrr(9sz!SentinelManagedConnection.connectF)disconnect_on_error push_requestcsHztj|||dWStk rB|jjr<|tdYnXdS)N)disable_decodingr9r:z"The previous master is now a slave)rr*r rr/ disconnectr)rr;r9r:rrrr*<sz'SentinelManagedConnection.read_response)F) rrrrr'r,r3r(rboolr* __classcell__rrrrrs   rc@s eZdZdS)SentinelManagedSSLConnectionNrrrrrr?Usr?c@s,eZdZddZddZddZddZd S) SentinelConnectionPoolProxycCs0t||_||_||_||_||_|dSr-)weakrefrefconnection_pool_refr/r)r"sentinel_managerreset)rrr/r)r"rDrrrrZs  z$SentinelConnectionPoolProxy.__init__cCsd|_d|_dSr-)master_addressslave_rr_counterr8rrrrEisz!SentinelConnectionPoolProxy.resetcCsD|j|j}|jr@|j|kr@||_|}|dk r@|jdd|S)NF)Zinuse_connections)rDdiscover_masterr"r/rFrCr<)rrFrrrrr0ms z.SentinelConnectionPoolProxy.get_master_addressccs|j|j}|rh|jdkr2tdt|d|_tt|D](}|jdt||_||j}|Vq>z|VWnt k rYnXt d|jdS)NrzNo slave found for ) rDdiscover_slavesr"rGrandomrandintlenranger0r r)rslaves_r2rrrr1xs  z)SentinelConnectionPoolProxy.rotate_slavesN)rrrrrEr0r1rrrrr@Ys r@cs\eZdZdZfddZddZfddZedd Zfd d Z d d Z ddZ 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|ddrtnt|d<|dd|_|dd|_t||j|j||d|_tj f||j|j d<||_ ||_ dS) NZconnection_classsslFr/Tr))rr/r)r"rDr) getrr?rr/r)r@proxyrrconnection_kwargsr"rD)rr"rDrrrrrs&  zSentinelConnectionPool.__init__c Cs:|jr dnd}dt|jdt|jd|jd|d S)NZmasterr2rrr (z))>)r/r!rrr")rrolerrrr's*zSentinelConnectionPool.__repr__cst|jdSr-)rrErTr8rrrrEs zSentinelConnectionPool.resetcCs|jjSr-)rTrFr8rrrrFsz%SentinelConnectionPool.master_addresscs4|j p|jo|j|j|jfk}t}|o2||Sr-)r/rFr#r$rowns_connection)r connectioncheckparentrrrrXs z&SentinelConnectionPool.owns_connectioncCs |jSr-)rTr0r8rrrr0sz)SentinelConnectionPool.get_master_addresscCs |jS)zRound-robin slave balancer)rTr1r8rrrr1sz$SentinelConnectionPool.rotate_slaves) rrr__doc__rr'rEpropertyrFrXr0r1r>rrrrrQs    rQc@sfeZdZdZdddZddZdd Zd d Zd d ZddZ ddZ e e fddZ e e fddZdS)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. rNc sJ|dkrdd|D}|_fdd|D_|_|_|_dS)NcSs i|]\}}|dr||qS)Zsocket_) startswith).0kvrrr s z%Sentinel.__init__..cs g|]\}}t||fjqSr)rsentinel_kwargs)r`hostnamer$r8rr sz%Sentinel.__init__..)itemsrd sentinelsmin_other_sentinelsrU_force_master_ip)rrhrirdZforce_master_iprUrr8rrs  zSentinel.__init__cOs\t|dd}d|kr&|d|r@t|jj||n|jD]}|j||qFdS)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)r=rSkeysrrKchoicerhexecute_command)rargsrrksentinelrrrrns   zSentinel.execute_commandcCsNg}|jD]}|d|jjq dt|jdt|jdd|dS)Nz {host}:{port}rrz (sentinels=[,z])>) rhappend format_maprrUr!rrjoin)rZsentinel_addressesrprrrr' s  (zSentinel.__repr__cCs2|dr|ds|drdS|d|jkr.dSdS)Nr/is_sdownis_odownFznum-other-sentinelsT)ri)rstater"rrrcheck_master_states zSentinel.check_master_statec Cst}t|jD]\}}z |}WnBttfk rf}z ||d|WYqW5d}~XYnX||}|r|||r||jd|jd<|j|<|j dk r|j n|d}||dfSqd} t |dkrdd |} t d || dS) 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 - Nripr$z : z, zNo master found for ) list enumeraterhZsentinel_mastersrr rrrSrxrjrMrtr ) rr"Zcollected_errorsZ sentinel_norpZmasterserwry error_inforrrrHs*   zSentinel.discover_mastercCs:g}|D],}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN statervruryr$)rr)rrOZ slaves_aliver2rrr filter_slavesBs zSentinel.filter_slavesc CsT|jD]H}z||}Wntttfk r6YqYnX||}|r|SqgS)z;Returns a list of alive slaves for service ``service_name``)rhZsentinel_slavesrr r r)rr"rprOrrrrJKs    zSentinel.discover_slavescKs0d|d<t|j}||||||f|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. Tr/dictrUupdateZ from_poolrr"Z redis_classZconnection_pool_classrrUrrr master_forWs    zSentinel.master_forcKs0d|d<t|j}||||||f|S)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. Fr/rrrrr slave_for{s    zSentinel.slave_for)rNN)rrrr\rrnr'rxrHrrJrrQrrrrrrr^s"  #  'r^)rKrAtypingrZ redis.clientrZredis.commandsrZredis.connectionrrrZredis.exceptionsrr r r Z redis.utilsr r rrr?r@rQr^rrrrs    A0=