Current File : //lib/python3.9/site-packages/ansible/module_utils/__pycache__/api.cpython-39.pyc
a
�)g� � @ s� d Z ddlmZmZmZ eZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
m m mZ ddd�Zddd�Zddd �Zdd
d �Zd d
d�Zd dd�Zdd� Zd dd�ZdS ) 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
� )�absolute_import�division�print_functionNc C s* t t dd�t dd�d�}| r&|�| � |S )z7Creates an argument spec for working with rate limiting�int��type)�rate�
rate_limit��dict�update��specZarg_spec� r �</usr/lib/python3.9/site-packages/ansible/module_utils/api.py�rate_limit_argument_spec' s �
r c C s, t t dd�t ddd�d�}| r(|�| � |S )z2Creates an argument spec for working with retryingr r �float� �r �default��retries� retry_pauser
r
r r r �retry_argument_spec2 s
�
r c C s> t t dd�t ddd�t dd�t ddd�d�}| r:|�| � |S )N�strr T)r Zno_log�boolr )Z api_usernameZ api_passwordZapi_urlZvalidate_certsr
r
r r r �basic_auth_argument_spec= s
�
r c s4 d� | dur$|dur$t |�t | � � � fdd�}|S )zrate limiting decoratorNc s dg�� ��fdd�}|S )Ng c sd t jdkrtj}ntj}�d urR|� �d }�| }|dkrHt�|� |� �d< � | i |��}|S )N)� � r )�sys� version_info�time� process_timeZclock�sleep)�args�kwargsZ real_time�elapsed�left�ret)�f�last�minrater r � ratelimitedR s
z0rate_limit.<locals>.wrapper.<locals>.ratelimitedr )r) r, �r+ )r) r* r �wrapperO s zrate_limit.<locals>.wrapper)r )r r r. r r- r r I s
r r c s � �fdd�}|S )zRetry decoratorc s � ��fdd�}|S )Nc sl d}�d urhd }|d7 }|�kr,t d� ��z� | i |��}W n t yP Y n0 |rXqdt��� q|S d S )Nr r zRetry limit exceeded: %d)� Exceptionr! r# )r$ r% Z retry_countr( )r) r r r r �retriedh s z'retry.<locals>.wrapper.<locals>.retriedr )r) r0 r )r) r r. f s zretry.<locals>.wrapperr )r r r. r r r �retryd s r1 �
r �< c c s0 t d| �D ] }t�dt||d| ��V q
dS )aD The "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.
r � N)�range�randomZrandint�min)r Z
delay_baseZdelay_thresholdr1 r r r �generate_jittered_backoff} s r8 c C s dS )NFr )Zexception_or_resultr r r � retry_never� s r9 c s0 dd� }|| �� ~ �du r t �� �fdd�}|S )z�Generic 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).
c s s& t �| �\}}|V t�|�V qd S )N)� itertools�tee�copy)Zoriginal_iteratorZ_copiable_iteratorZ_first_iterator_copyr r r � _emit_isolated_iterator_copies� s zGretry_with_delays_and_condition.<locals>._emit_isolated_iterator_copiesNc s t �� ��� �fdd��}|S )Nc sr t j�g| �R i |��}t� �D ]J}z |� W S ty^ } z�|�sJ� W Y d}~n
d}~0 0 t�|� q |� S )z�This 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)� functools�partial�nextr/ r! r# )r$ r% Zcall_retryable_functionZdelay�e)�backoff_iterator_generator�function�should_retry_errorr r � run_function� s zOretry_with_delays_and_condition.<locals>.function_wrapper.<locals>.run_function)r> �wraps)rC rE �rB rD )rC r �function_wrapper� s z9retry_with_delays_and_condition.<locals>.function_wrapper)r9 )Zbackoff_iteratorrD r= rH r rG r � retry_with_delays_and_condition� s rI )N)N)N)NN)Nr )r2 r r3 )N)�__doc__Z
__future__r r r r Z
__metaclass__r< r> r: r6 r r! Z"ansible.module_utils.compat.typingZ module_utils�compat�typing�tr r r r r1 r8 r9 rI r r r r �<module>
s"
Mini Shell