File: //lib/python3/dist-packages/botocore/retries/__pycache__/bucket.cpython-310.pyc
o
.&�a� � @ sD d Z ddlZddlZddlmZ G dd� de�ZG dd� de�ZdS )zEThis module implements token buckets used for client side throttling.� N)�CapacityNotAvailableErrorc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Clockc C s d S �N� ��selfr r �9/usr/lib/python3/dist-packages/botocore/retries/bucket.py�__init__ s zClock.__init__c C s t �|� d S r )�time�sleep�r �amountr r r r s z Clock.sleepc C s t � � S r )r
r r r r � current_time s zClock.current_timeN)�__name__�
__module__� __qualname__r r r r r r r r s r c @ sp e Zd ZdZefdd�Zedd� �Zejdd� �Zedd� �Zed d
� �Z dd
d�Z
dd� Z dd� Z dd� Z
dS )� TokenBucketg �?c C sF d | _ d | _d| _|| _d | _|| _t�� | _t� | j�| _
|| _ d S )Nr ) �
_fill_rate�
_max_capacity�_current_capacity�_clock�_last_timestamp� _min_rate� threading�Lock�_lock� Condition�_new_fill_rate_condition�max_rate)r r �clock�min_rater r r r s
zTokenBucket.__init__c C � | j S r )r r r r r r " � zTokenBucket.max_ratec C sr | j �, | �� t|| j�| _|dkr|| _nd| _t| j| j�| _| j �� W d � d S 1 s2w Y d S )N� ) r �_refill�maxr r r �minr �notify)r �valuer r r r & s � "�c C r! r )r r r r r � max_capacity< r" zTokenBucket.max_capacityc C r! r )r r r r r �available_capacity@ r" z TokenBucket.available_capacityr# Tc C s8 | j � | j||d�W d � S 1 sw Y dS )aJ Acquire token or return amount of time until next token available.
If block is True, then this method will block until there's sufficient
capacity to acquire the desired amount.
If block is False, then this method will return True is capacity
was successfully acquired, False otherwise.
)r
�blockN)r �_acquire)r r
r+ r r r �acquireD s
$�zTokenBucket.acquirec C sx | � � || jkr| j|8 _dS |st� �| �|�}|dkr3| j�|� | � � | �|�}|dks | j|8 _dS )NTr )r$ r r �
_sleep_amountr �wait)r r
r+ � sleep_amountr r r r, Q s
�
zTokenBucket._acquirec C s || j | j S r )r r r r r r r. h s zTokenBucket._sleep_amountc C sT | j �� }| jd u r|| _d S | j}|| j | j }t| j|| �}|| _|| _d S r )r r r r r r& r )r � timestamp�current_capacity� fill_amount� new_capacityr r r r$ k s
zTokenBucket._refillN)r# T)r r r � _MIN_RATEr �propertyr �setterr) r* r- r, r. r$ r r r r r s
r )�__doc__r
r �botocore.exceptionsr �objectr r r r r r �<module> s