File: //lib/python3/dist-packages/botocore/retries/__pycache__/quota.cpython-310.pyc
o
.&�a� � @ s d Z ddlZG dd� de�ZdS )z Retry quota implementation.
� Nc @ s: e Zd ZdZedfdd�Zdd� Zdd� Zed d
� �ZdS ) �
RetryQuotai� Nc C s& || _ || _|d u rt�� }|| _d S �N)�
_max_capacity�_available_capacity� threading�Lock�_lock)�self�initial_capacity�lock� r �8/usr/lib/python3/dist-packages/botocore/retries/quota.py�__init__ s
zRetryQuota.__init__c C sX | j � || jkr W d � dS | j|8 _ W d � dS 1 s%w Y dS )z�Attempt to aquire a certain amount of capacity.
If there's not sufficient amount of capacity available, ``False``
is returned. Otherwise, ``True`` is returned, which indicates that
capacity was successfully allocated.
NFT)r r )r �capacity_amountr r r
�acquire s
�$�zRetryQuota.acquirec C s\ | j | jkrdS | j� t| j | j |�}| j|7 _W d � dS 1 s'w Y dS )z�Release capacity back to the retry quota.
The capacity being released will be truncated if necessary
to ensure the max capacity is never exceeded.
N)r r r �min)r r �amountr r r
�release" s
�"�zRetryQuota.releasec C s | j S r )r )r r r r
�available_capacity7 s z RetryQuota.available_capacity) �__name__�
__module__� __qualname__�INITIAL_CAPACITYr r r �propertyr r r r r
r s r )�__doc__r �objectr r r r r
�<module> s