U h.@sddlZddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z mZddlmZddlmZeeZGdd d ZGd d d ZGd d d ZGdddZddZejdddZdS)N)datetimetimezone)sleep)Any AwaitableCallableUnion)RequestTokenErrTokenRenewalErr)IdentityProviderInterface) TokenResponsec@seZdZdZddZeeeegdfe fdddZ e j eeegdfe fddd dZ eeee gdfe fdd d Z e j eee gdfe fddd 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_errorselfrr9r)rr* acquire_tokenr r force_refreshtokenerrrrgszTokenManager.acquire_tokenc sz|j|}Wn~tk r}z`|j|jkrz|jd7_t|j dIdH| |IdHWY S|W5d}~XYnXd|_t |Srd) r=rfr rBr>r9r)rMrr*acquire_token_asyncr rhrrrrls z TokenManager.acquire_token_async) expire_date issue_datercCs4||}|||}t||}|dkr,dS|dS)Nrr^)_delay_for_lower_refresh_delay_for_ratio_refreshmin)rrmrnZdelay_for_lower_refreshZdelay_for_ratio_refreshdelayrrr_calculate_renewal_delays   z%TokenManager._calculate_renewal_delay)rmcCs"||jttjdSNr^)r>r7rnowrutc timestamp)rrmrrrros z%TokenManager._delay_for_lower_refresh)rmrncCs6||}|||j}||ttjdSrt)r>r6rrurrvrw)rrmrnZ token_ttlZrefresh_beforerrrrps z%TokenManager._delay_for_ratio_refresh)rHr[c CsFz.z|jdd}|||}|rBtd|jj dkrbt dWWdS|sz|j |Wn*t k r}z t|W5d}~XYnX|dkrWW~dSt }|||j|_t d|d|WWHSt k r,}z"|jjdkr||j|W5d}~XYnXW5|r@|XdS) zq Task to renew token from identity provider. Schedules renewal tasks based on token TTL. TriRequested token is expiredN@No registered callback for token renewal task. Renewal cancelledr Next token renewal scheduled in  seconds)setrgrs get_tokenget_expires_at_msget_received_at_ms is_expiredr r@rrCwarningr$rMrNrUrVr?rDr)rrHr[ token_resrrrkrYrrrrVs<        "zTokenManager._renew_tokenc sZzBz|jddIdH}|||}|rHtd|jj dkrht dWWdS|sz|j |IdHWn*t k r}z t|W5d}~XYnX|dkrWWdSt }t||j}t d|d|||WnHt k r@}z(|jjdkr||j|IdHW5d}~XYnXW5|rT|XdS) zx Async task to renew tokens from identity provider. Schedules renewal tasks based on token TTL. TrxNryrzrr{r|)r}rlrsr~rrrr r@rrCrr$rMrNr_r`rDrUr)rrHr[rrrrkrYrarrrr`(s<       (zTokenManager._renew_token_async)F)FrF)F)F)FN)FN)rrr r r-rrFr boolrrSr,rbrEr rgrlrsrorprMrTrVr`rrrrr:ysL     /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. cstjddS)N)rY)rMZ ensure_futurerrJ coro_funckwargsrYrrrabsz'_async_to_sync_wrapper..wrappedr)rYrrJrrarrrr_Ws r_Z 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)rMZset_event_loopZ run_foreverrrrrrRis rR)rMloggingrPrrtimertypingrrrrZredis.auth.errr r Zredis.auth.idpr Zredis.auth.tokenr getLoggerrrCr r%r-r:r_ZAbstractEventLooprRrrrrs     9_