U N8c@s@dZddlZddlZddlmZGdddZGdddZdS)zEThis module implements token buckets used for client side throttling.N)CapacityNotAvailableErrorc@s$eZdZddZddZddZdS)ClockcCsdSNselfrr;/tmp/pip-unpacked-wheel-ozje0y8b/botocore/retries/bucket.py__init__ szClock.__init__cCst|dSr)timesleepramountrrrr sz Clock.sleepcCstSr)r rrrr current_timeszClock.current_timeN)__name__ __module__ __qualname__r r rrrrrrsrc@speZdZdZefddZeddZejddZeddZed d Z dd dZ ddZ ddZ ddZ dS) TokenBucketg?cCsFd|_d|_d|_||_d|_||_t|_t |j|_ ||_ dS)Nr) _fill_rate _max_capacity_current_capacity_clock_last_timestamp _min_rate threadingLock_lock Condition_new_fill_rate_conditionmax_rate)rrZclockZmin_raterrrr s zTokenBucket.__init__cCs|jSr)rrrrrr"szTokenBucket.max_ratec Cs\|jL|t||j|_|dkr.||_nd|_t|j|j|_|jW5QRXdS)N) r_refillmaxrrrminrnotify)rvaluerrrr&scCs|jSr)rrrrr max_capacity;szTokenBucket.max_capacitycCs|jSr)rrrrravailable_capacity?szTokenBucket.available_capacityrTc Cs,|j|j||dW5QRSQRXdS)aJAcquire 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)rr r'rrracquireCs zTokenBucket.acquirecCsv|||jkr$|j|8_dS|s.t||}|dkr`|j||||}q8|j|8_dSdS)NTr)r rr _sleep_amountrwait)rr r'Z sleep_amountrrrr(Ps    zTokenBucket._acquirecCs||j|jSr)rrr rrrr*gszTokenBucket._sleep_amountcCsT|j}|jdkr||_dS|j}||j|j}t|j||}||_||_dSr)rrrrrr"r)r timestampZcurrent_capacityZ fill_amountZ new_capacityrrrr js  zTokenBucket._refillN)rT)rrrZ _MIN_RATEr propertyrsetterr%r&r)r(r*r rrrrrs      r)__doc__rr Zbotocore.exceptionsrrrrrrrs