)Jf_76ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZddlmZGdd e ZGd d e ZGd d e ZGddee ZGddZGdde ZGddeZdS)N)Optional)Redis)SentinelCommands) ConnectionConnectionPool SSLConnection)ConnectionError ReadOnlyError ResponseError TimeoutError) str_if_bytesceZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__Q/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/redis/sentinel.pyrr DrrceZdZdS)SlaveNotFoundErrorNrrrrrrrrrcreZdZfdZdZfdZdZdZ d ddddee d ee ffd Z xZ S) SentinelManagedConnectionc n|d|_tjdi|dS)Nconnection_poolr)poprsuper__init__)selfkwargs __class__s rr z"SentinelManagedConnection.__init__s;%zz*;<<""6"""""rc|j}t|jd|jd}|jrd|jd|j}||z}|S)N z,host=z,port=)rtyper service_namehostport)r!pools host_infos r__repr__z"SentinelManagedConnection.__repr__s]#Dzz" C CT-> C C C 9 ===$)==II Arc|\|_|_t|jjrI|dt|dkrtddSdS)NPINGPONGz PING failed) r(r)rconnectrcheck_connection send_commandr read_responser )r!addressr#s r connect_toz$SentinelManagedConnection.connect_to!s& 49    0 5   f % % %D..0011V;;%m444 5 5;;rc|jrdS|jjr.||jdS|jD])} ||cS#t $rY&wxYwtN)_sockr is_masterr6get_master_address rotate_slavesr r)r!slaves r_connect_retryz(SentinelManagedConnection._connect_retry)s :  F   ) % OOD0CCEE F F F F F-;;==  ??511111&H$ $s A77 BBcD|j|jdS)NcdSr8r)errors rz3SentinelManagedConnection.connect..7sTr)retrycall_with_retryr>r!s rr1z!SentinelManagedConnection.connect6s!z))$*=?Q?QRRRrF)disconnect_on_error push_requestrFrGc t|||S#t$r1|jjr#|t dwxYw)N)disable_decodingrFrGz"The previous master is now a slave)rr4r rr: disconnectr )r!rIrFrGr#s rr4z'SentinelManagedConnection.read_response9s 77((!1$7))     #- L !!!%&JKKK  s #';A")F) rrrr r-r6r>r1rboolr4 __classcell__r#s@rrrs#####55555 % % %SSS /4', &d^  tn rrceZdZdS)SentinelManagedSSLConnectionNrrrrrOrORrrrOc&eZdZdZdZdZdZdS)SentinelConnectionPoolProxyctj||_||_||_||_||_|dSr8)weakrefrefconnection_pool_refr:r2r'sentinel_managerreset)r!rr:r2r'rVs rr z$SentinelConnectionPoolProxy.__init__WsF$+;#?#? " 0( 0 rc"d|_d|_dSr8)master_addressslave_rr_counterrEs rrWz!SentinelConnectionPoolProxy.resetfs" $rc|j|j}|jr>|j|kr3||_|}||d|S)NF)inuse_connections)rVdiscover_masterr'r:rYrUrJ)r!rYrs rr;z.SentinelConnectionPoolProxy.get_master_addressjsq.>>t?PQQ > Dd1^CC"0D #6688O***U*CCCrc#K|j|j}|r|j*t jdt |dz |_tt |D]2}|jdzt |z|_||j}|V3 |Vn#t$rYnwxYwtd|j)NrzNo slave found for ) rVdiscover_slavesr'rZrandomrandintlenranger;rr)r!slaves_r=s rr<z)SentinelConnectionPoolProxy.rotate_slavesus&66t7HII  $,(.q#f++/(J(J%3v;;''  )-)>)Bc&kk(Q%t45  ))++ + + + +"    D  !Lt7H!L!LMMMs%B<< C C N)rrrr rWr;r<rrrrQrQVsU   %%%   NNNNNrrQc^eZdZdZfdZdZfdZedZfdZ dZ dZ xZ 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 |d|ddrtnt|d<|dd|_|dd|_t ||j|j|||_tj d i||j|j d<||_ ||_ dS) Nconnection_classsslFr:Tr2)rr:r2r'rVrr) getrrOrr:r2rQproxyrr connection_kwargsr'rV)r!r'rVr"r#s rr zSentinelConnectionPool.__init__s%+ZZ zz%'' + ( (* & & !"  K66 & +=u E E0 n!2%-      ""6"""48J01( 0rc\|jrdnd}t|jd|jd|dS)Nmasterr=r%())r:r&rr')r!roles rr-zSentinelConnectionPool.__repr__s=>6xxwt**%KK0AKKDKKKKrczt|jdSr8)rrWrm)r!r#s rrWzSentinelConnectionPool.resets.   rc|jjSr8)rmrYrEs rrYz%SentinelConnectionPool.master_addresss z((rc|j p|jo|j|j|jfk}t }|o||Sr8)r:rYr(r)rowns_connection)r! connectioncheckparentr#s rrwz&SentinelConnectionPool.owns_connectionsTN" N Xt2z 6XX ;// ;;;rc4|jSr8)rmr;rEs rr;z)SentinelConnectionPool.get_master_addresssz,,...rc4|jS)zRound-robin slave balancer)rmr<rEs rr<z$SentinelConnectionPool.rotate_slavessz'')))r) rrr__doc__r r-rWpropertyrYrwr;r<rLrMs@rrhrhs11111*LLL))X)<<<<<///*******rrhcZeZdZdZ d dZdZdZdZdZd Z d Z e e fd Z e e fd 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. rNc |d|D}|_fd|D_|_|_dS)NcDi|]\}}|d||S)socket_) startswith).0kvs r z%Sentinel.__init__..s?Aall9>U>U1rc<g|]\}}t||fijSr)rsentinel_kwargs)rhostnamer)r!s r z%Sentinel.__init__..sB   $ (D 9 9D$8 9 9   r)itemsr sentinelsmin_other_sentinelsrn)r!rrrrns` rr zSentinel.__init__s  "!2!8!8!:!:O /    "+   $7 !2rct|dd}d|vr|d|r%t j|jj|i|n|jD]}|j|i|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)rKrlkeysrrachoicerexecute_command)r!argsr"rsentinels rrzSentinel.execute_commands FJJvu--.. V[[]] " " JJv     : 9FM$. ) ) 94 J6 J J J J N : :(($9&9999trcg}|jD]4}|d|jj5t |jdd|dS)Nz {host}:{port}z )rappend format_maprrnr&rjoin)r!sentinel_addressesrs rr-zSentinel.__repr__s|  H  % %**8+C+UVV    t**%SS388%> 2 2 !K "3355#\2    ''8(=(=(=(=>>> KK --E 200 EE 2N1%?q!4>+#>T{E&M1111  1 $ $A3A..A3c|g}|D]6}|ds|dr||d|df7|S)z1Remove slaves that are in an ODOWN or SDOWN staterrrr))r)r!re slaves_aliver=s r filter_slaveszSentinel.filter_slaves/s\  > >EZ  E*$5    teFm < = = = =rc|jD]Q} ||}n#tttf$rY2wxYw||}|r|cSRgS)z;Returns a list of alive slaves for service ``service_name``)rsentinel_slavesr r r r)r!r'rres rr`zSentinel.discover_slaves8s  H !11,??#]LA    ''//F    s !;;c 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. Tr:dictrnupdate from_poolr!r' redis_classconnection_pool_classr"rns r master_forzSentinel.master_forDsd:#{ !788  ((($$ ! !, J J8I J J   rc d|d<t|j}||||||fi|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:rrs r slave_forzSentinel.slave_forhsd0${ !788  ((($$ ! !, J J8I J J   r)rN)rrrr}r rr-rr]rr`rrhrrrrrrrs> 3333,"TTTWWW:   4 " " " " N4       rr)rarStypingr redis.clientrredis.commandsrredis.connectionrrrredis.exceptionsr r r r redis.utilsr rrrrOrQrhrrrrrs ++++++FFFFFFFFFFXXXXXXXXXXXX$$$$$$     /           ;;;;; ;;;|     #