VhdZddlmZddlZddlZddlZddlZddlZddlZddl m cm cm Z d dZd dZd dZd dZddZdd Zd Zd d Zy)a This module adds shared support for generic api modules In order to use this module, include it as part of a custom module as shown below. The 'api' module provides the following common argument specs: * rate limit spec - rate: number of requests per time unit (int) - rate_limit: time window in which the limit is applied in seconds * retry spec - retries: number of attempts - retry_pause: delay between attempts in seconds ) annotationsNcnttdtd}|r|j||S)z7Creates an argument spec for working with rate limitinginttype)rate rate_limitdictupdatespecarg_specs H/home/dcms/DCMS/lib/python3.12/site-packages/ansible/module_utils/api.pyrate_limit_argument_specr&s2 u U#H  Ocpttdtdd}|r|j||S)z2Creates an argument spec for working with retryingrrfloatrdefault)retries retry_pauser r s rretry_argument_specr1s4% gq1H  Orc ttdtddtdtdd}|r|j||S)NstrrT)rno_logboolr) api_username api_passwordapi_urlvalidate_certsr r s rbasic_auth_argument_specr#<sFu%uT2% 6 H   OrcLd||t|t|z fd}|S)zrate limiting decoratorNc dgfd}|S)Ngctjdk\rtj}ntj}6|dz }|z }|dkDrtj ||d<|i|}|S)N)r)sys version_infotime process_timeclocksleep) argskwargs real_timeelapsedleftretflastminrates r ratelimitedz0rate_limit..wrapper..ratelimitedQsw6) --  JJ "#+Q/(!8JJt$#+QT$V$CJr)r5r8r6r7s` @rwrapperzrate_limit..wrapperNsu r)r)rr r:r7s @rr r Hs2G J2 #eDk1$ Nrcfd}|S)zRetry decoratorcfd}|S)Ncd}?d} |dz }|k\rtdz |i|}|r |Stj.wrapper..retriedgsK"1$K"g-'(BW(LMM00 JJ{+#%sA AAr9)r5r@rrs` rr:zretry..wrapperes "rr9)rrr:s`` rretryrAcs( Nrc #|Ktd|D])}tjt||d|zz+yw)aDThe "Full Jitter" backoff strategy. Ref: https://www.awsarchitectureblog.com/2015/03/backoff.html :param retries: The number of delays to generate. :param delay_base: The base time in seconds used to calculate the exponential backoff. :param delay_threshold: The maximum time in seconds for any delay. rN)rangesecrets randbelowmin)r delay_basedelay_thresholdrAs rgenerate_jittered_backoffrJ|sAq'"OOZ!u*5L MNNOs:<cy)NFr9)exception_or_results r retry_neverrMs rc>d}||~tfd}|S)zGeneric retry decorator. :param backoff_iterator: An iterable of delays in seconds. :param should_retry_error: A callable that takes an exception of the decorated function and decides whether to retry or not (returns a bool). c3tKtj|\}}| tj|wN) itertoolsteecopy)original_iterator_copiable_iterator_first_iterator_copys r_emit_isolated_iterator_copieszGretry_with_delays_and_condition.._emit_isolated_iterator_copiess:3<==AR3S00""))./ /s68cHtjfd}|S)Nctjg|i|}tD] } |cS|S#t$r}|sYd}~nd}~wwxYwt j |J)zThis assumes the function has not already been called. If backoff_iterator is empty, we should still run the function a single time with no delay. N) functoolspartialnextr>r+r.)r/r0call_retryable_functiondelayebackoff_iterator_generatorfunctionshould_retry_errors r run_functionzOretry_with_delays_and_condition..function_wrapper..run_functions~ '0&7&7&R4&R6&R #89 "244 "+, , !-a01 5!s; A AA)rZwraps)rarcr`rbs` rfunction_wrapperz9retry_with_delays_and_condition..function_wrappers&  " - # - r)rM)backoff_iteratorrbrWrer`s ` @rretry_with_delays_and_conditionrgs3 0 "@@P!Q!(& rrP)NN)Nr) r'<)__doc__ __future__rrSrZrQrEr)r+"ansible.module_utils.compat.typing module_utilscompattypingtrrr#r rArJrMrgr9rrrqsP #  .. 62 O%r