o >h.@sddlZddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z mZddlmZddlmZeeZGdd d ZGd d d ZGd d d ZGdddZddZdejfddZdS)N)datetimetimezone)sleep)Any AwaitableCallableUnion)RequestTokenErrTokenRenewalErr)IdentityProviderInterface) TokenResponsec@seZdZdZddZedeeegdfe ffddZ e j deeegdfe fddfd dZ edeee gdfe ffd d Z e j deee gdfe fddfd d Z dS) CredentialsListenerzz Listeners that will be notified on events related to credentials. Accepts callbacks and awaitable callbacks. cCsd|_d|_dSN)_on_next _on_errorselfrx/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/redis/auth/token_manager.py__init__ zCredentialsListener.__init__returnNcC|jSrrrrrron_nextzCredentialsListener.on_nextcallbackcC ||_dSrrrrrrrr cCrrrrrrron_error!rzCredentialsListener.on_errorcCrrr rrrrr!%r)__name__ __module__ __qualname____doc__rpropertyrrrrrsetter Exceptionr!rrrrr s"&"*r c@s:eZdZdedefddZdefddZdefdd Zd S) RetryPolicy max_attempts delay_in_mscCs||_||_dSr)r*r+)rr*r+rrrr+rzRetryPolicy.__init__rcCr)zW Retry attempts before exception will be thrown. :return: int )r*rrrrget_max_attempts/zRetryPolicy.get_max_attemptscCr)zI Delay between retries in seconds. :return: int )r+rrrrget_delay_in_ms7r-zRetryPolicy.get_delay_in_msN)r"r#r$intfloatrr,r.rrrrr)*sr)c@s^eZdZdedededefddZdefdd Zdefd d Zdefd d Z defddZ dS)TokenManagerConfigexpiration_refresh_ratiolower_refresh_bound_millis%token_request_execution_timeout_in_ms retry_policycCs||_||_||_||_dSr)_expiration_refresh_ratio_lower_refresh_bound_millis&_token_request_execution_timeout_in_ms _retry_policy)rr2r3r4r5rrrrAs  zTokenManagerConfig.__init__rcCr)a& Represents the ratio of a token's lifetime at which a refresh should be triggered. # noqa: E501 For example, a value of 0.75 means the token should be refreshed when 75% of its lifetime has elapsed (or when 25% of its lifetime remains). :return: float )r6rrrrget_expiration_refresh_ratioOs z/TokenManagerConfig.get_expiration_refresh_ratiocCr)a Represents the minimum time in milliseconds before token expiration to trigger a refresh, in milliseconds. This value sets a fixed lower bound for when a token refresh should occur, regardless of the token's total lifetime. If set to 0 there will be no lower bound and the refresh will be triggered based on the expirationRefreshRatio only. :return: int )r7rrrrget_lower_refresh_bound_millisZs z1TokenManagerConfig.get_lower_refresh_bound_milliscCr)z Represents the maximum time in milliseconds to wait for a token request to complete. :return: int )r8rrrr)get_token_request_execution_timeout_in_msgszys^        /r>csfdd}|S)a Wraps an asynchronous function so it can be used with loop.call_later. :param loop: The event loop in which the coroutine will be executed. :param coro_func: The coroutine function to wrap. :param args: Positional arguments to pass to the coroutine function. :param kwargs: Keyword arguments to pass to the coroutine function. :return: A regular function suitable for loop.call_later. cstjiddS)N)r_)rQ ensure_futurerrN coro_funckwargsr_rrrgbsz'_async_to_sync_wrapper..wrappedr)r_rrNrrgrrrreWs re event_loopcCst||dS)z Starts event loop in a thread. Used to be able to schedule tasks using loop.call_later. :param event_loop: :return: N)rQset_event_loop run_forever)rrrrrWis  rW)rQloggingrUrrtimertypingrrrrredis.auth.errr r redis.auth.idpr redis.auth.tokenr getLoggerr"rGr r)r1r>reAbstractEventLooprWrrrrs"    9_