o svXj-@sddlZddlZddlmZddlmZmZddlmZm Z m Z m Z ddl m Z mZmZGdddeZGdd d eZGd d d eZGd d d eZGdddeZdS)N)Redis)ConnectionPool Connection)ConnectionError ResponseError ReadOnlyError TimeoutError) iteritems nativestrxrangec@ eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rrK/var/www/html/myproject/venv/lib/python3.10/site-packages/redis/sentinel.pyr r c@r )SlaveNotFoundErrorNrrrrrrrrcsDeZdZfddZddZfddZddZfd d ZZS) SentinelManagedConnectionc s&|d|_tt|jdi|dS)Nconnection_poolr)poprsuperr__init__)selfkwargs __class__rrrs z"SentinelManagedConnection.__init__cCs<|j}dt|j|jf}|jrd|j|jf}||}|S)Nz%sz,host=%s,port=%s)rtyper service_namehostport)rpools host_inforrr__repr__s z"SentinelManagedConnection.__repr__csL|\|_|_tt||jjr"|dt| dkr$t ddSdS)NPINGPONGz PING failed) r!r"rrconnectrcheck_connection send_commandr read_responser)raddressrrr connect_to s  z$SentinelManagedConnection.connect_toc Cs\|jrdS|jjr||jdS|jD]}z||WSty+YqwtN)_sockr is_masterr.get_master_address rotate_slavesrr)rslaverrrr)(s z!SentinelManagedConnection.connectcs:ztt|WSty|jjr|tdw)Nz"The previous master is now a slave)rrr,rrr1 disconnectrrrrrr,5s z'SentinelManagedConnection.read_response) rrrrr&r.r)r, __classcell__rrrrrs    rcsPeZdZdZfddZddZfddZfdd Zd d Zd 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 sb|dt|d<|dd|_|dd|_tt|jdi|t ||j d<||_ ||_ dS)Nconnection_classr1Tr*Frr) getrrr1r*rr8rweakrefproxyconnection_kwargsr sentinel_manager)rr r>rrrrrLs zSentinelConnectionPool.__init__cCs dt|j|j|jr dp dfS)Nz%sdiscover_masterr r1rAr5)rrArrrr2is  z)SentinelConnectionPool.get_master_addressccs|j|j}|r5|jdurtdt|d|_tt|D]}|jdt||_||j}|Vq z|VWn t yEYnwt d|j)zRound-robin slave balancerNrzNo slave found for %r) r>discover_slavesr rBrandomrandintlenr r2r r)rslaves_r4rrrr3ts    z$SentinelConnectionPool.rotate_slaves) rrr__doc__rr&r@rCr2r3r7rrrrr8Ds    r8c@s^eZdZdZdddZddZdd Zd d Zd d ZddZ e e fddZ e e fddZ dS)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 sD|dur ddt|D}|_fdd|D_|_|_dS)NcSs i|] \}}|dr||qS)socket_) startswith).0kvrrr s z%Sentinel.__init__..cs$g|]\}}t||fijqSr)rsentinel_kwargs)rThostnamer"r6rr sz%Sentinel.__init__..)r rX sentinelsmin_other_sentinelsr=)rr[r\rXr=rr6rrs  zSentinel.__init__cCsJg}|jD]}|d|jjd|jjdfqdt|jd|fS)Nz%s:%sr!r"z%s,)r[appendrr=rrjoin)rsentinel_addressessentinelrrrr&s    zSentinel.__repr__cCs2|dr |ds |drdS|d|jkrdSdS)Nr1is_sdownis_odownFznum-other-sentinelsT)r\)rstater rrrcheck_master_states zSentinel.check_master_statec Cst|jD];\}}z|}Wn ttfyYqw||}|r@|||r@||jd|jd<|j|<|d|dfSqtd|f)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. ripr"zNo master found for %r) enumerater[sentinel_mastersrrr:rer )rr sentinel_noramastersrdrrrrHs   zSentinel.discover_mastercCs:g}|D]}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN statercrbrfr")r^)rrN slaves_aliver4rrr filter_slavess zSentinel.filter_slavesc CsN|jD]!}z||}Wn tttfyYqw||}|r$|SqgS)z;Returns a list of alive slaves for service ``service_name``)r[sentinel_slavesrrrrl)rr rarNrrrrJs  zSentinel.discover_slavescK4d|d<t|j}||||||fi|dS)aF Returns a redis client instance for the ``service_name`` master. A SentinelConnectionPool class is used to retrive 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 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. Tr1rdictr=updaterr redis_classconnection_pool_classrr=rrr master_fors   zSentinel.master_forcKrn)a Returns redis client instance for the ``service_name`` slave(s). A SentinelConnectionPool class is used to retrive the slave's address before establishing a new connection. By default clients will be a 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. Fr1rorprsrrr slave_fors   zSentinel.slave_for)rN)rrrrPrr&rerHrlrJrr8rvrwrrrrrQs     rQ)rKr; redis.clientrredis.connectionrrredis.exceptionsrrrr redis._compatr r r r rrr8objectrQrrrrs 1C