U n/e@s6dZddlZddlmZddlmZGdddZdS)zPAn async reimplementation of the blocking elements from botocore.retries.bucket.N)CapacityNotAvailableError)Clockc@sveZdZdZdZefddZeddZddZd d Z ed d Z ed dZ dddZ ddZ ddZddZdS)AsyncTokenBucketz5A reimplementation of TokenBucket that doesn't block.g?cCs<d|_d|_d|_||_d|_||_||t|_ dS)Nr) _fill_rate _max_capacity_current_capacity_clock_last_timestamp _min_rate _set_max_rateasyncio Condition_new_fill_rate_condition)selfmax_rateZclockZmin_rater>/tmp/pip-unpacked-wheel-8mnez3y6/aiobotocore/retries/bucket.py__init__s zAsyncTokenBucket.__init__cCs|jSN)rrrrrr!szAsyncTokenBucket.max_ratec s8|j4IdH|||jW5QIdHRXdSr)rr notifyrvaluerrr set_max_rate%s zAsyncTokenBucket.set_max_ratecCs@|t||j|_|dkr&||_nd|_t|j|j|_dS)N)_refillmaxr rrminrrrrrr *szAsyncTokenBucket._set_max_ratecCs|jSr)rrrrr max_capacity<szAsyncTokenBucket.max_capacitycCs|jSr)rrrrravailable_capacity@sz#AsyncTokenBucket.available_capacityrTc sF|j4IdH(|j||dIdHW5QIdHRSQIdHRXdS)ajAcquire token or return amount of time until next token available. If block is True, then this method will return when there's sufficient capacity to acquire the desired amount. This won't block the event loop. If block is False, then this method will return True if capacity was successfully acquired, False otherwise. N)amountblock)r_acquire)rr r!rrracquireDs zAsyncTokenBucket.acquirecs|||jkr$|j|8_dS|s.t||}|dkrzt|j|IdHWntjk rrYnX|||}q8|j|8_dSdS)NTr) rrr _sleep_amountr wait_forrwait TimeoutError)rr r!Z sleep_amountrrrr"Ps&   zAsyncTokenBucket._acquirecCs||j|jSr)rr)rr rrrr$fszAsyncTokenBucket._sleep_amountcCsT|j}|jdkr||_dS|j}||j|j}t|j||}||_||_dSr)r current_timer rrrr)r timestampZcurrent_capacityZ fill_amountZ new_capacityrrrris  zAsyncTokenBucket._refillN)rT)__name__ __module__ __qualname____doc__Z _MIN_RATErpropertyrrr rrr#r"r$rrrrrr s      r)r-r Zbotocore.exceptionsrZbotocore.retries.bucketrrrrrrs