o ;ha/@svddlZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z ddl mZmZGdddeZGd d d eZGd d d eZGd ddeZGdddeZGdddZGdddeZGdddeZGdddZGdddZGdddZGdddZGdd d eZGd!d"d"eZGd#d$d$eZGd%d&d&eZ Gd'd(d(eZ!Gd)d*d*eZ"dS)+N)ABCabstractmethod)Enum)ListOptionalUnion)TokenInterface)CredentialProviderStreamingCredentialProviderc@"eZdZdZedefddZdS)EventListenerInterfacez7 Represents a listener for given event object. eventcCdSNselfr rr>/var/www/html/venv/lib/python3.10/site-packages/redis/event.pylistenzEventListenerInterface.listenN__name__ __module__ __qualname____doc__robjectrrrrrr r c@r )AsyncEventListenerInterfacez> Represents an async listener for given event object. r cdSrrrrrrrz"AsyncEventListenerInterface.listenNrrrrrrrrc@s4eZdZdZedefddZedefddZdS)EventDispatcherInterfacezf Represents a dispatcher that dispatches events to listeners associated with given event. r cCrrrrrrrdispatch%rz!EventDispatcherInterface.dispatchcrrrrrrrdispatch_async)rz'EventDispatcherInterface.dispatch_asyncN)rrrrrrr!r"rrrrr s r cs*eZdZdZdedeffdd ZZS)EventExceptionzM Exception wrapper that adds an event object into exception context. exceptionr cs||_||_t|dSr)r$r super__init__)rr$r  __class__rrr&3szEventException.__init__)rrrr Exceptionrr& __classcell__rrr'rr#.sr#c@s0eZdZddZdefddZdefddZdS) EventDispatcherc Cs:ttgttgttgttgtt gt t gi|_ dS)zY Mapping should be extended for any new events or listeners to be added. N) AfterConnectionReleasedEventReAuthConnectionListener(AfterPooledConnectionsInstantiationEvent"RegisterReAuthForPooledConnections'AfterSingleConnectionInstantiationEvent!RegisterReAuthForSingleConnection'AfterPubSubConnectionInstantiationEventRegisterReAuthForPubSub#AfterAsyncClusterInstantiationEvent"RegisterReAuthForAsyncClusterNodes!AsyncAfterConnectionReleasedEventAsyncReAuthConnectionListener_event_listeners_mappingrrrrr&;s zEventDispatcher.__init__r cCs(|jt|}|D]}||q dSrr8gettyperrr Z listenersZlistenerrrrr!Ps zEventDispatcher.dispatchcs0|jt|}|D] }||IdHq dSrr:r=rrrr"Vs zEventDispatcher.dispatch_asyncN)rrrr&rr!r"rrrrr+9sr+c@s$eZdZdZddZeddZdS)r,zA Event that will be fired before each command execution. cCs ||_dSr _connection)r connectionrrrr&b z%AfterConnectionReleasedEvent.__init__cC|jSrr>r9rrrr@ez'AfterConnectionReleasedEvent.connectionN)rrrrr&propertyr@rrrrr,]s r,c@s eZdZdS)r6N)rrrrrrrr6jsr6c@seZdZdZdZdS) ClientType)sync)asyncN)rrrSYNCASYNCrrrrrEnsrEc@sfeZdZdZ ddededeefddZe dd Z e d efd d Z e d e edffd dZ dS)r.zQ Event that will be fired after pooled connection instances was created. Nconnection_pools client_typecredential_providercC||_||_||_dSr)_connection_pools _client_type_credential_provider)rrJrKrLrrrr&x z1AfterPooledConnectionsInstantiationEvent.__init__cCrBr)rNr9rrrrJrCz9AfterPooledConnectionsInstantiationEvent.connection_poolsreturncCrBrrOr9rrrrKrCz4AfterPooledConnectionsInstantiationEvent.client_typecCrBrrPr9rrrrLrCzr9rrrr@rCz2AfterSingleConnectionInstantiationEvent.connectionrRcCrBrrSr9rrrrKrCz3AfterSingleConnectionInstantiationEvent.client_typecCrBrrVr9rrrrUrCz7AfterSingleConnectionInstantiationEvent.connection_lockN)rrrrrEr threadingLockasyncior&rDr@rKrUrrrrr0s   r0c@sreZdZdedeejejffddZe ddZ e ddZ e d efd d Z e d eejejffd d Z dS)r2rKrUcCs||_||_||_||_dSr)_pubsub_connection_connection_poolrOrV)rpubsub_connectionconnection_poolrKrUrrrr&s z0AfterPubSubConnectionInstantiationEvent.__init__cCrBr)r[r9rrrr]rCz9AfterPubSubConnectionInstantiationEvent.pubsub_connectioncCrBr)r\r9rrrr^rCz7AfterPubSubConnectionInstantiationEvent.connection_poolrRcCrBrrSr9rrrrKrCz3AfterPubSubConnectionInstantiationEvent.client_typecCrBrrWr9rrrrUrCz7AfterPubSubConnectionInstantiationEvent.connection_lockN)rrrrErrXrYrZr&rDr]r^rKrUrrrrr2s    r2c@sVeZdZdZ d dedeefddZedefdd Z ede edffd d Z dS) r4z Event that will be fired after async cluster instance was created. Async cluster doesn't use connection pools, instead ClusterNode object manages connections. NnodesrLcCs||_||_dSr)_nodesrP)rr_rLrrrr&s z,AfterAsyncClusterInstantiationEvent.__init__rRcCrBr)r`r9rrrr_rCz)AfterAsyncClusterInstantiationEvent.nodescCrBrrTr9rrrrLrCz7AfterAsyncClusterInstantiationEvent.credential_providerr) rrrrdictrr r&rDr_rrLrrrrr4s  r4c@eZdZdZdefddZdS)r-zG Listener that performs re-authentication of given connection. r cCs|jdSrr@Zre_authrrrrrszReAuthConnectionListener.listenN)rrrrr,rrrrrr-r-c@rb)r7zM Async listener that performs re-authentication of given connection. r cs|jIdHdSrrcrrrrrsz$AsyncReAuthConnectionListener.listenN)rrrrr6rrrrrr7rdr7c@ReZdZdZddZdefddZddZd d Zd e fd d Z d e fddZ dS)r/z Listener that registers a re-authentication callback for pooled connections. Required by :class:`StreamingCredentialProvider`. cC d|_dSr_eventr9rrrr&rAz+RegisterReAuthForPooledConnections.__init__r cCsbt|jtr/||_|jtjkr|j|j|j |j dS|j|j |j |j dSdSr) isinstancerLr rhrKrErHon_next_re_authon_error_raise_on_error_re_auth_async_raise_on_error_asyncrrrrrs  z)RegisterReAuthForPooledConnections.listencCs|jjD]}||qdSrrhrJre_auth_callbackrtokenpoolrrrrk s  z+RegisterReAuthForPooledConnections._re_authcs$|jjD] }||IdHqdSrrprrrrrrns z1RegisterReAuthForPooledConnections._re_auth_asyncerrorcC t||jrr#rhrrurrrrm z2RegisterReAuthForPooledConnections._raise_on_errorct||jrrwrxrrrro z8RegisterReAuthForPooledConnections._raise_on_error_asyncN) rrrrr&r.rrkrnr)rmrorrrrr/s r/c@re)r1z Listener that registers a re-authentication callback for single connection. Required by :class:`StreamingCredentialProvider`. cCrfrrgr9rrrr& rAz*RegisterReAuthForSingleConnection.__init__r cCslt|jjtr4||_|jtjkr"|jj|j |jj |j dS|jj|j |jj |j dSdSr)rir@rLr rhrKrErHrjrkrlrmrnrorrrrr#s z(RegisterReAuthForSingleConnection.listencCsV|jj|jjd|d||jjWddS1s$wYdSNZAUTHoidrhrUr@ send_commandtry_get get_value read_responserrsrrrrk2s "z*RegisterReAuthForSingleConnection._re_authc sx|jj4IdH&|jjd|d|IdH|jjIdHWdIdHdS1IdHs5wYdSr|r~rrrrrn9s .z0RegisterReAuthForSingleConnection._re_auth_asyncrucCrvrrwrxrrrrm@ryz1RegisterReAuthForSingleConnection._raise_on_errorcrzrrwrxrrrroCr{z7RegisterReAuthForSingleConnection._raise_on_error_asyncN) rrrrr&r0rrkrnr)rmrorrrrr1sr1c@s>eZdZddZdefddZdefddZd efd d Z d S) r5cCrfrrgr9rrrr&HrAz+RegisterReAuthForAsyncClusterNodes.__init__r cCs6t|jtr||_|j|j|j|jdSdSr)rirLr rhrjrkrlrmrrrrrKs z)RegisterReAuthForAsyncClusterNodes.listenrscs,|jjD]}|jj||IdHqdSr)rhr_rq)rrskeyrrrrkQs z+RegisterReAuthForAsyncClusterNodes._re_authrucrzrrwrxrrrrmUr{z2RegisterReAuthForAsyncClusterNodes._raise_on_errorN) rrrr&r4rrrkr)rmrrrrr5Gs r5c@sZeZdZddZdefddZdefddZdefd d Zd e fd d Z d e fddZ dS)r3cCs"d|_d|_d|_d|_d|_dSr)r?r\rOrVrhr9rrrr&Zs  z RegisterReAuthForPubSub.__init__r cCst|jjtrK|jdvrM||_|j|_|j|_|j |_ |j |_ |j t jkr9|jj|j|jj|jdS|jj|j|jj|jdSdSdS)N)3)rir]rLr Z get_protocolrhr?r^r\rKrOrUrVrErHrjrkrlrmrnrorrrrras$ zRegisterReAuthForPubSub.listenrscCsZ|j|jd|d||jWdn1s wY|j|dSr|rVr?rrrrr\rqrrrrrkts z RegisterReAuthForPubSub._re_authc s|j4IdH#|jd|d|IdH|jIdHWdIdHn 1IdHs1wY|j|IdHdSr|rrrrrrn}s (z&RegisterReAuthForPubSub._re_auth_asyncrucCrvrrwrxrrrrmryz'RegisterReAuthForPubSub._raise_on_errorcrzrrwrxrrrror{z-RegisterReAuthForPubSub._raise_on_error_asyncN) rrrr&r2rrrkrnr)rmrorrrrr3Ys  r3)#rZrXabcrrenumrtypingrrrZredis.auth.tokenrZredis.credentialsr r r rr r)r#r+r,r6rEr.r0r2r4r-r7r/r1r5r3rrrrs2     $   #-