o ;h,9@sddlZddlZddlZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZmZmZmZddlmZddlmZmZmZmZddlmZGdd d eZGd d d eZGd d d eZGdddeeZGdddeZGdddeZ dS)N) AsyncIteratorIterableMappingOptionalSequenceTupleType)Redis) ConnectionConnectionPool EncodableT SSLConnection)AsyncSentinelCommands)ConnectionError ReadOnlyError ResponseError TimeoutError) str_if_bytesc@ eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rrI/var/www/html/venv/lib/python3.10/site-packages/redis/asyncio/sentinel.pyrrc@r)SlaveNotFoundErrorNrrrrrrrrc sxeZdZfddZddZfddZddZd d Z dd d ddede e de e de effddZ Z S)SentinelManagedConnectionc s"|d|_tjdi|dS)Nconnection_poolr)poprsuper__init__)selfkwargs __class__rrr"s z"SentinelManagedConnection.__init__cCsBd|jjd|jj}|jrd|jd|j}||7}|dS)N<.z,host=z,port=z)>)r&rrhostport)r#sZ host_inforrr__repr__s z"SentinelManagedConnection.__repr__cs\|\|_|_tIdH|jjr*|dIdHt|IdHdkr,t ddSdS)NZPINGZPONGz PING failed) r)r*r!connectrcheck_connectionZ send_commandr read_responser)r#addressr%rr connect_to&s z$SentinelManagedConnection.connect_toc sz|jrdS|jjr||jIdHIdHdS|j2z3dHW}z ||IdHWSty9Yqw6tN)_readerr is_masterr1get_master_address rotate_slavesrr)r#slaverrr_connect_retry.s  z(SentinelManagedConnection._connect_retrycs|j|jddIdHS)NcSs tdS)Nr)asynciosleep)errorrrr>s z3SentinelManagedConnection.connect..)retryZcall_with_retryr8r#rrrr-;s  z!SentinelManagedConnection.connectFNT)disconnect_on_error push_requestdisable_decodingtimeoutr?r@csNztj||||dIdHWSty&|jjr%|IdHtdw)N)rArBr?r@z"The previous master is now a slave)r!r/rrr4 disconnectr)r#rArBr?r@r%rrr/As z'SentinelManagedConnection.read_response)FN) rrrr"r,r1r8r-boolrfloatr/ __classcell__rrr%rrs(   rc@r)SentinelManagedSSLConnectionNrrrrrrG\rrGcs\eZdZdZfddZddZfddZdeffd d Zd d Z d e fddZ 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 sz|d|ddr tnt|d<|dd|_|dd|_tjdi|t ||j d<||_ ||_ d|_ d|_dS) NZconnection_classsslFr4Tr.rr)getr rGrr4r.r!r"weakrefproxyconnection_kwargs service_namesentinel_managermaster_addressslave_rr_counter)r#rNrOr$r%rrr"hs  zSentinelConnectionPool.__init__c Cs2d|jjd|jjd|jd|jrdpdd S)Nr'r(z (service=(Zmasterr7z))>)r&rrrNr4r>rrrr,zs  zSentinelConnectionPool.__repr__cstd|_d|_dSr2)r!resetrPrQr>r%rrrSs  zSentinelConnectionPool.reset connectioncs0|j p|jo|j|j|jfk}|ot|Sr2)r4rPr)r*r!owns_connection)r#rTcheckr%rrrUsz&SentinelConnectionPool.owns_connectioncsB|j|jIdH}|jr|j|kr||_|jddIdH|S)NF)Zinuse_connections)rOdiscover_masterrNr4rPrC)r#rPrrrr5s z)SentinelConnectionPool.get_master_addressreturncCs|j|jIdH}|r8|jdurtdt|d|_tt|D]}|jdt||_||j}|Vq#z |IdHVWn t yKYnwt d|j)zRound-robin slave balancerNrzNo slave found for ) rOdiscover_slavesrNrQrandomrandintlenranger5rr)r#slaves_r7rrrr6s   z$SentinelConnectionPool.rotate_slaves) rrr__doc__r"r,rSr rUr5rr6rFrrr%rrH`s   rHc@seZdZdZ   dddZddZdd Zd ed ed e fd dZ d efddZ de e d eeeeffddZd ed eeeeffddZeefd edeedeefddZeefd edeedeefddZdS)Sentinela~ 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) >>> await master.set('foo', 'bar') >>> slave = sentinel.slave_for('mymaster', socket_timeout=0.1) >>> await 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|dur dd|D}|_fdd|D_|_|_|_dS)NcSs i|] \}}|dr||qS)Zsocket_) startswith).0kvrrr s z%Sentinel.__init__..cs&g|]\}}td||djqS))r)r*r)r sentinel_kwargs)rdhostnamer*r>rr sz%Sentinel.__init__..)itemsrh sentinelsmin_other_sentinelsrM_force_master_ip)r#rlrmrhZforce_master_iprMrr>rr"s   zSentinel.__init__csxtdd}dvrd|r't|jjiIdHdSfdd|jD}tj |IdHdS)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. onceFNcs"g|] }t|jiqSr)r9Taskexecute_command)rdsentinelargsr$rrrjsz,Sentinel.execute_command..T) rDrJkeysr r[choicerlrqr9Zgather)r#rtr$rotasksrrsrrqs   zSentinel.execute_commandcCsXg}|jD]}||jjdd|jjdqd|jd|jjdd|dS) Nr):r*r'r(z (sentinels=[,z])>)rlappendrrMr&rjoin)r#Zsentinel_addressesrrrrrr,s  zSentinel.__repr__staterNrXcCs2|dr |ds |drdS|d|jkrdSdS)Nr4is_sdownis_odownFznum-other-sentinelsT)rm)r#r|rNrrrcheck_master_states zSentinel.check_master_statec st}t|jD][\}}z |IdH}Wnttfy4}z||d|WYd}~q d}~ww||}|rd|||rd||jd|jd<|j|<|j durX|j n|d}||dfSq d} t |dkrudd |} 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. Nz - ripr*z : z, zNo master found for ) list enumeraterlZsentinel_mastersrrrzrJrrnr]r{r) r#rNZcollected_errorsZ sentinel_norrZmasterser|r error_inforrrrWs2    zSentinel.discover_masterr_cCs:g}|D]}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN stater~r}rr*)rz)r#r_Z slaves_aliver7rrr filter_slaves's zSentinel.filter_slavesc sV|jD]$}z ||IdH}Wn tttfyYqw||}|r(|SqgS)z;Returns a list of alive slaves for service ``service_name``N)rlZsentinel_slavesrrrr)r#rNrrr_rrrrZ2s  zSentinel.discover_slaves redis_classconnection_pool_classcK8d|d<t|j}|||||fi|}||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. Tr4dictrMupdateZ from_poolr#rNrrr$rMrrrr master_for@s    zSentinel.master_forcKr)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. Fr4rrrrr slave_forgs    zSentinel.slave_for)rNN)rrrrar"rqr,rstrrDrrWrrrrr rrZr rHrrrrrrrrbsL  #    *rb)!r9r[rKtypingrrrrrrrZredis.asyncio.clientr Zredis.asyncio.connectionr r r r Zredis.commandsrZredis.exceptionsrrrrZ redis.utilsrrrrrGrHrbrrrrs$   BG